nsarchive 3.0.0b1__py3-none-any.whl → 3.0.0b2__py3-none-any.whl

nsarchive/interfaces/_justice.py

@@ -22,11 +22,36 @@ class JusticeInterface(Interface):
             headers = self.default_headers,
         )
 
-        if res.status_code != 200:
-            res.raise_for_status()
+
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
+
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
 
         report = Report(id)
-        report._load(res.json(), f"{self.url}/justice/reports/{id}", self.default_headers)
+        report._load(_data, f"{self.url}/justice/reports/{id}", self.default_headers)
 
         return report
 
@@ -41,14 +66,40 @@ class JusticeInterface(Interface):
             json = payload
         )
 
-        if res.status_code != 200:
-            res.raise_for_status()
 
-        report = Report(NSID(res.json()['id']))
-        report._load(res.json(), f"{self.url}/justice/reports/{report.id}", self.default_headers)
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
+
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
+
+        report = Report(NSID(_data['id']))
+        report._load(_data, f"{self.url}/justice/reports/{report.id}", self.default_headers)
 
         return report
 
+
     """
     PROCÈS
     """
@@ -59,11 +110,36 @@ class JusticeInterface(Interface):
             headers = self.default_headers,
         )
 
-        if res.status_code != 200:
-            res.raise_for_status()
+
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
+
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
 
         lawsuit = Lawsuit(id)
-        lawsuit._load(res.json(), f"{self.url}/justice/lawsuits/{id}", self.default_headers)
+        lawsuit._load(_data, f"{self.url}/justice/lawsuits/{id}", self.default_headers)
 
         return lawsuit
 
@@ -77,14 +153,40 @@ class JusticeInterface(Interface):
             json = payload
         )
 
-        if res.status_code != 200:
-            res.raise_for_status()
 
-        lawsuit = Lawsuit(NSID(res.json()['id']))
-        lawsuit._load(res.json(), f"{self.url}/justice/lawsuits/{report.id}", self.default_headers)
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
+
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
+
+        lawsuit = Lawsuit(NSID(_data['id']))
+        lawsuit._load(_data, f"{self.url}/justice/lawsuits/{report.id}", self.default_headers)
 
         return lawsuit
 
+
     """
     SANCTIONS
     """
@@ -95,11 +197,36 @@ class JusticeInterface(Interface):
             headers = self.default_headers,
         )
 
-        if res.status_code != 200:
-            res.raise_for_status()
+
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
+
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
 
         sanction = Sanction(id)
-        sanction._load(res.json(), f"{self.url}/justice/sanctions/{id}", self.default_headers)
+        sanction._load(_data, f"{self.url}/justice/sanctions/{id}", self.default_headers)
 
         return sanction
 
@@ -113,10 +240,35 @@ class JusticeInterface(Interface):
             json = payload
         )
 
-        if res.status_code != 200:
-            res.raise_for_status()
 
-        sanction = Sanction(NSID(res.json()['id']))
-        sanction._load(res.json(), f"{self.url}/justice/sanctions/{sanction.id}", self.default_headers)
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
+
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
+
+        sanction = Sanction(NSID(_data['id']))
+        sanction._load(_data, f"{self.url}/justice/sanctions/{sanction.id}", self.default_headers)
 
         return sanction

nsarchive/interfaces/_state.py

@@ -38,17 +38,40 @@ class StateInterface(Interface):
         id = NSID(id)
         res = requests.get(f"{self.url}/votes/{id}", headers = self.default_headers)
 
-        if not res:
-            return None
+
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
 
         _data = res.json()
 
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
+
         vote = Vote(id)
         vote._load(_data, url = f"{self.url}/votes/{id}", headers = self.default_headers)
 
         return vote
 
-    def open_vote(self, title: str = None, options: list[dict] = [], end: int = 0) -> Vote:
+    def open_vote(self, title: str = None, options: list[dict] = [], start: int = None, end: int = None) -> Vote:
         """
         Déclenche un vote dans la base de données.
 
@@ -57,29 +80,53 @@ class StateInterface(Interface):
             Titre du vote
         - options: list[dict]\n
             Liste des choix disponibles
-        - end: `int`\n
+        - start: `int` (optionnel)\n
+            Début du vote (timestamp)
+        - end: `int` (optionnel)\n
             Fin du vote (timestamp)
         """
 
         payload = {
             "options": options,
-            "end_date": end
         }
 
         if title:
             payload['title'] = title
 
-        res = requests.put(f"{self.url}/open_vote", headers = self.default_headers, json = payload)
+        res = requests.put(f"{self.url}/open_vote?a=b{('&start=' + str(start)) if start else ''}{('&end=' + str(end)) if end else ''}", headers = self.default_headers, json = payload)
+
+
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
+
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
 
-        if res.status_code == 200:
-            _data = res.json()
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
 
-            vote = Vote(_data['id'])
-            vote._load(_data, url = f"{self.url}/votes/{_data['id']}", headers = self.default_headers)
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
 
-            return vote
-        else:
-            res.raise_for_status()
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
+
+        vote = Vote(_data['id'])
+        vote._load(_data, url = f"{self.url}/votes/{_data['id']}", headers = self.default_headers)
+
+        return vote
 
     # Aucune possibilité de supprimer un vote
 
