devide project

src/docbot/handlers/admins_handler.py

@@ -1,10 +0,0 @@
-from telegram import Update
-from telegram.ext import (
-    ContextTypes,
-    ConversationHandler,
-    CommandHandler,
-    MessageHandler,
-    filters,
-)
-
-

src/docbot/handlers/doctors/register_handler.py

@@ -7,7 +7,7 @@ from telegram.ext import (
     filters,
 )
 
-from docbot.handlers.cancel_handler import get_cancel_handler
+from docbot.handlers.utils.cancel_handler import get_cancel_handler
 from docbot.services.referral_service import validate_referral_code, mark_referral_code_as_used
 from core.enums.consultation_types import Consultation
 

src/docbot/handlers/get_form_handler.py

@@ -7,7 +7,7 @@ from telegram.ext import (
     filters,
 )
 from docbot.services.session_service import create_session_code, get_pending_session
-from docbot.handlers.cancel_handler import get_cancel_handler
+from docbot.handlers.utils.cancel_handler import get_cancel_handler
 
 
 ASK_LINK = 1

src/docbot/handlers/patients/send_form_handler.py

@@ -7,7 +7,7 @@ from telegram.ext import (
     filters,
 )
 from docbot.services.session_service import create_session_code, get_pending_session
-from docbot.handlers.cancel_handler import get_cancel_handler
+from docbot.handlers.utils.cancel_handler import get_cancel_handler
 
 
 ASK_LINK = 1

src/docbot/handlers/utils/help.py

@@ -1,8 +1,6 @@
 from telegram import Update, ReplyKeyboardMarkup
-from telegram.ext import ContextTypes, CommandHandler
+from telegram.ext import ContextTypes
 
-from core.config import settings
-from docbot.services.referral_service import ReferralService
 from docbot.services.admins_service import get_admin_info
 from docbot.services.doctors_service import get_doctor_info
 

src/docbot/handlers/utils/unknown.py

@@ -1,5 +1,5 @@
 from telegram import Update
-from telegram.ext import ContextTypes, CommandHandler, MessageHandler, filters
+from telegram.ext import ContextTypes, MessageHandler, filters
 
 
 async def unknown_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:

src/docbot/main.py

@@ -3,11 +3,11 @@ from telegram.ext import ApplicationBuilder, PicklePersistence, ExtBot
 
 from core.config import settings
 from docbot.handlers.start_handler import get_start_handler
-from docbot.handlers.send_form_handler import get_send_form_handler
+from docbot.handlers.patients.send_form_handler import get_send_form_handler
-from docbot.handlers.doctors_handler import get_doctors_handler
+from docbot.handlers.admins.doctors_handler import get_doctors_handler
-from docbot.handlers.register_handler import get_register_doctor_handler
+from docbot.handlers.doctors.register_handler import get_register_doctor_handler
-from docbot.handlers.generate_ref import get_referral_handlers
+from docbot.handlers.admins.generate_ref import get_referral_handlers
-from docbot.handlers.unknown import get_unknown_handler
+from docbot.handlers.utils.unknown import get_unknown_handler
 
 
 def main():

src/docbot/services/admins_service.py

@@ -6,6 +6,15 @@ from db.models import Admins
 
 
 async def get_admin_info(telegram_id: int) -> Admins | None:
+    async with AsyncSessionLocal() as session:
+        result = await session.execute(
+            select(Admins.telegram_id)
+            .where(Admins.telegram_id.match(str(telegram_id)))
+        )
+        return result.scalar_one_or_none()
+
+
+async def mark_doctor_inactive(telegram_id: int) -> Admins | None:
     async with AsyncSessionLocal() as session:
         result = await session.execute(
             select(Admins.telegram_id)