From 84e991cef0e718b58db3cc29892b8fa798ccfdfc Mon Sep 17 00:00:00 2001
From: Emmanuel Viennet <emmanuel.viennet@gmail.com>
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 7db85a73..43ae68a5 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 = """<?xml version="1.0" encoding="UTF-8"?>
 <etudiants>"""
@@ -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