diff --git a/orm/__init__.py b/orm/__init__.py index e75104ed..b8d196e7 100644 --- a/orm/__init__.py +++ b/orm/__init__.py @@ -3,6 +3,6 @@ from orm.user import User from orm.message import Message from orm.base import Base, engine -__all__ = ["User", "Role", "Operation", "Permission"] +__all__ = ["User", "Role", "Operation", "Permission", "Message"] Base.metadata.create_all(engine) diff --git a/resolvers/__init__.py b/resolvers/__init__.py index cd35ddbb..bf479638 100644 --- a/resolvers/__init__.py +++ b/resolvers/__init__.py @@ -1,4 +1,14 @@ from resolvers.auth import sign_in, sign_out, register -from resolvers.inbox import create_message +from resolvers.inbox import create_message, delete_message, update_message, get_messages -__all__ = ["sign_in", "sign_out", "register"] +__all__ = [ + "sign_in", + "sign_out", + "register", + # TODO: "reset_password_code", + # TODO: "reset_password_confirm" + "create_message", + "delete_message", + "get_messages", + "update_message" + ]