simplify coding time today function

This commit is contained in:
Alan Hamlett 2019-05-07 00:50:08 -07:00
parent 7a82f3c846
commit 23e0778392
2 changed files with 8 additions and 8 deletions

View file

@ -163,7 +163,7 @@ def send_heartbeats(heartbeats, args, configs, use_ntlm_proxy=False):
return AUTH_ERROR if code == 401 else API_ERROR return AUTH_ERROR if code == 401 else API_ERROR
def get_coding_time(start, end, args, use_ntlm_proxy=False): def get_time_today(args, use_ntlm_proxy=False):
"""Get coding time from WakaTime API for given time range. """Get coding time from WakaTime API for given time range.
Returns total time as string or `None` when unable to fetch summary from Returns total time as string or `None` when unable to fetch summary from
@ -207,8 +207,8 @@ def get_coding_time(start, end, args, use_ntlm_proxy=False):
ssl_verify = args.ssl_certs_file ssl_verify = args.ssl_certs_file
params = { params = {
'start': start, 'start': 'today',
'end': end, 'end': 'today',
} }
# send request to api # send request to api
@ -219,7 +219,7 @@ def get_coding_time(start, end, args, use_ntlm_proxy=False):
verify=ssl_verify) verify=ssl_verify)
except RequestException: except RequestException:
if should_try_ntlm: if should_try_ntlm:
return get_coding_time(start, end, args, use_ntlm_proxy=True) return get_time_today(args, use_ntlm_proxy=True)
session_cache.delete() session_cache.delete()
if log.isEnabledFor(logging.DEBUG): if log.isEnabledFor(logging.DEBUG):
@ -233,7 +233,7 @@ def get_coding_time(start, end, args, use_ntlm_proxy=False):
except: # delete cached session when requests raises unknown exception except: # delete cached session when requests raises unknown exception
if should_try_ntlm: if should_try_ntlm:
return get_coding_time(start, end, args, use_ntlm_proxy=True) return get_time_today(args, use_ntlm_proxy=True)
session_cache.delete() session_cache.delete()
if log.isEnabledFor(logging.DEBUG): if log.isEnabledFor(logging.DEBUG):
@ -264,7 +264,7 @@ def get_coding_time(start, end, args, use_ntlm_proxy=False):
return None, API_ERROR return None, API_ERROR
else: else:
if should_try_ntlm: if should_try_ntlm:
return get_coding_time(start, end, args, use_ntlm_proxy=True) return get_time_today(args, use_ntlm_proxy=True)
session_cache.delete() session_cache.delete()
log.debug({ log.debug({

View file

@ -22,7 +22,7 @@ sys.path.insert(0, os.path.dirname(pwd))
sys.path.insert(0, os.path.join(pwd, 'packages')) sys.path.insert(0, os.path.join(pwd, 'packages'))
from .__about__ import __version__ from .__about__ import __version__
from .api import send_heartbeats, get_coding_time from .api import send_heartbeats, get_time_today
from .arguments import parse_arguments from .arguments import parse_arguments
from .compat import u, json from .compat import u, json
from .constants import SUCCESS, UNKNOWN_ERROR, HEARTBEATS_PER_REQUEST from .constants import SUCCESS, UNKNOWN_ERROR, HEARTBEATS_PER_REQUEST
@ -43,7 +43,7 @@ def execute(argv=None):
setup_logging(args, __version__) setup_logging(args, __version__)
if args.show_time_today: if args.show_time_today:
text, retval = get_coding_time('today', 'today', args) text, retval = get_time_today(args)
if text: if text:
print(text) print(text)
return retval return retval