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

nsarchive/mandate.py

@@ -0,0 +1,50 @@
+import math
+import time
+
+EPOCH = 1577833200 # 1er Janvier 2020
+PATIENTAGE_DATE = 1725141600 # 1er Septembre 2024
+OPEN_DATE = 1756677600 # 1er Septembre 2025
+
+MANDATE_DURATION = 2419200
+
+def get_cycle(ts: int = round(time.time())):
+    if EPOCH <= ts < PATIENTAGE_DATE:
+        return 0
+    elif PATIENTAGE_DATE <= ts < OPEN_DATE:
+        return 1
+    elif OPEN_DATE <= ts:
+        return math.floor((ts - OPEN_DATE) / MANDATE_DURATION) + 2
+    else:
+        raise ValueError(f"Timestamp {ts} is out of range (must be greater than or equal to {EPOCH}).")
+
+def get_day(ts: int = round(time.time())) -> int:
+    cycle = get_cycle(ts)
+
+    if cycle == 0:
+        return math.floor((ts - EPOCH) / 86400)
+    elif cycle == 1:
+        return math.floor((ts - PATIENTAGE_DATE) / 86400)
+    else:
+        return math.floor(((ts - OPEN_DATE) % MANDATE_DURATION) / 86400)
+
+def get_phase(ts: int = round(time.time())) -> str:
+    cycle = get_cycle(ts)
+
+    if cycle < 2: # Les deux premiers cycles durent (beaucoup) plus longtemps qu'un cycle normal.
+        return 'undefined'
+
+    day = get_day(ts)
+
+    if day == 0:
+        return 'investiture' # Investiture du PR
+    elif 1 <= day < 24:
+        return 'paix' # Rien du tout
+    elif 24 <= day < 28:
+        return 'partials' # Élections législatives
+    elif 28 <= day < 52:
+        return 'paix' # Toujours rien
+    elif 52 <= day < 56:
+        return 'elections' # Élections législatives et présidentielles
+
+    else:
+        raise ValueError(f"Idk what happened but it seems that {day} is greater than 55...")

nsarchive/{cls → models}/base.py

@@ -1,4 +1,3 @@
-import io
 import json
 import requests
 import typing
@@ -40,23 +39,28 @@ class NSID(str):
         if value.startswith("0x"):
             value = value[2:]
 
-        instance = super(NSID, cls).__new__(cls, value.upper())
-        return instance
+        interface = super(NSID, cls).__new__(cls, value.upper())
+        return interface
 
-class Instance:
+class Interface:
     """
-    Instance qui servira de base à toutes les instances.
+    Instance qui servira de base à toutes les interfaces.
     """
 
     def __init__(self, url: str, token: str = None):
         self.url = url
         self.token = token
+        self.zone = 20 # 10 = Serveur test, 20 = Serveur principal, 30 = Serveur de patientage, 40 = Scratch World
 
         self.default_headers = {
             "Authorization": f"Bearer {self.token}",
             "Content-Type": "application/json",
         }
 
+        return
+
+        # Vérification (ralentit considérablement les requêtes)*
+
         try:
             test_res = requests.get(f'{self.url}/ping')
 
@@ -71,6 +75,24 @@ class Instance:
         except:
             utils.warn("NationDB is not responding.")
 
