package name fix

basis-company · Nov 13, 2023 · 58f9a0e · 58f9a0e
1 parent 9fb16fc
commit 58f9a0e
Show file tree

Hide file tree

Showing 7 changed files with 16 additions and 15 deletions.
diff --git a/sharded/__init__.py → registry/__init__.py b/sharded/__init__.py → registry/__init__.py
diff --git a/sharded/drivers.py → registry/drivers.py b/sharded/drivers.py → registry/drivers.py
@@ -1,7 +1,7 @@
 from typing import Protocol
 
-from sharded.entity import Entity
-from sharded.schema import StorageDriver
+from registry.entity import Entity
+from registry.schema import StorageDriver
 
 
 class Driver(Protocol):

diff --git a/sharded/entity.py → registry/entity.py b/sharded/entity.py → registry/entity.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 
-from sharded.schema import BucketStatus, StorageClass, StorageDriver
+from registry.schema import BucketStatus, StorageClass, StorageDriver
 
 
 @dataclass

diff --git a/sharded/registry.py → registry/registry.py b/sharded/registry.py → registry/registry.py
@@ -2,11 +2,12 @@
 from functools import cache
 from typing import Any, Optional
 
-from sharded.drivers import Driver, get_driver
-from sharded.entity import Bucket, Entity, Storage
-from sharded.repository import (BucketRepository, Repository,
-                                StorageRepository, get_entity_repository_class)
-from sharded.schema import BucketStatus, StorageClass, StorageDriver
+from registry.drivers import Driver, get_driver
+from registry.entity import Bucket, Entity, Storage
+from registry.repository import (BucketRepository, Repository,
+                                 StorageRepository,
+                                 get_entity_repository_class)
+from registry.schema import BucketStatus, StorageClass, StorageDriver
 
 
 @dataclass
@@ -151,7 +152,7 @@ async def context(self, entity: type[Entity], key: Any) -> QueryContext:
 
         return QueryContext(bucket, driver, entity, repository)
 
-    async def get_bucket(self, repository: Repository, key: Any) -> Bucket:
+    async def get_bucket(self, repository: Repository, key: Any):
         if isinstance(repository, BucketRepository | StorageRepository):
             buckets = self.get_repository(BucketRepository)
             bucket = buckets.map[Bucket][repository.bucket_id]

diff --git a/sharded/repository.py → registry/repository.py b/sharded/repository.py → registry/repository.py
@@ -2,9 +2,9 @@
 from functools import cache
 from typing import Optional
 
-from sharded.drivers import Driver
-from sharded.entity import Bucket, Entity, Storage
-from sharded.schema import BucketStatus
+from registry.drivers import Driver
+from registry.entity import Bucket, Entity, Storage
+from registry.schema import BucketStatus
 
 
 @dataclass

diff --git a/sharded/schema.py → registry/schema.py b/sharded/schema.py → registry/schema.py
diff --git a/tests/test_repository.py b/tests/test_repository.py
@@ -2,9 +2,9 @@
 
 from pytest import mark
 
-from sharded.entity import Entity
-from sharded.registry import Registry
-from sharded.repository import Index, Repository
+from registry.entity import Entity
+from registry.registry import Registry
+from registry.repository import Index, Repository
 
 
 @dataclass