From e02a964fb9bc5f82890a94be281f191ec7976c9f Mon Sep 17 00:00:00 2001
From: Untone <anton.rewin@gmail.com>
Date: Fri, 2 Jul 2021 10:31:30 +0300
Subject: [PATCH] minor fixes

---
 orm/__init__.py       |  2 +-
 resolvers/__init__.py | 14 ++++++++++++--
 2 files changed, 13 insertions(+), 3 deletions(-)

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"
+    ]