add part of territories

back/tgbot/tgbot.py

@@ -31,7 +31,7 @@ from django.db import models
 
 from .apps import TgBotUpdater
 from .models import TgItem, TmcElement, TmcField
-from tmc.models import CustomTable, BaseCustomField
+from tmc.models import CustomTable, BaseCustomField, Territory, TerritoryItem
 
 
 logger = logging.getLogger(__name__)
@@ -84,6 +84,7 @@ class TgBot:
         TgBot.app.add_handler(
             CommandHandler("start", self.start, filters.ChatType.PRIVATE)
         )
+
         TgBot.app.add_handler(CommandHandler("my", self.my, filters.ChatType.PRIVATE))
         TgBot.app.add_handler(CommandHandler("inv", self.inv, filters.ChatType.PRIVATE))
         TgBot.app.add_handler(
@@ -96,6 +97,15 @@ class TgBot:
                 self.inv,
             )
         )
+        TgBot.app.add_handler(
+            MessageHandler(
+                filters.ChatType.PRIVATE & filters.Regex(r"/ter_(\d.*)"),
+                self.ter_deep,
+            )
+        )
+        TgBot.app.add_handler(CommandHandler("ter", self.ter, filters.ChatType.PRIVATE))
+        TgBot.app.add_handler(CallbackQueryHandler(self.ter_back, "ter_back"))
+        TgBot.app.add_handler(CallbackQueryHandler(self.ter_next, "ter_next"))
         TgBot.app.add_handler(CallbackQueryHandler(self.stop_inv, "stop_inv"))
         TgBot.app.add_handler(CallbackQueryHandler(self.get_inv, r"get_inv@(.*?)"))
         TgBot.app.add_handler(CallbackQueryHandler(self.add_tmc, r"add_tmc@(.*?)"))
@@ -110,6 +120,7 @@ class TgBot:
                 "Это бот для проведения инвентаризации\n"
                 "/my \-\- продолжить инвентаризацию\n"
                 "/inv \-\- начать новую инвентаризацию"
+                "/ter \-\- список территорий"
             ),
             # reply_markup=ForceReply(selective=True),
             reply_parameters=ReplyParameters(message_id=update.message.message_id),
@@ -178,6 +189,20 @@ class TgBot:
             )
         ]
 
+    async def ter_back(self, update: Update, context: CallbackContext):
+        query = update.callback_query
+        await query.answer()
+        context.chat_data["territory_start"] -= context.chat_data["territory_count"]
+        context.chat_data["territory_renew"] = True
+        await self.inv(update, context)
+
+    async def ter_next(self, update: Update, context: CallbackContext):
+        query = update.callback_query
+        await query.answer()
+        context.chat_data["territory_start"] += context.chat_data["territory_count"]
+        context.chat_data["territory_renew"] = True
+        await self.inv(update, context)
+
     async def stop_inv(self, update: Update, context: CallbackContext):
         query = update.callback_query
         await update.effective_message.edit_reply_markup(InlineKeyboardMarkup([]))
@@ -237,13 +262,122 @@ class TgBot:
 
         await self.inv(update, context)
 
