helpers.py

#!/usr/bin/env python3
"""
__author__ = "Praveen Kumar Pendyala"
__email__ = "mail@pkp.io"
"""
import logging
import json
import fitbit
import httplib2
from oauth2client.file import Storage
from apiclient.discovery import build


class Helper(object):
	"""Helper methods to hide trivial methods"""

	def __init__(self, fitbitCredsFile, googleCredsFile):
		""" Intialize a helper object.

		fitbitCredsFile -- Fitbit credentials file
		googleCredsFile -- Google Fits credentials file
		"""
		self.fitbitCredsFile = fitbitCredsFile
		self.googleCredsFile = googleCredsFile

	def GetFitbitClient(self):
		"""Returns an authenticated fitbit client object"""
		logging.debug("Creating Fitbit client")
		credentials = json.load(open(self.fitbitCredsFile))
		client = fitbit.Fitbit(
			refresh_cb = lambda token: self.UpdateFitbitCredentials(token),
			**credentials)

		# Use v1.2 sleep API: https://github.com/orcasgit/python-fitbit/issues/128
		client.API_VERSION = 1.2

		logging.debug("Fitbit client created")
		return client

	def GetGoogleClient(self):
		"""Returns an authenticated google fit client object"""
		logging.debug("Creating Google client")
		credentials = Storage(self.googleCredsFile).get()
		http = credentials.authorize(httplib2.Http())
		client = build('fitness', 'v1', http=http)
		logging.debug("Google client created")
		return client

	def UpdateFitbitCredentials(self, token):
		"""Persists new fitbit credentials to local storage

		fitbitClient -- fitbit client object that contains the latest credentials
		"""
		logging.debug("Refreshed Fitbit token")
		credentials = json.load(open(self.fitbitCredsFile))
		for t in ('access_token', 'refresh_token'):
			credentials[t] = token[t]
		json.dump(credentials, open(self.fitbitCredsFile, 'w'))