Update

app.py

@@ -195,11 +195,12 @@ 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")
+@app.get("/reconnect", tags=["Misc"], summary="Reconnects to ETSI portal for document access", include_in_schema=False)
 def reconnect():
-    etsi_doc_finder.connect()
-    return {"status": "OK !"}
-
+    data = etsi_doc_finder.connect()
+    if data.get('error', None) and not data.get('error'):
+        return data['message']
+    raise HTTPException(status_code=400, detail=data['message'])
 @app.post("/find/single", response_model=DocResponse, tags=["Document Retrieval"], summary="Retrieve a single document by ID", responses={
              200: {
                  "description": "Document found successfully",

classes.py

@@ -1,3 +1,4 @@
+from fastapi import HTTPException
 import requests
 import re
 from bs4 import BeautifulSoup
@@ -7,13 +8,16 @@ import json
 class ETSIDocFinder:
     def __init__(self):
         self.main_ftp_url = "https://docbox.etsi.org/SET"
-        self.session = self.connect()
+        req_data = self.connect()
+        print(req_data['message'])
+        self.session = req_data['session']
     
     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
+        if req.text == "Failed":
+            return {"error": True, "session": session, "message": "Login failed ! Check your credentials"}
+        return {"error": False, "session": session, "message": "Login successful"}
     
     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