From 84e991cef0e718b58db3cc29892b8fa798ccfdfc Mon Sep 17 00:00:00 2001 From: Emmanuel Viennet Date: Fri, 3 Sep 2021 09:00:43 +0200 Subject: [PATCH] Ameliore portail de test: photos et interruptions --- tools/fakeportal/fakeportal.py | 43 ++++++++++++++++++++++++---------- 1 file changed, 31 insertions(+), 12 deletions(-) diff --git a/tools/fakeportal/fakeportal.py b/tools/fakeportal/fakeportal.py index 7db85a7395..43ae68a5c3 100755 --- a/tools/fakeportal/fakeportal.py +++ b/tools/fakeportal/fakeportal.py @@ -9,18 +9,22 @@ from urllib.parse import urlparse import http.server import os import random +import signal import socketserver import sys import time from gen_nomprenoms import nomprenom -cur_dir = Path(os.path.abspath(__file__)).parent +script_dir = Path(os.path.abspath(__file__)).parent + +# Les "photos" des étudiants +FAKE_FACES_FILENAMES = list((script_dir / "faces").glob("*.jpg")) # Etudiant avec tous les champs (USPN) -ETUD_TEMPLATE_FULL = open(cur_dir / "etud_template.xml").read() +ETUD_TEMPLATE_FULL = open(script_dir / "etud_template.xml").read() # Etudiant avec seulement les champs requis -ETUD_TEMPLATE_MINI = open(cur_dir / "etud_minimal_template.xml").read() +ETUD_TEMPLATE_MINI = open(script_dir / "etud_minimal_template.xml").read() ETUD_HEAD = """ """ @@ -70,13 +74,16 @@ def make_random_etape_etuds(etape, annee): return "\n".join(L) -FAKE_FACES_FILENAME = list(Path("faces").glob("*.jpg")) - - -def get_photo_filename(nip): - """get an existing filename for a fake photo, found in faces/""" +def get_photo_filename(nip: str) -> str: + """get an existing filename for a fake photo, found in faces/ + Returns a path relative to the current working dir + """ # - return FAKE_FACES_FILENAME[hash(nip) % len(FAKE_FACES_FILENAME)] + nb_faces = len(FAKE_FACES_FILENAMES) + if nb_faces == 0: + print("WARNING: aucun fichier image disponible !") + return "" + return Path(FAKE_FACES_FILENAMES[hash(nip) % nb_faces]).relative_to(Path.cwd()) class MyHttpRequestHandler(http.server.SimpleHTTPRequestHandler): @@ -122,8 +129,8 @@ class MyHttpRequestHandler(http.server.SimpleHTTPRequestHandler): return elif "getPhoto" in self.path or "scodocPhoto" in self.path: nip = query_components["nip"][0] - print(f"photo for nip={nip}") self.path = str(get_photo_filename(nip)) + print(f"photo for nip={nip}: {self.path}") else: print(f"Error 404: path={self.path}") self.send_response(404) @@ -137,9 +144,21 @@ class MyHttpRequestHandler(http.server.SimpleHTTPRequestHandler): PORT = 8678 -my_server = socketserver.TCPServer(("", PORT), MyHttpRequestHandler) + + +def signal_handler(sig, frame): + print("You pressed Ctrl+C!") + raise SystemExit() + + +signal.signal(signal.SIGINT, signal_handler) if __name__ == "__main__": # Start the server print(f"Server listening on port {PORT}...") - my_server.serve_forever() + my_server = socketserver.TCPServer(("", PORT), MyHttpRequestHandler) + try: + my_server.serve_forever() + finally: + print("shutting down...") + my_server.shutdown() \ No newline at end of file