+    async def ter(self, update: Update, context: CallbackContext):
+        user = update.effective_user
+        current_step = context.chat_data.get("step", None)
+
+        logger.info(f"Step {current_step} from user {user.full_name}")
+        context.chat_data["step"] = "territory"
+        if "territory_start" not in context.chat_data:
+            context.chat_data["territory_start"] = 10
+        if "territory_count" not in context.chat_data:
+            context.chat_data["territory_count"] = 10
+
+        locations = []
+        async for e in Territory.objects.all():
+            locations.append({"name": e.name, "id": e.id})
+
+        ter_start = context.chat_data["territory_start"]
+        ter_step = context.chat_data["territory_count"]
+        ter_end = ter_start + ter_step
+
+        text = "\n".join(
+            [
+                self.format_username(user),
+                "Выберите объект:",
+                "\n".join(
+                    [
+                        f"/ter_{i['id']} {i['name']}"
+                        for i in locations[ter_start:ter_end]
+                    ]
+                ),
+            ]
+        )
+
+        keyboard = []
+        if ter_start > 0:
+            keyboard.append(InlineKeyboardButton(text="←", callback_data="ter_back"))
+        if ter_end < len(locations):
+            keyboard.append(InlineKeyboardButton(text="→", callback_data="ter_next"))
+
+        keyboard = [keyboard]
+
+        if "territory_renew" in context.chat_data:
+            await update.effective_message.edit_text(
+                text,
+                reply_markup=InlineKeyboardMarkup(inline_keyboard=keyboard),
+            )
+        else:
+            await update.effective_message.reply_text(
+                text,
+                reply_parameters=ReplyParameters(
+                    message_id=update.effective_message.message_id
+                ),
+                reply_markup=InlineKeyboardMarkup(inline_keyboard=keyboard),
+            )
+            
+    async def ter_deep(self, update: Update, context: CallbackContext):
+        user = update.effective_user
+        current_step = context.chat_data.get("step", None)
+
+        logger.info(f"Step {current_step} from user {user.full_name}")
+        context.chat_data["step"] = "territory"
+        # if "territory_start" not in context.chat_data:
+        #     context.chat_data["territory_start"] = 10
+        # if "territory_count" not in context.chat_data:
+        #     context.chat_data["territory_count"] = 10
+            
+        parent = re.findall(r"/ter_(\d.*)", update.effective_message.text)
+        ter = await Territory.objects.aget(id=parent[0])
+        locations = []
+        async for e in TerritoryItem.objects.filter(parent=ter):
+            locations.append({"name": e.name, "id": e.id})
+        logger.info(parent[0])
+        # ter_start = context.chat_data["territory_start"]
+        # ter_step = context.chat_data["territory_count"]
+        ter_start = 0
+        ter_step = 10
+        ter_end = ter_start + ter_step
+
+        text = "\n".join(
+            [
+                "Выберите:",
+                "\n".join(
+                    [
+                        f"/ter_{i['id']} {i['name']}"
+                        for i in locations[ter_start:ter_end]
+                    ]
+                ),
+            ]
+        )
+
+        keyboard = []
+        if ter_start > 0:
+            keyboard.append(InlineKeyboardButton(text="←", callback_data="ter_back"))
+        if ter_end < len(locations):
+            keyboard.append(InlineKeyboardButton(text="→", callback_data="ter_next"))
+
+        keyboard = [keyboard]
+
+        if "territory_renew" in context.chat_data:
+            await update.effective_message.edit_text(
+                text,
+                reply_markup=InlineKeyboardMarkup(inline_keyboard=keyboard),
+            )
+        else:
+            await update.effective_message.reply_text(
+                text,
+                reply_parameters=ReplyParameters(
+                    message_id=update.effective_message.message_id
+                ),
+                reply_markup=InlineKeyboardMarkup(inline_keyboard=keyboard),
+            )
+            
     async def inv(self, update: Update, context: CallbackContext):
         user = update.effective_user
         current_step = context.chat_data.get("step", None)
 
-        logger.info(
+        logger.info(f"Step {current_step} from user {user.full_name}")
-            f"Step {current_step} from user {user.full_name} in {update.message}"
-        )
 
         if update.effective_message.text == "/inv":
             context.chat_data["step"] = "name"
@@ -254,9 +388,10 @@ class TgBot:
                 ),
             )
 
-        elif current_step == "name":
+        if current_step == "name":
             if not context.chat_data.get("inv", None):
-                inv = await TgItem.objects.acreate(user_id=user.id)
+                loc = await TerritoryItem.objects.aget(id=35)
+                inv = await TgItem.objects.acreate(user_id=user.id, location=loc)
                 inv.name = update.message.text
                 await inv.asave()
             else:
@@ -363,6 +498,7 @@ class TgBot:
                 ),
                 reply_markup=ReplyKeyboardRemove(),
             )
+
         elif current_step == "add_element_data":
             inv = await TgItem.objects.aget(id=context.chat_data["inv"])
             tmc_element = await inv.tmc.aget(id=context.chat_data["tmc"])
@@ -393,7 +529,7 @@ class TgBot:
                 await self.inv(update, context)
 
         else:
-            logger.info(update.message.entities)
+            # logger.info(update.message.entities)
             logger.info(f"no step for update {update}")
 
         if "step" in context.chat_data and context.chat_data["step"] == current_step: