mixpeek 0.7.3__py3-none-any.whl → 0.8.1__py3-none-any.whl

mixpeek/client.py

@@ -5,6 +5,7 @@ from .endpoints.collections import Collections
 from .endpoints.index import Index
 from .endpoints.search import Search
 from .endpoints.tools import Tools
+from .endpoints.register import Register
 
 
 class Mixpeek:
@@ -20,6 +21,7 @@ class Mixpeek:
         self.collections = Collections(self.base_url, self.headers)
         self.index = Index(self.base_url, self.headers)
         self.search = Search(self.base_url, self.headers)
+        self.register = Register(self.base_url, self.headers)
 
         # tools is all local
-        self.tools = Tools()
+        self.tools = Tools()

mixpeek/endpoints/collections.py

@@ -58,6 +58,15 @@ class Collections:
         except requests.RequestException as e:
             return {"error": str(e)}
 
+    def get_full_file(self, file_id):
+        try:
+            url = f"{self.base_url}collections/file/{file_id}/full"
+            response = requests.get(url, headers=self.headers)
+            response.raise_for_status()
+            return response.json()
+        except requests.RequestException as e:
+            return {"error": str(e)}
+
     def delete_file_by_id(self, file_id):
         try:
             url = f"{self.base_url}collections/file/{file_id}"

mixpeek/endpoints/index.py

@@ -1,4 +1,5 @@
 import requests
+from .tasks import Task
 
 class Index:
     def __init__(self, base_url, headers):
@@ -12,33 +13,37 @@ class Index:
             base_data["settings"] = settings
         return base_data
 
-    # def upload(self, file, collection_id, metadata=None, settings=None):
-    #     try:
-    #         api_url = f"{self.base_url}index/upload"
-    #         files = {"file": file}
-    #         data = self._prepare_data({"collection_id": collection_id}, metadata, settings)
-    #         response = requests.post(api_url, files=files, data=data, headers=self.headers)
-    #         response.raise_for_status()
-    #         return response.json()
-    #     except requests.RequestException as e:
-    #         return {"error": str(e)}
-
     def url(self, target_url, collection_id, metadata=None, settings=None):
         try:
             endpoint = f"{self.base_url}index/url"
             data = self._prepare_data({"url": target_url, "collection_id": collection_id}, metadata, settings)
+
             response = requests.post(endpoint, json=data, headers=self.headers)
             response.raise_for_status()
-            return response.json()
+            task_id = response.json().get("task_id")
+            if task_id:
+                return Task(self.base_url, self.headers, task_id)
+            else:
+                return response.json()
         except requests.RequestException as e:
             return {"error": str(e)}
-
-    def youtube(self, youtube_video_id, collection_id, metadata=None, settings=None):
+        
+    def upload(self, file_path, collection_id, metadata=None, settings=None):
         try:
-            api_url = f"{self.base_url}index/youtube"
-            data = self._prepare_data({"youtube_video_id": youtube_video_id, "collection_id": collection_id}, metadata, settings)
-            response = requests.post(api_url, json=data, headers=self.headers)
+            endpoint = f"{self.base_url}index/upload"
+            data = self._prepare_data({"collection_id": collection_id}, metadata, settings)
+            
+            with open(file_path, 'rb') as file:
+                files = [('file', (file.name, file, 'application/octet-stream'))]
+                response = requests.post(endpoint, headers=self.headers, data=data, files=files)
+            
             response.raise_for_status()
-            return response.json()
+            task_id = response.json().get("task_id")
+            if task_id:
+                return Task(self.base_url, self.headers, task_id)
+            else:
+                return response.json()
         except requests.RequestException as e:
-            return {"error": str(e)}
+            return {"error": str(e)}
+        except IOError as e:
+            return {"error": f"File error: {str(e)}"}

mixpeek/endpoints/register.py

@@ -0,0 +1,34 @@
+import requests
+from .tasks import Task
+
+class Register:
+    def __init__(self, base_url, headers):
+        self.base_url = base_url
+        self.headers = headers
+
+    def _prepare_data(self, base_data, metadata=None, settings=None):
+        if metadata is not None:
+            base_data["metadata"] = metadata
+        if settings is not None:
+            base_data["settings"] = settings
+        return base_data
+
+    def faces(self, file_path, collection_id, metadata=None, settings=None):
+        try:
+            endpoint = f"{self.base_url}register/faces"
+            data = self._prepare_data({"collection_id": collection_id}, metadata, settings)
+            
+            with open(file_path, 'rb') as file:
+                files = [('file', (file.name, file, 'application/octet-stream'))]
+                response = requests.post(endpoint, headers=self.headers, data=data, files=files)
+            
+            response.raise_for_status()
+            task_id = response.json().get("task_id")
+            if task_id:
+                return Task(self.base_url, self.headers, task_id)
+            else:
+                return response.json()
+        except requests.RequestException as e:
+            return {"error": str(e)}
+        except IOError as e:
+            return {"error": f"File error: {str(e)}"}

