Se agregó nuevo endpoint Authorize

avaimet.py

@@ -45,6 +45,35 @@ def obtenTokens(sftp, caja):
 
       return tokens
     
+def autoriza(tokens, work):
+
+  print(tokens)
+
+  #Standard cost.
+  costo_tarea = 2
+
+  #Aplica reglas de cobro de tokens.
+  #Posteriormente las equivalencias de tardeas y costos vendrán de una tabla aparte.
+  #Por ahora se definen via éste IF: 
+  if work == 'picswap':
+    costo_tarea = 1
+    print(f"Work: {work}, tokens cost: {costo_tarea}")
+    time.sleep(1)
+  else:
+    print("The work specified doesn't exists.")
+
+  #Ahora evaluaremos si se tiene el suficiente crédito como para ejecutar la tarea.
+  if tokens >= costo_tarea:
+    print("Tarea autorizada...")
+    result = True
+  else:
+     print("Tarea no autorizada, no tienes suficientes tokens...")
+     result = False
+  
+  return result
+
+
+    
 def restaToken(sftp, caja, tokens, work):
 
   #Standard cost.

funciones.py

@@ -1,7 +1,7 @@
 import time
 import avaimet
 
-def getAccess(userfile):
+def getTokens(userfile):
 
     #Genera conexión inicial.       
     sshListo, sftpListo = avaimet.conecta()
@@ -14,6 +14,25 @@ def getAccess(userfile):
     
     return tokens
 
+def authorize(userfile, work):
+
+    print(f"Task received : {work}, type: {type(work)} ...")
+
+    #Genera conexión inicial.
+    sshListo, sftpListo = avaimet.conecta()
+    #Obtiene la caja donde está guardados los tokens.
+    caja = avaimet.obtenCaja(userfile)
+    #Obtiene los tokens que hay en esa caja.
+    tokens = avaimet.obtenTokens(sftpListo, caja)
+    
+    #True si autoriza o false si no autoriza.
+    result = avaimet.autoriza(tokens, work)
+    
+    #Cierra la conexión.  
+    avaimet.cierraConexion(sshListo, sftpListo)
+
+    return result
+
 def debitTokens(userfile, work):
 
     print(f"Task received : {work}, type: {type(work)} ...")

main.py

@@ -8,24 +8,49 @@ def start():
 
     return {"Status":"Deployed"}
 
+## GET TOKENS ##
+
+#Vía Path
 @app.get("/getTokens/{userfile}")
 def getTokens(userfile: str):
-    tokens = funciones.getAccess(userfile)
+    tokens = funciones.getTokens(userfile)
     print("Tipo de resultado:", type(tokens))
     return tokens
 
+#Vía Query 
 @app.get("/getTokensQ/")
 def getTokens(userfile: str = "gAAAAABmEZA4SLBC2YczouOrjIEi9WNCNGOIvyUcqBUnzxNsftXTdy54KaX9x8mAjFkABSI6FJrdZDQKk_5lpJOgJoMChxlniw=="):
-    tokens = funciones.getAccess(userfile)
+    tokens = funciones.getTokens(userfile)
     print("Tipo de resultado:", type(tokens))
     return tokens
 
+## AUTHORIZE WORK ##
+
+#Vía Parameters
+@app.get("/authorize/{tokens}/{work}")
+def getTokens(tokens: int, work: str):
+    autorizacion = funciones.authorize(tokens, work)
+    print("Tipo de resultado:", type(autorizacion))
+    return autorizacion
+
+#Vía Query
+@app.get("/authorizeQ/")
+def debitTokens(tokens: int, work: str = "picswap"):
+    autorizacion = funciones.authorize(tokens,work) 
+    print("Tipo de resultado:", type(autorizacion))
+    return tokens
+
+
+## DEBIT TOKENS ##
+
+#Vía Parámeters
 @app.get("/debitTokens/{userfile}/{work}")
 def debitTokens(userfile: str, work: str):
     tokens = funciones.debitTokens(userfile,work) 
     print("Tipo de resultado:", type(tokens))
     return tokens
 
+#Vía Query
 @app.get("/debitTokensQ/")
 def debitTokens(userfile: str, work: str = "picswap"):
     tokens = funciones.debitTokens(userfile,work)