Improve imports

src/compub/db/__init__.py

@@ -1 +1,2 @@
-from .base import LocalAsyncSession, ORMBase, engine
+from .geography import *
+from .companies import *

src/compub/db/companies.py

@@ -17,6 +17,15 @@ from compub.utils import multi_max
 from .base import AbstractBase, ORMBase
 
 
+__all__ = [
+    'LegalForm',
+    'LegalFormSubcategory',
+    'Integer',
+    'Company',
+    'CompanyName',
+]
+
+
 class LegalForm(AbstractBase):
     __tablename__ = 'legal_form'
 

src/compub/db/geography.py

@@ -8,6 +8,14 @@ from sqlalchemy_utils.types import CountryType, UUIDType
 from .base import AbstractBase
 
 
+__all__ = [
+    'StateProvince',
+    'City',
+    'Street',
+    'Address',
+]
+
+
 class StateProvince(AbstractBase):
     __tablename__ = 'state_province'