@@ -98,11 +145,34 @@ class StateInterface(Interface):
         id = NSID(id)
         res = requests.get(f"{self.url}/elections/{id}", headers = self.default_headers)
 
-        if res.status_code != 200:
-            res.raise_for_status()
+
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
 
         _data = res.json()
 
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
+
         election = Election(id)
         election._load(_data, url = f"{self.url}/elections/{id}", headers = self.default_headers)
 
@@ -123,15 +193,38 @@ class StateInterface(Interface):
 
         res = requests.put(f"{self.url}/open_election?vote={vote.id}&type={'full' if full else 'partial'}{('&date=' + str(start)) if start else ''}", headers = self.default_headers, json = {})
 
-        if res.status_code == 200:
-            _data = res.json()
 
-            election = Election(_data['id'])
-            election._load(_data, url = f"{self.url}/elections/{_data['id']}", headers = self.default_headers)
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
 
-            return election
-        else:
-            res.raise_for_status()
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
+
+        election = Election(_data['id'])
+        election._load(_data, url = f"{self.url}/elections/{_data['id']}", headers = self.default_headers)
+
+        return election
 
     """
     PARTIS
@@ -152,11 +245,34 @@ class StateInterface(Interface):
         id = NSID(id)
         res = requests.get(f"{self.url}/parties/{id}", headers = self.default_headers)
 
-        if res.status_code != 200:
-            res.raise_for_status()
+
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
 
         _data = res.json()
 
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
+
+        # TRAITEMENT
+
         party = Party(id)
         party._load(_data, url = f"{self.url}/parties/{id}", headers = self.default_headers)
 
@@ -173,24 +289,51 @@ class StateInterface(Interface):
             Couleur du parti
         - motto: `str, optional`\n
             Devise du parti
-        - politiscales: `.Scale`\n
+        - scale: `.Scale`\n
             Résultats du parti au test Politiscales
         """
 
         payload = {
             "color": color,
             "motto": motto,
-            "scale": scale if isinstance(scale, dict) else scale._to_dict()
+            "scale": {}
+        }
+
+        params = {
+            "candidate": id
         }
 
-        res = requests.put(f"{self.url}/register_party?candidate={id}", headers = self.default_headers, json = payload)
+        res = requests.put(f"{self.url}/register_party", headers = self.default_headers, params = params, json = payload)
+
+
+        # ERREURS
+
+        if 500 <= res.status_code < 600:
+            raise errors.globals.ServerDownError()
+
+        _data = res.json()
+
+        if res.status_code == 400:
+            if _data['message'] == "MissingParam":
+                raise errors.globals.MissingParamError(f"Missing parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidParam":
+                raise errors.globals.InvalidParamError(f"Invalid parameter '{_data['param']}'.")
+            elif _data['message'] == "InvalidToken":
+                raise errors.globals.AuthError("Token is not valid.")
+
+        elif res.status_code == 401:
+            raise errors.globals.AuthError(_data['message'])
+
+        elif res.status_code == 403:
+            raise errors.globals.PermissionError(_data['message'])
+
+        elif res.status_code == 404:
+            return
+
 
-        if res.status_code == 200:
-            _data = res.json()
+        # TRAITEMENT
 
-            party = Party(_data['org_id'])
-            party._load(_data, url = f"{self.url}/parties/{_data['org_id']}", headers = self.default_headers)
+        party = Party(_data['org_id'])
+        party._load(_data, url = f"{self.url}/parties/{_data['org_id']}", headers = self.default_headers)
 
-            return party
-        else:
-            res.raise_for_status()
+        return party

nsarchive/mandate.py

@@ -5,7 +5,7 @@ EPOCH = 1577833200 # 1er Janvier 2020
 PATIENTAGE_DATE = 1725141600 # 1er Septembre 2024
 OPEN_DATE = 1756677600 # 1er Septembre 2025
 
-MANDATE_DURATION = 2419200
+MANDATE_DURATION = 2419200 # 28 jours
 
 def get_cycle(ts: int = round(time.time())):
     if EPOCH <= ts < PATIENTAGE_DATE:
@@ -47,4 +47,23 @@ def get_phase(ts: int = round(time.time())) -> str:
         return 'elections' # Élections législatives et présidentielles
 
     else:
-        raise ValueError(f"Idk what happened but it seems that {day} is greater than 55...")
+        raise ValueError(f"Idk what happened but it seems that {day} is greater than 55...")
+
+
+def next_election(type: str = 'partial') -> int:
+    if get_cycle() == 1:
+        return PATIENTAGE_DATE
+    elif get_cycle() == 2:
+        return OPEN_DATE
+    else:
+        if type == 'partial':
+            ts = OPEN_DATE + get_cycle() * MANDATE_DURATION
+        elif type == 'full':
+            ts = OPEN_DATE + get_cycle() * MANDATE_DURATION
+
+            if get_cycle() % 2 == 1:
+                ts += 28 * 86400
+
+        ts = round(ts / 86400) * 86400
+
+        return ts