diff --git a/src/infra/__init__.py b/src/infra/__init__.py index e69de29..8851384 100644 --- a/src/infra/__init__.py +++ b/src/infra/__init__.py @@ -0,0 +1,27 @@ +from .db import Base, CRUD, CRUDRepo, UpdateMissingEntryError, get_database +from .users import UserModel, UserRepoImpl +from .publication import ( + PublicationEntryModel, + PublicationSequenceModel, + PublicationOrderModel, + PublicationModel, + PublicationRepoImpl, +) +from .subscription import SubscriptionModel, SubscriptionRepoImpl + +__all__ = [ + 'Base', + 'CRUD', + 'CRUDRepo', + 'UpdateMissingEntryError', + 'get_database', + 'UserModel', + 'UserRepoImpl', + 'PublicationEntryModel', + 'PublicationSequenceModel', + 'PublicationOrderModel', + 'PublicationModel', + 'PublicationRepoImpl', + 'SubscriptionModel', + 'SubscriptionRepoImpl', +] diff --git a/src/main.py b/src/main.py index aca6e66..f9fb8fd 100644 --- a/src/main.py +++ b/src/main.py @@ -6,8 +6,7 @@ from src.services.email import EmailService, get_email_service from src.services.auth import AuthService from src.services.users.service import UserService from dataclasses import dataclass -from src.infra.db import get_database -from src.infra.users import UserRepoImpl +from src.infra import get_database, UserRepoImpl from waitress import serve from flask import Flask import argparse