CaesarFinance setup code implemented

CaesarFinance/caesarfinance.py

@@ -6,8 +6,9 @@ class CaesarFinance:
     def __init__(self) -> None:
         self.CLIENT_ID = config('CLIENT_ID')
         self.CLIENT_SECRET = config('CLIENT_SECRET')
+        self.access_token = None
     def create_access_code(self):
-        print(f"https://auth.monzo.com/?client_id={self.CLIENT_ID}&redirect_uri=https://palondomus-caesarai.hf.space/caesaraisavecode&response_type=code&state=state_token")
+        print(f"https://auth.monzo.com/?client_id={self.CLIENT_ID}&redirect_uri=https://palondomus-caesarai.hf.space&response_type=code&state=state_token")
         val = input("Have you created the access code? (y), (n)")
         if val.lower() == "y":
             return True
@@ -16,9 +17,7 @@ class CaesarFinance:
         else:
             return False
 
-    def fetch_access_code(self):
-        resp = requests.get("https://palondomus-caesarai.hf.space/caesaraigetcode")
-        return resp.json()["messagecode"]
+
     def get_access_token(self,code):
         form_data = {'grant_type': 'authorization_code','client_id':self.CLIENT_ID,'client_secret':self.CLIENT_SECRET,
                      "redirect_uri":"https://palondomus-caesarai.hf.space",
@@ -27,6 +26,22 @@ class CaesarFinance:
         server = requests.post('https://api.monzo.com/oauth2/token', data=form_data)
         output = server.json()["access_token"]
         return output
+    def whoami(self):
+        response = requests.get("https://api.monzo.com/ping/whoami",headers={'Authorization': 'Bearer ' + self.access_token})
+        return response.json()
+    def get_accounts(self):
+        response = requests.get("https://api.monzo.com/accounts",headers={'Authorization': 'Bearer ' + self.access_token})
+        return response.json()
+    def phone_auth(self):
+        phone_auth_val = input("Authenticated on phone? (y),(n)")
+        if phone_auth_val.lower() == "y":
+            return True
+        elif phone_auth_val.lower() == "n":
+            return False
+        else:
+            return False
+
+
 
 if __name__ == "__main__":
     caesarfinance = CaesarFinance()
@@ -35,8 +50,22 @@ if __name__ == "__main__":
         access_code_bool = caesarfinance.create_access_code()
 
     #code = caesarfinance.fetch_access_code()
-    #access_token = caesarfinance.get_access_token(code)
-    #print(access_token)
+    access_token_bool = False
+    while access_token_bool == False:
+        code = input("Insert access token:")
+        if "eyJhb" in code:
+            access_token_bool = True
+        elif "eyJhb" not in code:
+            access_token_bool = False
+
+    access_token = caesarfinance.get_access_token(code)
+    caesarfinance.access_token = access_token
+    access_phone_auth = False
+    while access_phone_auth == False:
+        access_phone_auth = caesarfinance.phone_auth()
 
+        
+    accountinfo = caesarfinance.get_accounts()
+    print(accountinfo)
 
     #caesarfinance.get_access_code()

main.py

@@ -55,24 +55,6 @@ def caesaraihome():
 def caesarailogo():
     return FileResponse(f"{CURRENT_DIR}/CaesarAILogo.png")
 
-#@app.get("/caesaraisavecode")
-#def caesaraisavecode(code: Union[str,None]):
-#    if code != None:
-#        with open(f"{CURRENT_DIR}/CaesarFinance/current_code.txt","w+") as f:
-#            f.write(code)
-#        return {"message":"Temporary fetching code stored."}
-#    elif code== None:
-#        return {"message":"code doesn't exist."}
-#@app.get("/caesaraigetcode")
-#def caesaraigetcode():
-#    if "current_code.txt" in os.listdir(f"{CURRENT_DIR}/CaesarFinance"):
-#        f = open(f"{CURRENT_DIR}/CaesarFinance/current_code.txt","r")
-#        code = f.read()
-#        f.close()
-#        os.remove(f"{CURRENT_DIR}/CaesarFinance/current_code.txt")
-#        return {"messagecode":code}
-#    elif "current_code.txt" not in os.listdir(f"{CURRENT_DIR}/CaesarFinance"):
-#        return {"message":"temporary code doesn't exist. Please authenticate"}
 
 @app.websocket("/caesarobjectdetectws")
 async def caesarobjectdetectws(websocket: WebSocket):