diff --git a/impression/digicode.py b/impression/digicode.py index 0c0c6a3e..486ff6ef 100644 --- a/impression/digicode.py +++ b/impression/digicode.py @@ -76,18 +76,16 @@ def __init__(): # ############################### # save_code # ############################### -# enregistre le codes avec "monString" dans le fichier +# enregistre le codes pour user_name sur l'intranet2 # -def save_code(code, monString=""): - """enregistre le codes avec "monString" dans le fichier """ +def save_code(code, user_name): + """enregistre le codes pour ``user_name``""" 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() + response = requests.post(CREATION_LINK + code, data={'password':digicode_pass, 'user':user_name}, verify=CERTIFICATE, timeout=2) + try: + code = int(response.content) + except (TypeError, ValueError): + raise ValueError(response.content) return code