コード例 #1
0
ファイル: handler.py プロジェクト: rohanprabhu/giraffe
 def __init__(self):
     self.ds = DataStore()
     self.dividely = DividelyManager()
     self.ds.connect()
コード例 #2
0
ファイル: handler.py プロジェクト: rohanprabhu/giraffe
class GiraffeHandler:
    ds = None
    dividely = None

    @classmethod
    def __init__(self):
        self.ds = DataStore()
        self.dividely = DividelyManager()
        self.ds.connect()

    @classmethod
    def add_expense(self, user, title, bills):
        try:
            username, password = self.ds.get_credentials(user)

            if(username == None or password == None):
                return False

            credentials = Credentials(username, password)
            short_codes = [x[1] for x in bills]
            friends_list = self.ds.get_friends_list(user, short_codes)
            emails = {x["short_code"]: x["email"] for x in friends_list}
            emails["You"] = "You"

            bill_objects = []

            for bill in bills:
                bill_objects.append((bill[0], emails[bill[1]]))

            self.dividely.add_expense(credentials, title, bill_objects, datetime.today().strftime("%m/%d/%Y"))
        except Exception as e:
            logging.debug("Failed request [%s]" % e)
            try:
            	last_exc_info()
                logging.debug("Last ditch attempt to logout")
                self.dividely.logout()
            except Exception as e:
            	logging.debug("Last ditch logout attempt fail. Need to figure this out immediately [%s]" % e)
                pass

            return False

        return True

    @classmethod
    def get_accounts(self, user):
        username, password = self.ds.get_credentials(user)
        friends = self.ds.get_friends_list(user, None)
        name_map = {x["email"]: x["name"] for x in friends}

        accounts = self.dividely.get_accounts(Credentials(username, password))
        accounts_final = []

        for account in accounts:
        	account["friend"] = name_map[account["friend"]]
        	accounts_final.append(account)

        return accounts_final

    @classmethod
    def __del__(self):
        try:
            self.ds.disconnect()
        except:
            pass