time_api.py

"""This module connect aTimeLogger API."""
import requests
import json
import arrow
from datetime import datetime
from requests.auth import HTTPBasicAuth

INTERVAL_MAX = 100000
DAYS_NEW = 2


def get_auth_header():
    """
    Generate basic auth header for aTimeLogger.
    See the post for more info about the API, http://blog.timetrack.io/rest-api/
    """

    # get username and password from text file
    lines = [line.rstrip('\n') for line in open('pass.txt')]
    username, password, dev_token = lines
    return HTTPBasicAuth(username, password)


def get_types(auth_header):
    """
    Retrieve types data from aTimeLogger.
    :param auth_header: auth header for request data.
    :return: A list of dict for types data.
    """
    r_type = requests.get("https://app.atimelogger.com/api/v2/types",
                          auth=auth_header)
    types = json.loads(r_type.text)
    return types['types']


def get_all_intervals(auth_header):
    """
    Retrieve all intervals data from aTimeLogger. Number of entries is limited by INTERVAL_MAX.
    :param auth_header: auth header for request data.
    :return: A list of dict for intervals data.
    """
    r_interval = requests.get("https://app.atimelogger.com/api/v2/intervals",
                              params={'limit': INTERVAL_MAX, 'order': 'asc'},
                              auth=auth_header)
    intervals = json.loads(r_interval.text)
    return intervals['intervals']


def get_new_intervals(auth_header):
    """
    Retrieve new intervals data from aTimeLogger. New intervals is defined by DAYS_NEW.
    :param auth_header: auth header for request data.
    :return: A list of dict for intervals data.
    """
    now = arrow.get(datetime.now(), 'US/Eastern')
    start = now.replace(days=-DAYS_NEW).floor('day')
    r_interval = requests.get("https://app.atimelogger.com/api/v2/intervals",
                              params={'from': str(start.timestamp), 'to': str(now.timestamp)},
                              auth=auth_header)
    intervals = json.loads(r_interval.text)
    return intervals['intervals']