Fix ETSI error

app.py

@@ -195,6 +195,11 @@ valid_etsi_spec_format = re.compile(r'^\d{3} \d{3}(?:-\d+)?')
 def frontend():
     return FileResponse(os.path.join('templates', 'index.html'))
 
+@app.get("/reconnect", tags=["Misc"], summary="Reconnects to ETSI portal for document access")
+def reconnect():
+    etsi_doc_finder.connect()
+    return {"status": "OK !"}
+
 @app.post("/find/single", response_model=DocResponse, tags=["Document Retrieval"], summary="Retrieve a single document by ID", responses={
              200: {
                  "description": "Document found successfully",
@@ -229,9 +234,9 @@ def find_document(request: DocRequest):
     elif valid_3gpp_spec_format.match(document):
         url = get_spec_url(document)
     elif valid_etsi_doc_format.match(document):
-        etsi_doc_finder.search_document(document)
+        url = etsi_doc_finder.search_document(document)
     elif valid_etsi_spec_format.match(document):
-        etsi_spec_finder.search_document(document)
+        url = etsi_spec_finder.search_document(document)
     elif document.startswith("GP"):
         for sp in gp_spec_locations:
             if document.lower() in sp.lower():

classes.py

@@ -7,9 +7,13 @@ import json
 class ETSIDocFinder:
     def __init__(self):
         self.main_ftp_url = "https://docbox.etsi.org/SET"
-        self.session = requests.Session()
-        req = self.session.post("https://portal.etsi.org/ETSIPages/LoginEOL.ashx", verify=False, headers={"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36"}, data=json.dumps({"username": os.environ.get("EOL_USER"), "password": os.environ.get("EOL_PASSWORD")}))
+        self.session = self.connect()
+    
+    def connect(self):
+        session = requests.Session()
+        req = session.post("https://portal.etsi.org/ETSIPages/LoginEOL.ashx", verify=False, headers={"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36"}, data=json.dumps({"username": os.environ.get("EOL_USER"), "password": os.environ.get("EOL_PASSWORD")}))
         print(req.content, req.status_code)
+        return session
     
     def get_workgroup(self, doc: str):
         main_tsg = "SET-WG-R" if any(doc.startswith(kw) for kw in ["SETREQ", "SCPREQ"]) else "SET-WG-T" if any(doc.startswith(kw) for kw in ["SETTEC", "SCPTEC"]) else "SET" if any(doc.startswith(kw) for kw in ["SET", "SCP"]) else None