7122f040e6
* digicode.py appelle désormais la fonction create de l'intranet 2. * digicode_server2.py est renommé en digicode_server.py. * Edition de gen_code.py pour le rendre compatible avec le nouveau digicode.py.
151 lines
4.7 KiB
Python
151 lines
4.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# #############################################################
|
|
# ..
|
|
# .... ............ ........
|
|
# . ....... . .... ..
|
|
# . ... .. .. .. .. ..... . ..
|
|
# .. .. ....@@@. .. . ........ .
|
|
# .. . .. ..@.@@..@@. .@@@@@@@ @@@@@@. ....
|
|
# .@@@@. .@@@@. .@@@@..@@.@@..@@@..@@@..@@@@.... ....
|
|
# @@@@... .@@@.. @@ @@ .@..@@..@@...@@@. .@@@@@. ..
|
|
# .@@@.. . @@@. @@.@@..@@.@@..@@@ @@ .@@@@@@.. .....
|
|
# ...@@@.... @@@ .@@.......... ........ ..... ..
|
|
# . ..@@@@.. . .@@@@. .. ....... . .............
|
|
# . .. .... .. .. . ... ....
|
|
# . . .... ............. .. ...
|
|
# .. .. ... ........ ... ...
|
|
# ................................
|
|
#
|
|
# #############################################################
|
|
"""
|
|
digicode.py
|
|
|
|
Fonctions pour controler le digicode du 4@J (simulation)
|
|
|
|
Copyright (c) 2006, 2007, 2008, 2009 by Cr@ns (http://www.crans.org)
|
|
"""
|
|
import sys
|
|
import time
|
|
import tempfile
|
|
import os
|
|
import commands
|
|
import string
|
|
import random
|
|
import requests
|
|
sys.path.append("/usr/scripts/")
|
|
import cranslib.utils.files
|
|
import secrets_new
|
|
|
|
digicode_pass = secrets_new.get("digicode_pass")
|
|
# #############################################################
|
|
# CONSTANTES
|
|
# #############################################################
|
|
CODES_SERVERS = ["zamok", "asterisk"]
|
|
CODES_DIR = "/usr/scripts/var/digicode/"
|
|
CREATION_LINK = "https://intranet2.crans.org/digicode/create/"
|
|
CERTIFICATE = "/etc/ssl/certs/cacert.org.pem"
|
|
|
|
# #############################################################
|
|
# EXCEPTIONS
|
|
# #############################################################
|
|
def CodeAlreadyExists(Exception):
|
|
pass
|
|
|
|
# #############################################################
|
|
# FONCTIONS
|
|
# #############################################################
|
|
# test pour voir si on est bien sur la bonne machine
|
|
import socket
|
|
if socket.gethostname() not in CODES_SERVERS:
|
|
CODES_DIR = tempfile.mkdtemp(prefix='fake_digicode')
|
|
FAKE_DIR = True
|
|
raise EnvironmentError("La manipulation des codes pour le digicode n'est possible que sur %s" %
|
|
', '.join(CODES_SERVERS))
|
|
else:
|
|
FAKE_DIR = False
|
|
|
|
def __del__():
|
|
if FAKE_DIR:
|
|
os.rmdir(CODES_DIR)
|
|
|
|
|
|
def __init__():
|
|
pass
|
|
# ###############################
|
|
# save_code
|
|
# ###############################
|
|
# enregistre le codes avec "monString" dans le fichier
|
|
#
|
|
def save_code(code, monString=""):
|
|
"""enregistre le codes avec "monString" dans le fichier """
|
|
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):
|
|
"""On contacte l'intranet 2 pour générer le code et on récupère le résultat"""
|
|
response = requests.post(CREATION_LINK, data={'password':digicode_pass, 'user':user_name}, verify=CERTIFICATE, timeout=2)
|
|
return response.content
|
|
|
|
|
|
# ###############################
|
|
# list_code
|
|
# ###############################
|
|
# liste les codes et leur age en secondes
|
|
#
|
|
def list_code():
|
|
"""
|
|
Renvoie la liste des codes existants.
|
|
La liste est sous la forme [(code, age (en sec), contenu du fichier),...]
|
|
"""
|
|
files = os.listdir(CODES_DIR)
|
|
code_list = []
|
|
for aCode in files:
|
|
age = cranslib.utils.files.ageOfFile(os.path.join(CODES_DIR, aCode ) )
|
|
content = read_code_file(aCode)
|
|
code_list.append((aCode, age, content ) )
|
|
return code_list
|
|
|
|
def get_codes(login):
|
|
return [code for (code, age, uid) in list_code() if uid == login]
|
|
|
|
def read_code_file(code):
|
|
"""
|
|
Lis le fichier correspondant au code.
|
|
Renvoie le contenu du fichier.
|
|
"""
|
|
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():
|
|
"""
|
|
Supprime les codes vieux de plus de 24h
|
|
"""
|
|
fileList = os.listdir(CODES_DIR)
|
|
for aFile in fileList:
|
|
aFilePath = os.path.join(CODES_DIR, aFile)
|
|
if os.path.isfile(aFilePath):
|
|
if cranslib.utils.files.fileIsOlderThan(aFilePath, days=1):
|
|
os.remove(aFilePath)
|