mixpeek/endpoints/tasks.py

@@ -0,0 +1,26 @@
+import requests
+import time
+
+class Task:
+    def __init__(self, base_url, headers, task_id):
+        self.base_url = base_url
+        self.headers = headers
+        self.task_id = task_id
+
+    def get_task_status(self, task_id):
+        try:
+            endpoint = f"{self.base_url}tasks/{task_id}"
+            response = requests.get(endpoint, headers=self.headers)
+            response.raise_for_status()
+            return response.json()
+        except requests.RequestException as e:
+            return {"error": str(e)}
+
+    def wait_for_done(self, sleep_interval=2, callback=None):
+        while True:
+            status = self.get_task_status(self.task_id)
+            if callback:
+                callback(status)
+            if status.get("status") == "DONE":
+                return status
+            time.sleep(sleep_interval)

{mixpeek-0.7.3.dist-info → mixpeek-0.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mixpeek
-Version: 0.7.3
+Version: 0.8.1
 Summary: Mixpeek Python SDK
 Home-page: https://github.com/mixpeek/mixpeek-python
 Author: Ethan Steininger

mixpeek-0.8.1.dist-info/RECORD

@@ -0,0 +1,15 @@
+mixpeek/__init__.py,sha256=XDdcK7wTEOEcF1cp-GeWmgPJ21Ny1R9pB0PPNrdDTMo,28
+mixpeek/client.py,sha256=FcSjc73WMMKPHzZKye2_XicwJP1bY_iaMWdIv-ft1Rw,906
+mixpeek/exceptions.py,sha256=Orhdo5UFLn3fcWVJtlgkznW8Iy5ndL96h0qTY8zOlDA,235
+mixpeek/endpoints/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mixpeek/endpoints/collections.py,sha256=-mxmI4Bs60OH6HRPzE03rr0XVnzqUxL_iHgF-_4R6Qc,3232
+mixpeek/endpoints/embed.py,sha256=ybDqyIt0oJiwIdH1QSwuV-CPeJgd-zW0etNAcBdgZYE,2290
+mixpeek/endpoints/index.py,sha256=kG5bf0ycIWvooicA5xowXrWH4x6nbzP6XdyK-XQHqU0,1976
+mixpeek/endpoints/register.py,sha256=ZVes4BPxQsPRNk70tGeVe_CYr4eVQKL-CTVmzpJAHM4,1314
+mixpeek/endpoints/search.py,sha256=u3onbss7Llx0AgcTSRtel6bGO5Ly3jvraXPcA26GzsY,2387
+mixpeek/endpoints/tasks.py,sha256=GBkscCYVv46JkJ4MVOtM6tL__rymnM3bmt4PrdRK04Y,839
+mixpeek/endpoints/tools.py,sha256=T8qb9zngY3V_vwtmBBrTqPvqZl1QIL8q3at3UGIj4bg,4709
+mixpeek-0.8.1.dist-info/METADATA,sha256=aCVFiT_BXaKmv-DydThZYI6HJhgpcaeg1NXnFwN-9vk,10616
+mixpeek-0.8.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+mixpeek-0.8.1.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
+mixpeek-0.8.1.dist-info/RECORD,,

{mixpeek-0.7.3.dist-info → mixpeek-0.8.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (73.0.1)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

mixpeek-0.7.3.dist-info/RECORD

@@ -1,13 +0,0 @@
-mixpeek/__init__.py,sha256=XDdcK7wTEOEcF1cp-GeWmgPJ21Ny1R9pB0PPNrdDTMo,28
-mixpeek/client.py,sha256=JFQO-hLFe7MESdkNoweuoaGrc4Vnqtl55_JRvgDCspg,804
-mixpeek/exceptions.py,sha256=Orhdo5UFLn3fcWVJtlgkznW8Iy5ndL96h0qTY8zOlDA,235
-mixpeek/endpoints/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mixpeek/endpoints/collections.py,sha256=ffOBenNHG-mVpjT7kQcOfn78-wjgjdQTVyrloyeAy90,2890
-mixpeek/endpoints/embed.py,sha256=ybDqyIt0oJiwIdH1QSwuV-CPeJgd-zW0etNAcBdgZYE,2290
-mixpeek/endpoints/index.py,sha256=QgMgys1CCukepCpK351aVXRxEsxBfI2YriX7aaH1NrU,1939
-mixpeek/endpoints/search.py,sha256=u3onbss7Llx0AgcTSRtel6bGO5Ly3jvraXPcA26GzsY,2387
-mixpeek/endpoints/tools.py,sha256=T8qb9zngY3V_vwtmBBrTqPvqZl1QIL8q3at3UGIj4bg,4709
-mixpeek-0.7.3.dist-info/METADATA,sha256=AO5vsb1itiebws3tEFrXnHuywAZei3TBgRXVW3Lfgy4,10616
-mixpeek-0.7.3.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
-mixpeek-0.7.3.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
-mixpeek-0.7.3.dist-info/RECORD,,