Spaces:
Sleeping
Sleeping
Todas las funciones listas
Browse files- funciones.py +32 -0
- main.py +9 -9
funciones.py
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import time
|
| 2 |
+
import avaimet
|
| 3 |
+
|
| 4 |
+
def getAccess(userfile):
|
| 5 |
+
|
| 6 |
+
#Genera conexi贸n inicial.
|
| 7 |
+
sshListo, sftpListo = avaimet.conecta()
|
| 8 |
+
#Obtiene la caja donde est谩 guardados los tokens.
|
| 9 |
+
caja = avaimet.obtenCaja(userfile)
|
| 10 |
+
#Obtiene los tokens que hay en esa caja.
|
| 11 |
+
tokens = avaimet.obtenTokens(sftpListo, caja)
|
| 12 |
+
#Cierra la conexi贸n.
|
| 13 |
+
avaimet.cierraConexion(sshListo, sftpListo)
|
| 14 |
+
|
| 15 |
+
return tokens
|
| 16 |
+
|
| 17 |
+
def debitTokens(userfile, work):
|
| 18 |
+
|
| 19 |
+
print(f"Task received : {work}, type: {type(work)} ...")
|
| 20 |
+
|
| 21 |
+
#Genera conexi贸n inicial.
|
| 22 |
+
sshListo, sftpListo = avaimet.conecta()
|
| 23 |
+
#Obtiene la caja donde est谩 guardados los tokens.
|
| 24 |
+
caja = avaimet.obtenCaja(userfile)
|
| 25 |
+
#Obtiene los tokens que hay en esa caja.
|
| 26 |
+
tokens = avaimet.obtenTokens(sftpListo, caja)
|
| 27 |
+
#Aplica las reglas de 茅sta app para debitar lo correspondiente.
|
| 28 |
+
resultado_debitado = avaimet.restaToken(sftpListo, caja, tokens, work)
|
| 29 |
+
#Cierra la conexi贸n.
|
| 30 |
+
avaimet.cierraConexion(sshListo, sftpListo)
|
| 31 |
+
|
| 32 |
+
return resultado_debitado
|
main.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
from fastapi import FastAPI
|
| 2 |
-
import
|
| 3 |
|
| 4 |
app = FastAPI()
|
| 5 |
|
|
@@ -8,12 +8,12 @@ def hello():
|
|
| 8 |
|
| 9 |
return {"Status":"Deployed"}
|
| 10 |
|
| 11 |
-
@app.get("/getTokens")
|
| 12 |
-
def getTokens(
|
|
|
|
|
|
|
| 13 |
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
return {"Status":"Debitting tokens..."}
|
|
|
|
| 1 |
from fastapi import FastAPI
|
| 2 |
+
import funciones
|
| 3 |
|
| 4 |
app = FastAPI()
|
| 5 |
|
|
|
|
| 8 |
|
| 9 |
return {"Status":"Deployed"}
|
| 10 |
|
| 11 |
+
@app.get("/getTokens/{userfile}")
|
| 12 |
+
def getTokens(userfile: str):
|
| 13 |
+
tokens = funciones.getAccess(userfile)
|
| 14 |
+
return tokens
|
| 15 |
|
| 16 |
+
@app.get("/debitTokens/{userfile}/{work}")
|
| 17 |
+
def debitTokens(userfile: str, work: str):
|
| 18 |
+
tokens = funciones.debitTokens(userfile,work)
|
| 19 |
+
return tokens
|
|
|
|
|
|