+    def alias(self, alias: NSID) -> typing.Self:
+        """
+        Duplique l'interface en se faisant passer pour une autre entité. Aucune erreur ne sera levée si l'entité n'existe pas (hormis les éventuels 401 ou 404 renvoyés par le serveur).
+
+        ## Paramètres
+        alias: `NSID`\n
+            ID de l'entité à simuler
+
+        ## Renvoie
+        - `self` avec le token de l'alias
+        """
+
+        alias = NSID(alias)
+
+        token = self.token + ':' + str(alias)
+
+        return self.__class__(self.url, token)
+
     def request_token(self, username: str, password: str) -> str | None:
         res = requests.post(f"{self.url}/auth/login", json = {
             "username": username,
@@ -120,7 +142,7 @@ class Instance:
 
         return _data
 
-    def _put_in_db(self, endpoint: str, body: dict, headers: dict = None, use_PUT: bool = False) -> None:
+    def _put_in_db(self, endpoint: str, body: dict = {}, headers: dict = None, use_PUT: bool = False) -> None:
         """
         Publie des données JSON dans une table nation-db.
 
@@ -144,7 +166,6 @@ class Instance:
         if 200 <= res.status_code < 300:
             return res.json()
         else:
-            print(res.text)
             res.raise_for_status()
 
     def _delete(self, _class: str, ids: list[NSID]) -> None:

nsarchive/{cls → models}/economy.py

@@ -4,7 +4,6 @@ import urllib
 
 from .base import NSID
 
-default_headers = {}
 
 class BankAccount:
     """
@@ -27,6 +26,7 @@ class BankAccount:
 
     def __init__(self, owner_id: NSID) -> None:
         self._url: str = ""
+        self._headers: dict = {}
 
         self.id: NSID = NSID(owner_id)
         self.owner_id: NSID = NSID(owner_id)
@@ -40,7 +40,12 @@ class BankAccount:
         self.frozen: bool = False
         self.flagged: bool = False
 
-    def _load(self, _data: dict):
+    def _load(self, _data: dict, url: str, headers: dict) -> None:
+        self._url = url + '/bank/accounts/' + _data['id']
+        self._headers = headers
+
+        self.id = NSID(_data['id'])
+
         self.owner_id = NSID(_data['owner_id'])
         self.register_date = _data['register_date']
         self.tag = _data['tag']
@@ -53,18 +58,17 @@ class BankAccount:
         self.flagged = _data['flagged']
 
     def freeze(self, frozen: bool = True, reason: str = None) -> None:
-        res = requests.post(f"{self._url}/freeze?frozen={str(frozen).lower()}", headers = default_headers, json = {
+        res = requests.post(f"{self._url}/freeze?frozen={str(frozen).lower()}", headers = self._headers, json = {
             "reason": reason
         })
 
         if res.status_code == 200:
             self.frozen = frozen
         else:
-            print(res.text)
             res.raise_for_status()
 
     def flag(self, flagged: bool = True, reason: str = None) -> None:
-        res = requests.post(f"{self._url}/flag?flagged={str(flagged).lower()}", headers = default_headers, json = {
+        res = requests.post(f"{self._url}/flag?flagged={str(flagged).lower()}", headers = self._headers, json = {
             "reason": reason
         })
 
@@ -77,7 +81,7 @@ class BankAccount:
         _target_query = f"&target={target}"
         _loan_query = f"&loan_id={loan}"
 
-        res = requests.post(f"{self._url}/debit?amount={amount}{_target_query if target else ''}{_loan_query if loan else ''}", headers = default_headers, json = {
+        res = requests.post(f"{self._url}/debit?amount={amount}{_target_query if target else ''}{_loan_query if loan else ''}", headers = self._headers, json = {
             "reason": reason,
             "digicode": digicode
         })
@@ -88,7 +92,7 @@ class BankAccount:
             res.raise_for_status()
 
     def deposit(self, amount: int, reason: str = None) -> None:
-        res = requests.post(f"{self._url}/deposit?amount={amount}", headers = default_headers, json = {
+        res = requests.post(f"{self._url}/deposit?amount={amount}", headers = self._headers, json = {
             "reason": reason,
         })
 
@@ -111,20 +115,28 @@ class Item:
     """
 
     def __init__(self) -> None:
+        self._url: str = ""
+        self._headers: dict = {}
+
         self.id: NSID = NSID(round(time.time()))
         self.name: str = "Unknown Object"
         self.emoji: str = ":light_bulb:"
         self.category: str = "common"
         self.craft: dict = {}
 
-    def _load(self, _data: dict):
+    def _load(self, _data: dict, url: str, headers: dict) -> None:
+        self._url = url + '/marketplace/items/' + _data['id']
+        self._headers = headers
+
+        self.id = NSID(_data['id'])
+
         self.name = _data['name']
         self.emoji = _data['emoji']
         self.category = _data['category']
         self.craft = _data['craft']
 
     def rename(self, new_name: str):
-        res = requests.post(f"{self._url}/rename?name={new_name}", headers = default_headers)
+        res = requests.post(f"{self._url}/rename?name={new_name}", headers = self._headers)
 
         if res.status_code == 200:
             self.name = new_name
@@ -150,6 +162,7 @@ class Sale:
 
     def __init__(self, item: Item) -> None:
         self._url: str = ""
+        self._headers: dict = {}
 
         self.id: NSID = NSID(round(time.time()))
         self.open: bool = True
@@ -159,8 +172,11 @@ class Sale:
         self.quantity: int = 1
         self.price: int = 0
 
-    def _load(self, _data: dict):
-        self.id = _data['json']
+    def _load(self, _data: dict, url: str, headers: dict) -> None:
+        self._url = url + '/marketplace/sales/' + _data['id']
+        self._headers = headers
+
+        self.id = _data['id']
         self.open = _data['open']
         self.seller_id = NSID(_data['seller_id'])
 
@@ -187,6 +203,7 @@ class Inventory:
 
     def __init__(self, owner_id: NSID) -> None:
         self._url: str = ""
+        self._headers: dict = {}
 
         self.id: NSID = NSID(owner_id)
         self.owner_id: NSID = NSID(owner_id)
@@ -196,7 +213,10 @@ class Inventory:
 
         self.items: dict[NSID, int] = {}
 
-    def _load(self, _data: dict):
+    def _load(self, _data: dict, url: str, headers: dict):
+        self._url = url + '/bank/inventories/' + _data['id']
+        self._headers = headers
+
         self.id = NSID(_data['id'])
         self.owner_id = NSID(_data['owner_id'])
 
@@ -206,7 +226,7 @@ class Inventory:
         self.items = _data['items']
 
     def deposit_item(self, item: Item, giver: NSID = None, quantity: int = 1, digicode: str = None):
-        res = requests.post(f"{self._url}/deposit?item={item.id}&amount={quantity}", headers = default_headers, json = {
+        res = requests.post(f"{self._url}/deposit?item={item.id}&amount={quantity}", headers = self._headers, json = {
             "giver": giver,
             "digicode": digicode
         })
@@ -220,7 +240,7 @@ class Inventory:
             res.raise_for_status()
 
     def sell_item(self, item: Item, price: int, quantity: int = 1, digicode: str = None) -> NSID:
-        res = requests.post(f"{self._url}/sell_item?item={item.id}&quantity={quantity}&price={price}", headers = default_headers, json = {
+        res = requests.post(f"{self._url}/sell_item?item={item.id}&quantity={quantity}&price={price}", headers = self._headers, json = {
             "digicode": digicode
         })