1
0
forked from ScoDoc/ScoDoc
ScoDoc/tests/fakeportal/fakeportal.py
2020-11-26 10:23:52 +01:00

143 lines
4.4 KiB
Python
Executable File

#!/usr/bin/python3
"""Simple fake HTTP serveur
emulating "Apogee" Web service
"""
import os
import sys
import random
import time
from pathlib import Path
import http.server
import socketserver
from urllib.parse import urlparse
from urllib.parse import parse_qs
sys.path.append("..")
from demo.gen_nomprenoms import nomprenom
# Etudiant avec tous les champs (USPN)
ETUD_TEMPLATE_FULL = open("etud_template.xml").read()
# Etudiant avec seulement les champs requis
ETUD_TEMPLATE_MINI = open("etud_minimal_template.xml").read()
ETUD_HEAD = """<?xml version="1.0" encoding="UTF-8"?>
<etudiants>"""
ETUD_TAIL = """</etudiants>
"""
def make_random_etud(nip, etape=None, annee=None, template=ETUD_TEMPLATE_FULL):
"""return XML for a student"""
random.seed(nip) # deterministic choice based on nip
gender = random.choice(("M", "F"))
nom, prenom = nomprenom(gender)
if not etape:
etape = random.choice(("V1RT", "V2RT", "V2RT2", ""))
if not annee:
annee = time.strftime("%Y") # current year
diplome = "VDRT"
data = template.format(
nip=nip,
gender=gender,
nom=nom,
prenom=prenom,
etape=etape,
diplome=diplome,
annee=annee,
)
return data
def make_random_etape_etuds(etape, annee):
"""Liste d'etudiants d'une etape"""
random.seed(etape + annee)
nb = random.randint(0, 50)
L = []
for i in range(nb):
if i % 2:
template = ETUD_TEMPLATE_MINI
else:
template = ETUD_TEMPLATE_FULL
nip = str(random.randint(10000000, 99999999)) # 8 digits
L.append(make_random_etud(nip, etape=etape, annee=annee, template=template))
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/"""
#
return FAKE_FACES_FILENAME[hash(nip) % len(FAKE_FACES_FILENAME)]
class MyHttpRequestHandler(http.server.SimpleHTTPRequestHandler):
def send_xml(self, data):
self.send_response(200)
self.send_header("Content-type", "text/xml;charset=UTF-8")
self.end_headers()
self.wfile.write(bytes(data, "utf8"))
def do_GET(self):
query_components = parse_qs(urlparse(self.path).query)
print(f"path={self.path}", file=sys.stderr)
print(query_components, file=sys.stderr)
if "etapes" in self.path.lower():
self.path = "etapes.xml"
elif "scodocEtudiant" in self.path:
# 2 forms: nip=xxx or etape=eee&annee=aaa
if "nip" in query_components:
nip = query_components["nip"][0]
print(f"requesting nip={nip}")
data = ETUD_HEAD + make_random_etud(nip) + ETUD_TAIL
return self.send_xml(data)
elif "etape" in query_components:
etape = query_components["etape"][0]
print(f"requesting etape={etape}", file=sys.stderr)
if "annee" in query_components:
annee = query_components["annee"][0]
data = ETUD_HEAD + make_random_etape_etuds(etape, annee) + ETUD_TAIL
return self.send_xml(data)
else:
print(
f"Error 404: (missing annee) path={self.path}", file=sys.stderr
)
self.send_response(404)
return
else:
print(
f"Error 404: (missing nip or etape) path={self.path}",
file=sys.stderr,
)
self.send_response(404)
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))
else:
print(f"Error 404: path={self.path}")
self.send_response(404)
return
# Sending an '200 OK' response
self.send_response(200)
http.server.SimpleHTTPRequestHandler.do_GET(self)
return
# Create an object of the above class
handler_object = MyHttpRequestHandler
PORT = 8678
my_server = socketserver.TCPServer(("", PORT), handler_object)
if __name__ == "__main__":
# Start the server
print(f"Server listening on port {PORT}...")
my_server.serve_forever()