export script

service/auth_service.py

@@ -2,15 +2,81 @@ import logging as log
 from typing import Type
 import requests
 from sqlalchemy.orm import Session
-from sqlalchemy import select, desc
+from sqlalchemy import select, desc, delete
 
 import constants
 from db.helpers import get_engine
 from db.models import ApiToken
 from constants import *
+import logging
+import datetime
+
+logger = logging.getLogger(__name__)
+
+def get_auth_token() -> str:    
+    delete_old_tokens()
+    all_tokens: list[ApiToken] = get_all_tokens()    
+    
+    if len(all_tokens) == 0:
+        return generate_token().access_token
+    
+    # check if there is at least one active
+    active_tokens: list[ApiToken] = []
+    tokens_to_refresh: list[ApiToken] = []
+    
+    for token in all_tokens:
+        if token.date_expiration_access_token > datetime.datetime.now():
+            active_tokens.append(token)
+        else:
+            tokens_to_refresh.append(token)
+    
+    with Session(get_engine()) as session:        
+        if len(active_tokens) > 0:
+            logger.info("There are active tokens. Returning first and deleting the rest")
+            for idx, active_token in enumerate(active_tokens):
+                if idx > 0:
+                    logger.info(f"More than 1 active token. Deleting auth token.")
+                    session.delete(active_token)
+                    
+           
+            return active_tokens[0].access_token   
+    
+        for idx, ref_token in enumerate(tokens_to_refresh):
+            if idx > 0:
+                logger.info(f"More than 1 token to refresh. Deleting token expired in {ref_token.date_expiration_refresh_token}")
+                session.delete(ref_token)
+                
+        session.commit()
+    
+    return refresh_token(tokens_to_refresh[0]).access_token
+        
+        
+def delete_old_tokens():
+    with Session(get_engine()) as session:
+        logger.info("Deleting old tokens.")
+        delete(ApiToken).where(ApiToken.date_expiration_refresh_token > (datetime.datetime.now() - datetime.timedelta(1)))
+        session.commit()
+                
+def refresh_token(token: ApiToken) -> ApiToken:   
+    response = requests.get(f"{constants.api_url}/auth?refresh_token={token.refresh_token}")
+    refreshed_token = persist_auth_token(response.json())
+    
+    with Session(get_engine()) as session:
+        session.delete(token)
+        session.commit()
+    
+    return refresh_token
+    
 
 
-def generate_token(persist=False):
+    
+def get_all_tokens():
+    with Session(get_engine()) as session:
+        return session.query(ApiToken).all()
+
+
+
+def generate_token() -> ApiToken:
     body = {
         "consumer_key": api_consumer_key,
         "consumer_secret": api_consumer_secret,
@@ -21,30 +87,9 @@ def generate_token(persist=False):
     if response.status_code in [201, 200]:
         log.info('Generated tray access token: %s', response.json())
 
-    if persist:
-        log.info('Persisting token')
-        persist_auth_token(response.json())
+    return persist_auth_token(response.json())
 
-    return response
-
-
-def refresh_access_token():
-    generate_token(persist=True)
-    token = get_last_api_token()
-
-    log.info('Refreshing token')
-    response = requests.get(f"{api_url}/auth?refresh_token={token.refresh_token}").json()
-
-    try:
-        persist_auth_token(response)
-        return response['access_token']
-    except Exception as e:
-        log.error(e)
-
-    return None
-
-
-def persist_auth_token(api_token):
+def persist_auth_token(api_token) -> ApiToken:
     with Session(get_engine()) as session:
         token = ApiToken(message=api_token["message"],
                          code=api_token["code"],
@@ -58,6 +103,8 @@ def persist_auth_token(api_token):
 
         session.add(token)
         session.commit()
+    
+    return token
 
 
 def delete_token(token):