96 lines
2.5 KiB
Python
96 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
wakatime.main
|
|
~~~~~~~~~~~~~
|
|
|
|
Module entry point.
|
|
|
|
:copyright: (c) 2013 Alan Hamlett.
|
|
:license: BSD, see LICENSE for more details.
|
|
"""
|
|
|
|
from __future__ import print_function
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
import time
|
|
import traceback
|
|
|
|
pwd = os.path.dirname(os.path.abspath(__file__))
|
|
sys.path.insert(0, os.path.dirname(pwd))
|
|
sys.path.insert(0, os.path.join(pwd, 'packages'))
|
|
|
|
from .__about__ import __version__
|
|
from .api import send_heartbeats, get_time_today
|
|
from .arguments import parse_arguments
|
|
from .compat import u, json
|
|
from .constants import SUCCESS, UNKNOWN_ERROR, HEARTBEATS_PER_REQUEST
|
|
from .logger import setup_logging
|
|
|
|
log = logging.getLogger('WakaTime')
|
|
|
|
from .heartbeat import Heartbeat
|
|
from .offlinequeue import Queue
|
|
|
|
|
|
def execute(argv=None):
|
|
if argv:
|
|
sys.argv = ['wakatime'] + argv
|
|
|
|
args, configs = parse_arguments()
|
|
|
|
setup_logging(args, __version__)
|
|
|
|
if args.today:
|
|
text, retval = get_time_today(args, configs)
|
|
if text:
|
|
print(text)
|
|
return retval
|
|
|
|
try:
|
|
heartbeats = []
|
|
|
|
hb = Heartbeat(vars(args), args, configs)
|
|
if hb:
|
|
heartbeats.append(hb)
|
|
else:
|
|
log.debug(hb.skip)
|
|
|
|
if args.extra_heartbeats:
|
|
try:
|
|
for extra_data in json.loads(sys.stdin.readline()):
|
|
hb = Heartbeat(extra_data, args, configs)
|
|
if hb:
|
|
heartbeats.append(hb)
|
|
else:
|
|
log.debug(hb.skip)
|
|
except json.JSONDecodeError as ex:
|
|
log.warning(u('Malformed extra heartbeats json: {msg}').format(
|
|
msg=u(ex),
|
|
))
|
|
|
|
retval = SUCCESS
|
|
while heartbeats:
|
|
retval = send_heartbeats(heartbeats[:HEARTBEATS_PER_REQUEST], args, configs)
|
|
heartbeats = heartbeats[HEARTBEATS_PER_REQUEST:]
|
|
if retval != SUCCESS:
|
|
break
|
|
|
|
if heartbeats:
|
|
Queue(args, configs).push_many(heartbeats)
|
|
|
|
if retval == SUCCESS:
|
|
queue = Queue(args, configs)
|
|
for offline_heartbeats in queue.pop_many(args.sync_offline_activity):
|
|
time.sleep(1)
|
|
retval = send_heartbeats(offline_heartbeats, args, configs)
|
|
if retval != SUCCESS:
|
|
break
|
|
|
|
return retval
|
|
|
|
except:
|
|
log.traceback(logging.ERROR)
|
|
print(traceback.format_exc())
|
|
return UNKNOWN_ERROR
|