cs fix

basis-company · Nov 15, 2023 · 74aaff0 · 74aaff0
1 parent bacc319
commit 74aaff0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/registry/drivers.py b/registry/drivers.py
@@ -25,6 +25,8 @@ async def find_one(
         if len(rows):
             return rows[0]
 
+        return None
+
     async def find_or_create(
         self, entity: type[Entity], query: dict, data: dict
     ) -> dict:
@@ -41,7 +43,7 @@ async def find_or_fail(
     ) -> dict:
         instance = await self.find_one(entity, queries)
         if not instance:
-            raise LookupError(f'{entity.nick} not found')
+            raise LookupError(f'{entity.__name__} not found')
 
         return instance
 

diff --git a/registry/registry.py b/registry/registry.py
@@ -40,13 +40,13 @@ def get_repository(self, cls: type[Repository]) -> Repository:
 
         return self.repositories[cls]
 
-    async def bootstrap(self):
+    async def bootstrap(self) -> None:
         if self.ready:
             return
 
-        primary: Optional[Storage] = None
         self.ready = True
 
+        primary: Optional[Storage] = None
         for candidate in self.storages:
             if candidate.id == StorageRepository.storage_id:
                 primary = self.storages[0]