erioon 0.0.3__tar.gz → 0.0.5__tar.gz

{erioon-0.0.3 → erioon-0.0.5}/PKG-INFO

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

{erioon-0.0.3 → erioon-0.0.5}/erioon/client.py

@@ -35,7 +35,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,49 +55,39 @@ 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}
@@ -101,10 +95,13 @@ class ErioonClient:
 
         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 +169,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-0.0.3 → erioon-0.0.5}/erioon/collection.py

@@ -1,8 +1,5 @@
 import json
 import requests
-from colorama import Fore, Style, init
-
-init(autoreset=True)
 
 class Collection:
     def __init__(
@@ -11,6 +8,8 @@ class Collection:
         db_id,
         coll_id,
         metadata,
+        database,
+        cluster,
         base_url: str = "https://sdk.erioon.com",
     ):
         """
@@ -27,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(Fore.CYAN + "Erioon is loading..." + Style.RESET_ALL)
+        print("Erioon is loading...")
+        
+    def _is_read_only(self):
+        return self.database == "read"
+
 
     # ---------- READ ---------- #
     def get_all(self):
@@ -90,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:
@@ -110,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:
@@ -130,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:
@@ -149,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:
@@ -173,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-0.0.3 → erioon-0.0.5}/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.3 → erioon-0.0.5}/erioon.egg-info/PKG-INFO

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

{erioon-0.0.3 → erioon-0.0.5}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup
 
 setup(
     name='erioon',
-    version='0.0.3',
+    version='0.0.5',
     author='Zyber Pireci',
     author_email='zyber.pireci@erioon.com',
     description='Erioon SDF for Python',