nsarchive 3.0.0a8__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,10 +66,209 @@ 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
+    """
+
+    def get_lawsuit(self, id: NSID) -> Lawsuit:
+        res = requests.get(
+            f"{self.url}/justice/lawsuits/{id}",
+            headers = 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(id)
+        lawsuit._load(_data, f"{self.url}/justice/lawsuits/{id}", self.default_headers)
+
+        return lawsuit
+
+    def open_lawsuit(self, target: NSID, title: str = None, report: Report = None) -> Lawsuit:
+        payload = {}
+        if title: payload['title'] = title
+
+        res = requests.put(
+            f"{self.url}/justice/open_lawsuit?target={target}{('&report=' + report.id) if report 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.")
+
+        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
+    """
+
+    def get_sanction(self, id: NSID) -> Sanction:
+        res = requests.get(
+            f"{self.url}/justice/sanctions/{id}",
+            headers = 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(id)
+        sanction._load(_data, f"{self.url}/justice/sanctions/{id}", self.default_headers)
+
+        return sanction
+
+    def add_sanction(self, target: NSID, _type: str, duration: int = None, title: str = None, lawsuit: Lawsuit = None) -> Sanction:
+        payload = {}
+        if title: payload['title'] = title
+
+        res = requests.put(
+            f"{self.url}/justice/add_sanction?type={_type}&target={target}&date={str(round(time.time()))}{('&duration=' + str(duration)) if duration else ''}{('&case=' + lawsuit.id) if lawsuit 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.")
+
+        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 report
+        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()
 
-        vote = Vote(id, _data['title'])
+        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,30 +80,152 @@ 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 = _data)
+        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)
+
 
-        if res.status_code == 200:
-            _data = res.json()
+        # ERREURS
 
-            vote = Vote()
-            vote._load(_data, url = f"{self.url}/votes/{_data['id']}", headers = self.default_headers)
-        else:
-            res.raise_for_status()
+        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(_data['id'])
+        vote._load(_data, url = f"{self.url}/votes/{_data['id']}", headers = self.default_headers)
+
+        return vote
 
     # Aucune possibilité de supprimer un vote
 
+    def get_election(self, id: NSID) -> Election:
+        """
+        Récupère une élection.
+
+        ## Paramètres
+        id: `NSID`\n
+            ID de l'élection.
+
+        ## Renvoie
+        - `.Election`
+        """
+
+        id = NSID(id)
+        res = requests.get(f"{self.url}/elections/{id}", headers = 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
+
+        election = Election(id)
+        election._load(_data, url = f"{self.url}/elections/{id}", headers = self.default_headers)
+
+        return election
+
+    def open_election(self, vote: Vote, start: int = None, full: bool = False) -> Election:
+        """
+        Déclenche une élection dans la base de données.
+
+        ## Paramètres
+        - vote: `.Vote`\n
+            Vote associé
+        - start: `int` (optionnel)\n
+            Date de début du vote (timestamp, dure 4 jours)
+        - full: `bool` (optionnel)\n
+            Choix du type d'élections (True = présidentielles, False = législatives)
+        """
+
+        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 = {})
+
+
+        # 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(_data['id'])
+        election._load(_data, url = f"{self.url}/elections/{_data['id']}", headers = self.default_headers)
+
+        return election
+
     """
     PARTIS
     """
@@ -100,11 +245,34 @@ class StateInterface(Interface):
         id = NSID(id)
         res = requests.get(f"{self.url}/parties/{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
+
         party = Party(id)
         party._load(_data, url = f"{self.url}/parties/{id}", headers = self.default_headers)
 
@@ -121,22 +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
+            "scale": {}
         }
 
-        res = requests.put(f"{self.url}/register_party?candidate={id}", headers = self.default_headers, json = _data)
+        params = {
+            "candidate": id
+        }
+
+        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
+
+
+        # TRAITEMENT
 
-        if res.status_code == 200:
-            _data = res.json()
+        party = Party(_data['org_id'])
+        party._load(_data, url = f"{self.url}/parties/{_data['org_id']}", headers = self.default_headers)
 
-            vote = Vote()
-            vote._load(_data, url = f"{self.url}/votes/{_data['id']}", headers = self.default_headers)
-        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