From d581d0ced2fae6cf71f60b4f384dbbf8d05764b4 Mon Sep 17 00:00:00 2001
From: Luna <git@l4.pm>
Date: Thu, 6 Jun 2019 13:59:35 -0300
Subject: [PATCH] pull vim-rana changes into rana-cli

---
 .gitignore       |  2 ++
 wakatime/api.py  | 19 +++++++++++--------
 wakatime/main.py |  2 +-
 3 files changed, 14 insertions(+), 9 deletions(-)

diff --git a/.gitignore b/.gitignore
index 077173e..07307e8 100644
--- a/.gitignore
+++ b/.gitignore
@@ -38,3 +38,5 @@ virtualenv
 venv/
 venv3/
 .DS_Store
+
+.mypy_cache/
diff --git a/wakatime/api.py b/wakatime/api.py
index 16286a4..dc56f42 100644
--- a/wakatime/api.py
+++ b/wakatime/api.py
@@ -38,6 +38,11 @@ except ImportError:  # pragma: nocover
     sys.exit(UNKNOWN_ERROR)
 
 
+def _make_url(configs, path):
+    base_url = configs.get('settings', 'base_url') or 'https://api.wakatime.com'
+    return '%s%s' % (base_url, path)
+
+
 def send_heartbeats(heartbeats, args, configs, use_ntlm_proxy=False):
     """Send heartbeats to WakaTime API.
 
@@ -47,9 +52,7 @@ def send_heartbeats(heartbeats, args, configs, use_ntlm_proxy=False):
     if len(heartbeats) == 0:
         return SUCCESS
 
-    api_url = args.api_url
-    if not api_url:
-        api_url = 'https://api.wakatime.com/api/v1/users/current/heartbeats.bulk'
+    api_url = _make_url(configs, '/api/v1/users/current/heartbeats.bulk')
     log.debug('Sending heartbeats to api at %s' % api_url)
     timeout = args.timeout
     if not timeout:
@@ -163,7 +166,7 @@ def send_heartbeats(heartbeats, args, configs, use_ntlm_proxy=False):
     return AUTH_ERROR if code == 401 else API_ERROR
 
 
-def get_time_today(args, use_ntlm_proxy=False):
+def get_time_today(args, configs, use_ntlm_proxy=False):
     """Get coding time from WakaTime API for given time range.
 
     Returns total time as string or `None` when unable to fetch summary from
@@ -171,7 +174,7 @@ def get_time_today(args, use_ntlm_proxy=False):
     fetch summary.
     """
 
-    url = 'https://api.wakatime.com/api/v1/users/current/summaries'
+    url = _make_url(configs, '/api/v1/users/current/summaries')
     timeout = args.timeout
     if not timeout:
         timeout = 60
@@ -219,7 +222,7 @@ def get_time_today(args, use_ntlm_proxy=False):
                                verify=ssl_verify)
     except RequestException:
         if should_try_ntlm:
-            return get_time_today(args, use_ntlm_proxy=True)
+            return get_time_today(args, configs, use_ntlm_proxy=True)
 
         session_cache.delete()
         if log.isEnabledFor(logging.DEBUG):
@@ -233,7 +236,7 @@ def get_time_today(args, use_ntlm_proxy=False):
 
     except:  # delete cached session when requests raises unknown exception
         if should_try_ntlm:
-            return get_time_today(args, use_ntlm_proxy=True)
+            return get_time_today(args, configs, use_ntlm_proxy=True)
 
         session_cache.delete()
         if log.isEnabledFor(logging.DEBUG):
@@ -264,7 +267,7 @@ def get_time_today(args, use_ntlm_proxy=False):
             return None, API_ERROR
     else:
         if should_try_ntlm:
-            return get_time_today(args, use_ntlm_proxy=True)
+            return get_time_today(args, configs, use_ntlm_proxy=True)
 
         session_cache.delete()
         log.debug({
diff --git a/wakatime/main.py b/wakatime/main.py
index 02a40dc..a17b626 100644
--- a/wakatime/main.py
+++ b/wakatime/main.py
@@ -43,7 +43,7 @@ def execute(argv=None):
     setup_logging(args, __version__)
 
     if args.today:
-        text, retval = get_time_today(args)
+        text, retval = get_time_today(args, configs)
         if text:
             print(text)
         return retval