erioon 0.0.4__py3-none-any.whl → 0.0.6__py3-none-any.whl

erioon/auth.py

@@ -16,5 +16,5 @@ def Auth(credential_string):
     >>> client = Auth("<EMAIL>:<PASSWORD>")
     >>> print(client)  # prints user_id if successful or error message if not
     """
-    email, password = credential_string.split(":")
-    return ErioonClient(email, password)
+    api, email, password = credential_string.split(":")
+    return ErioonClient(api ,email, password)

erioon/client.py

@@ -1,6 +1,7 @@
 import os
 import json
 import requests
+from werkzeug.security import generate_password_hash
 from erioon.database import Database
 
 class ErioonClient:
@@ -18,7 +19,7 @@ class ErioonClient:
         token_path (str): Local path to cached authentication token.
     """
 
-    def __init__(self, email, password, base_url="https://sdk.erioon.com"):
+    def __init__(self, api, email, password, base_url="https://sdk.erioon.com"):
         """
         Initialize ErioonClient instance, attempts to load cached token or perform login.
 
@@ -27,6 +28,7 @@ class ErioonClient:
             password (str): User password for authentication.
             base_url (str, optional): Base API URL. Defaults to "https://sdk.erioon.com".
         """
+        self.api = api
         self.email = email
         self.password = password
         self.base_url = base_url
@@ -35,7 +37,11 @@ class ErioonClient:
         self.token_path = os.path.expanduser(f"~/.erioon_token_{self._safe_filename(email)}")
 
         try:
-            self.user_id = self._load_or_login()
+            metadata = self._load_or_login()
+            self.user_id = metadata.get("_id")
+            self.database = metadata.get("database")
+            self.cluster = metadata.get("cluster")
+            self.login_metadata = metadata
         except Exception as e:
             self.error = str(e)
 
@@ -51,60 +57,54 @@ class ErioonClient:
         """
         return "".join(c if c.isalnum() else "_" for c in text)
 
-    def _load_or_login(self):
+    def _do_login_and_cache(self):
         """
-        Load cached user_id token from local storage or perform login if not cached.
+        Perform login to API and cache the metadata locally.
 
         Returns:
-            str: User ID from token or fresh login.
-
-        Raises:
-            Exception: If login fails.
+            dict: Login metadata including user_id, database, cluster.
         """
-        if os.path.exists(self.token_path):
-            with open(self.token_path, "r") as f:
-                token_data = json.load(f)
-                user_id = token_data.get("user_id")
-                if user_id:
-                    return user_id
-
-        return self._do_login_and_cache()
+        metadata = self._login()
+        with open(self.token_path, "w") as f:
+            json.dump(metadata, f)
+        return metadata
 
-    def _do_login_and_cache(self):
+    def _load_or_login(self):
         """
-        Perform login to API and cache the user_id token locally.
+        Load cached metadata or perform login.
 
         Returns:
-            str: User ID from successful login.
-
-        Raises:
-            Exception: If login fails.
+            dict: Login metadata.
         """
-        user_id = self._login()
-        with open(self.token_path, "w") as f:
-            json.dump({"user_id": user_id}, f)
-        return user_id
+        if os.path.exists(self.token_path):
+            with open(self.token_path, "r") as f:
+                metadata = json.load(f)
+                if "user_id" in metadata:
+                    return metadata
+
+        return self._do_login_and_cache()
 
     def _login(self):
         """
-        Authenticate with Erioon API using email and password.
+        Authenticate and return full login metadata.
 
         Returns:
-            str: User ID on successful authentication.
-
-        Raises:
-            Exception: If credentials are invalid.
+            dict: Metadata with user_id, database, cluster, etc.
         """
         url = f"{self.base_url}/login_with_credentials"
-        payload = {"email": self.email, "password": self.password}
+        hashed_key = generate_password_hash(self.api)
+        payload = {"api_key": hashed_key,"email": self.email, "password": self.password}
         headers = {"Content-Type": "application/json"}
 
         response = requests.post(url, json=payload, headers=headers)
         if response.status_code == 200:
-            return response.text.strip()
+            data = response.json()
+            self.login_metadata = data
+            return data
         else:
             raise Exception("Invalid account")
 
+
     def _clear_cached_token(self):
         """
         Remove cached token file and reset user_id to None.
@@ -172,7 +172,12 @@ class ErioonClient:
 
         if response.status_code == 200:
             db_info = response.json()
-            return Database(self.user_id, db_info)
+            return Database(
+                user_id=self.user_id,
+                metadata=db_info,
+                database=self.database,
+                cluster=self.cluster
+            )
         else:
             try:
                 error_json = response.json()

erioon/collection.py

@@ -8,6 +8,8 @@ class Collection:
         db_id,
         coll_id,
         metadata,
+        database,
+        cluster,
         base_url: str = "https://sdk.erioon.com",
     ):
         """
@@ -24,11 +26,17 @@ class Collection:
         self.db_id = db_id
         self.coll_id = coll_id
         self.metadata = metadata
+        self.database = database 
+        self.cluster = cluster
         self.base_url = base_url.rstrip("/")
 
     def _print_loading(self) -> None:
         """Print a green loading message to the terminal."""
         print("Erioon is loading...")
+        
+    def _is_read_only(self):
+        return self.database == "read"
+
 
     # ---------- READ ---------- #
     def get_all(self):
@@ -87,6 +95,9 @@ class Collection:
             result = collection.insert_one(new_doc)
         """
         self._print_loading()
+        if self._is_read_only():
+            return {"status": "KO", "error": "Method not allowed. Access is only read."}
+        
         url = f"{self.base_url}/{self.user_id}/{self.db_id}/{self.coll_id}/insert_one"
         response = requests.post(url, json=document, headers={"Content-Type": "application/json"})
         try:
