dataroom-client 0.0.0.post1.dev0__py3-none-any.whl

dataroom_client/counter.py

@@ -0,0 +1,45 @@
+import datetime
+import zoneinfo
+
+from .print_utils import print_success
+
+
+class Counter:
+    def __init__(self, total, use_print=True, print_every=10):
+        self.total = total
+        self.count = 0
+        self.use_print = use_print
+        self.print_every = print_every
+        self.start_time = None
+        self.end_time = None
+
+    @property
+    def time_left(self):
+        elapsed = datetime.datetime.now(tz=zoneinfo.ZoneInfo('UTC')) - self.start_time
+        seconds_left = (self.total - self.count) * elapsed.total_seconds() / self.count
+        return datetime.timedelta(seconds=seconds_left)
+
+    @property
+    def time_took(self):
+        return self.end_time - self.start_time
+
+    @property
+    def percent(self):
+        return round(self.count / self.total * 100, 1)
+
+    def increment(self):
+        self.count += 1
+        if self.use_print and self.print_every and self.count % self.print_every == 0:
+            print_success(
+                f'Processed {self.percent}% - {self.count} out of {self.total}. Time left: {self.time_left}',
+            )
+
+    def start(self):
+        self.start_time = datetime.datetime.now(tz=zoneinfo.ZoneInfo('UTC'))
+
+    def end(self):
+        self.end_time = datetime.datetime.now(tz=zoneinfo.ZoneInfo('UTC'))
+        if self.use_print:
+            print_success(
+                f'Done! Took: {self.time_took}',
+            )

dataroom_client/loader.py

@@ -0,0 +1,61 @@
+import asyncio
+import os
+
+from .client import DataRoomFile, DataRoomClient, DataRoomError
+from .counter import Counter
+from .print_utils import print_error, print_status
+
+
+class DataRoomLoader:
+
+    def __init__(
+        self,
+        client: DataRoomClient,
+        folder_path: str,
+        concurrency: int = 2,
+        verbose=True,
+        image_source: str = 'loader',
+        image_id_prefix: str = None,
+    ):
+        self.client = client
+        self.concurrency = concurrency
+        self.folder_path = folder_path
+        self.counter = Counter(0)
+        self.verbose = verbose
+        self.image_source = image_source
+        self.image_id_prefix = image_id_prefix
+
+    async def run(self):
+        images = os.listdir(self.folder_path)
+
+        self.counter = Counter(len(images))
+        self.counter.start()
+
+        semaphore = asyncio.Semaphore(self.concurrency)
+        tasks = []
+        for image_path in images:
+            tasks.append(self._process_image(image_path=image_path, semaphore=semaphore))
+
+        await asyncio.gather(*tasks)
+
+        self.counter.end()
+
+    async def _process_image(self, image_path: str, semaphore: asyncio.Semaphore):
+        async with semaphore:
+            image_id = image_path.split('.')[0]
+            if self.image_id_prefix:
+                image_id = f"{self.image_id_prefix}{image_id}"
+            full_path = os.path.join(self.folder_path, image_path)
+            if self.verbose:
+                print_status(f"Processing image: {full_path}")
+            try:
+                image_file = DataRoomFile.from_path(full_path)
+            except DataRoomError as e:
+                print_error(f"Failed to read image: {image_path} - Error: {e}")
+            else:
+                try:
+                    await self.client.create_image(image_id=image_id, image_file=image_file, source=self.image_source)
+                except DataRoomError as e:
+                    print_error(f"Failed to create image: {image_path} - Error: {e}")
+            finally:
+                self.counter.increment()

dataroom_client/print_utils.py

@@ -0,0 +1,11 @@
+
+def print_error(string):
+    print(f"\033[31m{string}\033[0m")  # noqa
+
+
+def print_success(string):
+    print(f"\033[32m{string}\033[0m")  # noqa
+
+
+def print_status(string):
+    print(string)  # noqa

dataroom_client-0.0.0.post1.dev0.dist-info/METADATA

@@ -0,0 +1,41 @@
+Metadata-Version: 2.3
+Name: dataroom-client
+Version: 0.0.0.post1.dev0
+Summary: A python client to interface with the Dataroom backend API
+Author: Ales Kocjancic
+Author-email: hi@ales.io
+Requires-Python: >=3.11,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: httpx (>=0.24.1,<1)
+Description-Content-Type: text/markdown
+
+
+# Installation
+
+```
+pip install dataroom-client
+```
+
+
+# Usage
+
+After getting an account you can find your API key on the settings page.
+
+```
+from dataroom_client import DataRoomClient
+
+DataRoom = DataRoomClient(api_key='YOUR_SECRET_API_KEY_HERE', api_url='YOUR_API_URL_HERE')
+
+images = await DataRoom.get_images()
+```
+
+For more examples see [client_example.ipynb](./notebooks/client_example.ipynb).
+
+
+# Developing
+
+Check out the `dataroom` repo and follow the instructions in the README.
+

dataroom_client-0.0.0.post1.dev0.dist-info/RECORD

@@ -0,0 +1,8 @@
+dataroom_client/__init__.py,sha256=lDoFtI0GopBlcVyfsJxnSpScN3bqTr4zeSyTmAhfuCI,135
+dataroom_client/client.py,sha256=6uK6HBinqXqJvZM3u7aX_E96Wv00hzaYaIgllxmsqps,58960
+dataroom_client/counter.py,sha256=tQqLTtE2NHAguv0sPK-7RufZ9p2ZhF8GTRpdf4LEbcc,1386
+dataroom_client/loader.py,sha256=hLNmmI-AcKnsgZmpH42Db5Nrj1Uokped7M2Uqz69Alo,2060
+dataroom_client/print_utils.py,sha256=Aak26E5Bp9TAv1v8rRcPe-qcFObGyF_VlZI4GlxAmj8,201
+dataroom_client-0.0.0.post1.dev0.dist-info/METADATA,sha256=ivnuxY5FcSYpAVfp8MBw_jjTxnw7lee89PRTyc12UGQ,982
+dataroom_client-0.0.0.post1.dev0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dataroom_client-0.0.0.post1.dev0.dist-info/RECORD,,

dataroom_client-0.0.0.post1.dev0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 2.1.3
+Root-Is-Purelib: true
+Tag: py3-none-any