All Quota Related Ok!

avaimet.py

@@ -3,6 +3,7 @@ import tools
 import paramiko
 import nycklar.nodes as nodes
 
+
 #AVAIMET CONTIENE LAS FUNCIONES QUE INTERACTUAN CON EL SERVIDOR REMOTO.
 def conecta():
 
@@ -13,14 +14,16 @@ def conecta():
   #Future: Para usar ésto el método connnect necesitaría aceptar la pk como var string.
   #go = os.getenv("go")
   
+  #llave = paramiko.RSAKey(data=base64.b64decode(nycklar.go.texto))
+  #clientPrivateKey = paramiko.RSAKey.from_private_key(nycklar.go.texto)
 
   #Ahora obtendremos nuestra secret key para poder entrar a ese servidor.
   project_dir = os.getcwd()
-  print("Ésto es project dir: ", project_dir)
   #Ruta de go.
   key_filename = os.path.join(project_dir, "nycklar", "go")  
   
   ssh.connect(nodes.realm, username=nodes.master, key_filename=key_filename)
+  #ssh.connect(nodes.realm, username=nodes.master, pkey=llave)
   sftp = ssh.open_sftp()
 
   return ssh, sftp
@@ -119,6 +122,36 @@ def restaToken(sftp, caja, tokens, work):
     # Escribir el contenido final en el archivo
     archivo.write(contenido_final)
 
+  contenido_final = int(contenido_final)
+  return 
+
+def modificaQuota(sftp, dir_quota, quota, costo_proceso):
+
+  print("Éste es quota: ", quota)
+  print("Éste es costo_proceso: ", costo_proceso)
+
+  #Standard cost.
+  #cuantos = 30
+
+  #Aplica reglas de quotas dinámicas.
+  # if work == 'picswap':
+  #   cuantos = 1
+  #   print(f"Work: {work}, tokens cost: {cuantos}")
+  # else:
+  #   print("The work specified doesn't exists.")
+
+  # Agregar el texto "- Revisado." al string
+  contenido_final = int(quota) - int(costo_proceso)
+  contenido_final = str(contenido_final)
+
+  # Imprimir el contenido
+  print(contenido_final)
+
+  #Actualiza el nuevo valor en el servidor en modo escritura.
+  with sftp.open(dir_quota, 'w') as archivo:
+    # Escribir el contenido final en el archivo
+    archivo.write(contenido_final)
+
   contenido_final = int(contenido_final)
   return contenido_final
 

funciones.py

@@ -75,13 +75,48 @@ def debitTokens(userfile, work, env):
 
     return resultado_debitado
 
+def getQuota():
+
+    print(f"Estoy en getQuota...")
+
+    #Genera conexión inicial.
+    sshListo, sftpListo = avaimet.conecta()
+    dir_quota = nodes.quota + globales.quota
+    print("Ésto es dir quota: ", dir_quota)
+
+    #Obtiene el json con los datos.
+    data = avaimet.obtenContenidoArchivo(sftpListo, dir_quota)
+    print("Y ésto es data: ", data)
+    #Cierra la conexión.  
+    avaimet.cierraConexion(sshListo, sftpListo)
+
+    return data
+
+def updateQuota(costo_proceso):
+
+    print(f"Estoy en updateQuota...")
+
+    #Genera conexión inicial.
+    sshListo, sftpListo = avaimet.conecta()
+    dir_quota = nodes.quota + globales.quota
+    print("Ésto es dir quota: ", dir_quota)
+
+    #Obtiene el json con los datos.
+    quota_actual = avaimet.obtenContenidoArchivo(sftpListo, dir_quota)
+    print("Y ésto es data: ", quota_actual)
+
+    #Aplica las reglas de ésta app para debitar lo correspondiente.
+    quota_modificada = avaimet.modificaQuota(sftpListo, dir_quota, quota_actual, costo_proceso)
+    #Cierra la conexión.  
+    avaimet.cierraConexion(sshListo, sftpListo)
+
+    return quota_modificada
+
 
 
 def getUserNovelty(userfile, aplicacion):
 
-    #print("Estoy en getUserNovelty")
     usuario = tools.decompileUser(userfile)
-    #print("El usuario es...", usuario)
 
     #Genera conexión inicial (general para cualquier función.)  
     sshListo, sftpListo = avaimet.conecta()

globales.py

@@ -1,2 +1,3 @@
 novelty = "/novelty.py"
 data = "/data.py"
+quota = "/quota.txt"

main.py

@@ -60,6 +60,31 @@ def debitTokens(userfile: str, work: str = "picswap", env: str = "dev"):
     tokens = funciones.debitTokens(userfile,work, env) 
     return tokens
 
+## CONTROL QUOTA ##
+#Vía Parámeters
+@app.get("/getQuota/")
+def getQuota():
+    quota = funciones.getQuota() 
+    return quota
+
+#Vía Query
+@app.get("/getQuota/")
+def getQuota():
+    quota = funciones.getQuota() 
+    return quota
+
+#Vía Parámeters
+@app.get("/updateQuota/{costo_proceso}")
+def updateQuota(costo_proceso: int):
+    quota = funciones.updateQuota(costo_proceso) 
+    return quota
+
+#Vía Query
+@app.get("/updateQuota/")
+def updateQuota(costo_proceso: int):
+    quota = funciones.updateQuota(costo_proceso) 
+    return quota
+
 ## GET USER Novelty ##
 #Vía Parámeters
 @app.get("/getUserNovelty/{userfile}/{aplicacion}")

nycklar/nodes.py

@@ -2,4 +2,5 @@ master = "root"
 realm = "165.22.53.200"
 key=b'kiSHsm7Y0hbyNdcFIav6OMT39gi29nIzT-rCrr0Tyc8=' #importante conn con server no es decompiler.
 users_data = "chips/sulkusers-data/"
-users_credits = "chips/sulkusers-credits/"
+users_credits = "chips/sulkusers-credits/"
+quota = "chips/sulku-quota"