@@ -107,6 +118,9 @@ class Collection:
             result = collection.insert_many(docs)
         """
         self._print_loading()
+        if self._is_read_only():
+            return {"status": "KO", "error": "Method not allowed. Access is only read."}
+        
         url = f"{self.base_url}/{self.user_id}/{self.db_id}/{self.coll_id}/insert_many"
         response = requests.post(url, json={"records": documents})
         try:
@@ -127,6 +141,9 @@ class Collection:
             result = collection.delete_one({"name": "John"})
         """
         self._print_loading()
+        if self._is_read_only():
+            return {"status": "KO", "error": "Method not allowed. Access is only read."}
+        
         url = f"{self.base_url}/{self.user_id}/{self.db_id}/{self.coll_id}/delete_one"
         response = requests.delete(url, json=filter_query)
         try:
@@ -146,6 +163,9 @@ class Collection:
             result = collection.delete_many({"status": "inactive"})
         """
         self._print_loading()
+        if self._is_read_only():
+            return {"status": "KO", "error": "Method not allowed. Access is only read."}
+        
         url = f"{self.base_url}/{self.user_id}/{self.db_id}/{self.coll_id}/delete_many"
         response = requests.delete(url, json=filter_query)
         try:
@@ -170,6 +190,9 @@ class Collection:
             )
         """
         self._print_loading()
+        if self._is_read_only():
+            return {"status": "KO", "error": "Method not allowed. Access is only read."}
+        
         url = f"{self.base_url}/{self.user_id}/{self.db_id}/{self.coll_id}/update_query"
         response = requests.patch(url, json={"filter_query": filter_query, "update_query": update_query})
         try:

erioon/database.py

@@ -2,10 +2,12 @@ import json
 from erioon.collection import Collection
 
 class Database:
-    def __init__(self, user_id, metadata):
+    def __init__(self, user_id, metadata, database=None, cluster=None):
         self.user_id = user_id
         self.metadata = metadata
         self.db_id = metadata.get("database_info", {}).get("_id")
+        self.database = database
+        self.cluster = cluster
 
     def __getitem__(self, collection_id):
         try:
@@ -19,7 +21,9 @@ class Database:
                 user_id=self.user_id,
                 db_id=self.db_id,
                 coll_id=collection_id,
-                metadata=coll_meta
+                metadata=coll_meta,
+                database = self.database, 
+                cluster = self.cluster
             )
         except Exception:
             return "Connection error"
@@ -28,4 +32,5 @@ class Database:
         return json.dumps(self.metadata, indent=4)
 
     def __repr__(self):
-        return f"<Database db_id={self.db_id}>"
+        return f"<Database db_id={self.db_id}, cluster={self.cluster}, database={self.database}>"
+

{erioon-0.0.4.dist-info → erioon-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: erioon
-Version: 0.0.4
+Version: 0.0.6
 Summary: Erioon SDF for Python
 Author: Zyber Pireci
 Author-email: zyber.pireci@erioon.com

erioon-0.0.6.dist-info/RECORD

@@ -0,0 +1,9 @@
+erioon/auth.py,sha256=zSMRnkoU2-5khZNYhcWb48PtYTbtYYbAI--ydrFeMbo,714
+erioon/client.py,sha256=xFdzTJgZNo3RU7FStEqXVCKI8EjmOxlQG5HXjKm6nMU,6816
+erioon/collection.py,sha256=i0mmOj79D02suXhTtfJA8yH-oM7JiHyehh574JHlJn0,7686
+erioon/database.py,sha256=IjtZYJtQ-8shojxYwNKnN1ZaRwwWZfOhw3PDAgapE8w,1185
+erioon-0.0.6.dist-info/LICENSE,sha256=xwnq3DNlZpQyteOK9HvtHRhMdYviXTTaCDljEodFRnQ,569
+erioon-0.0.6.dist-info/METADATA,sha256=sJILmB27Ztfcpzt7cj0RFF1e6QBe9WrLpCGD6V58Sc4,715
+erioon-0.0.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+erioon-0.0.6.dist-info/top_level.txt,sha256=yjKEg85X5Q5ot46IMML_xukvIGG5YfdrLWcemjalItc,7
+erioon-0.0.6.dist-info/RECORD,,

erioon-0.0.4.dist-info/RECORD

@@ -1,9 +0,0 @@
-erioon/auth.py,sha256=-a8g7JZKsQ_yir_iZMYbfhiBJT0hYDmfJSj6B1u3M84,704
-erioon/client.py,sha256=5VKxfHUdChwFRA1bRWCt6p10y1SF7XR1PWp6wt7qQgg,6532
-erioon/collection.py,sha256=f5Uw7-Y7-Il8uocUBVshNLTots2Ja3GQEO887vmImr4,6830
-erioon/database.py,sha256=n3arbUCYzJ1WAIh_1hgjJjUuCpyRVRP1quXGkTVgrsQ,953
-erioon-0.0.4.dist-info/LICENSE,sha256=xwnq3DNlZpQyteOK9HvtHRhMdYviXTTaCDljEodFRnQ,569
-erioon-0.0.4.dist-info/METADATA,sha256=mp2Pb68frefC8uz7pceZnKW7_NBaW5FmiGZdkS-Wffo,715
-erioon-0.0.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-erioon-0.0.4.dist-info/top_level.txt,sha256=yjKEg85X5Q5ot46IMML_xukvIGG5YfdrLWcemjalItc,7
-erioon-0.0.4.dist-info/RECORD,,