Moibe commited on
Commit
c51594b
1 Parent(s): 0bf4b92

Todas las funciones listas

Browse files
Files changed (2) hide show
  1. funciones.py +32 -0
  2. 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 aplicacion
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
- return {"Status":"Getting Tokens..."}
15
-
16
- @app.get("/debitTokens")
17
- def debitTokens():
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