on evite que l'impression garde longtemps un utilisateur locke en ecriture + fonctions pour la gestion des codes du digicode par l'intranet

darcs-hash:20061009172138-f46e9-a6141b3b4fa7cb02d9253b4436632bccf7859208.gz

lib/impression/digicode.py

@@ -23,9 +23,54 @@
 # Copyright (c) 2006 by www.crans.org                          
 # #############################################################
 import sys, time, tempfile, os, commands, string, random
+import crans.utils.files
 def __init__():
     pass
+# #############################################################
+# CONSTANTES
+# #############################################################
+CODES_SERVER = "zamok"
+CODES_DIR = "/var/impression/codes/"
 
+# test pour voir si on est bien sur la bonne machine
+import socket
+if socket.gethostname() != CODES_SERVER:
+    raise EnvironmentError, "La manipulation des codes pour le digicode n'est possible que sur %s" % CODES_SERVER
+
+# #############################################################
+# EXCEPTIONS
+# #############################################################
+def CodeAlreadyExists(Exception):
+    pass
+
+# #############################################################
+# FONCTIONS
+# #############################################################
+
+
+# ###############################
+# save_code
+# ###############################
+# enregistre le codes avec "monString" dans le fichier 
+#
+def save_code(code, monString=""):
+    code = str(code)
+    if os.path.exists( os.path.join( CODES_DIR, code ) ):
+        raise CodeAlreadyExists
+    # On enregistre le fichier avec le code pour numero
+    codefichier = open( os.path.join( CODES_DIR, code ) , 'w')
+    codefichier.write(monString)
+    codefichier.write("\n")
+    codefichier.close()
+    return code
+
+
+# ###############################
+# gen_code
+# ###############################
+# genere un code aleatoire
+# et l'enregistre 
+#
 def gen_code(user_name):
     """ Genere le code et l'enregistre dans /var/impression/codes pour radius """
     # Generation du code et ecriture du code
@@ -37,7 +82,7 @@ def gen_code(user_name):
         # On genere un code
         code = rand.randint(100000, 999999)
         # Si le code est libre, on sort de la boucle
-        if not os.path.exists("/var/impression/codes/%d" % code):
+        if not os.path.exists( os.path.join( CODES_DIR, str( code ) ) ):
             break
             
     else:
@@ -51,9 +96,40 @@ def gen_code(user_name):
             sys.stderr.write("ERROR: Plus de codes disponibles.\n")
             sys.stderr.write("ERROR: Penser a ouvrir a l'adherent debite...\n")
         return
-
-    # On enregistre le fichier avec le code pour numero
-    codefichier = open("/var/impression/codes/%d" % code, 'w')
-    codefichier.write("Utilisateur %s\n" % user_name)
-    codefichier.close()
+    save_code(code, user_name)
     return code
+
+
+# ###############################
+# list_code
+# ###############################
+# liste les codes et leur age en secondes
+#
+def list_code():
+    files = os.listdir(CODES_DIR)
+    code_list = []
+    for aCode in files:
+        age = crans.utils.files.ageOfFile(os.path.join(CODES_DIR, aCode ) )
+        content = read_code_file(aCode)
+        code_list.append((aCode, age, content ) )
+    return code_list
+
+def read_code_file(code):
+    myfile = open(os.path.join(CODES_DIR, code ), 'r')
+    lineStr = myfile.readline()
+    myfile.close()
+    return lineStr.replace('\n','')
+    
+    
+# ###############################
+# menage
+# ###############################
+# supprime les codes vieux de plus de 24h
+#
+def menage():
+    fileList = os.listdir(CODES_DIR)
+    for aFile in fileList:
+        aFilePath = os.path.join(CODES_DIR, aFile)
+        if os.path.isfile(aFilePath):
+            if crans.utils.files.fileIsOlderThan(aFilePath, days=1):
+                os.remove(aFilePath)