xync-bot 0.3.5__py3-none-any.whl → 0.3.7.dev1__py3-none-any.whl

xync_bot/__init__.py

@@ -0,0 +1,20 @@
+import logging
+from asyncio import run
+
+from PGram import Bot
+from aiogram.client.default import DefaultBotProperties
+from x_model import init_db
+
+from xync_bot.routers.cond import r
+
+if __name__ == "__main__":
+    from xync_bot.loader import TOKEN, TORM
+
+    logging.basicConfig(level=logging.INFO)
+
+    async def main() -> None:
+        cn = await init_db(TORM)
+        bot = Bot(TOKEN, [r], cn, default=DefaultBotProperties(parse_mode="HTML"))
+        await bot.start()
+
+    run(main())

xync_bot/loader.py

@@ -0,0 +1,21 @@
+from dotenv import load_dotenv
+from os import getenv as env
+
+from xync_schema import models
+
+load_dotenv()
+
+PG_DSN = (
+    f"postgres://{env('POSTGRES_USER')}:{env('POSTGRES_PASSWORD')}@{env('POSTGRES_HOST', 'dbs')}"
+    f":{env('POSTGRES_PORT', 5432)}/{env('POSTGRES_DB', env('POSTGRES_USER'))}"
+)
+TOKEN = env("TOKEN")
+# API_URL = "https://" + env("API_DOMAIN")
+# WH_URL = API_URL + "/wh/" + TOKEN
+
+TORM = {
+    "connections": {"default": PG_DSN},
+    "apps": {"models": {"models": [models, "aerich.models"]}},
+    "use_tz": False,
+    "timezone": "UTC",
+}

xync_bot/routers/cond/__init__.py

@@ -0,0 +1,179 @@
+import re
+from enum import IntEnum
+from inspect import isclass
+from typing import Coroutine
+
+from aiogram import Router, F
+from aiogram.filters import CommandStart
+from aiogram.fsm.context import FSMContext
+from aiogram.types import (
+    Message,
+    ReplyKeyboardMarkup,
+    KeyboardButton,
+    InlineKeyboardMarkup,
+    InlineKeyboardButton,
+    CallbackQuery,
+    # ReplyKeyboardRemove,
+)
+from cyrtranslit import to_latin
+from xync_schema import models
+from xync_schema.enums import SynonymType, Party, Slip, AbuserType, NameType, Boundary
+
+r = Router()
+
+
+async def wrap_cond(txt: str):
+    bnks = await models.Synonym.filter(typ=SynonymType.bank)
+    bnks = "|".join("\\b" + b.txt + ("\\b" if b.boundary & Boundary.right else "") for b in bnks)
+    for syn in await models.Synonym.all():
+        lb, rb = "\\b" if syn.boundary & Boundary.left else "", "\\b" if syn.boundary & Boundary.right else ""
+        if syn.typ == SynonymType.bank_side:
+            syn.txt.replace("#banks#", f"({bnks})")
+        if syn.is_re or syn.txt in txt:
+            pattern = re.compile(lb + syn.txt + rb)
+            if match := re.search(pattern, txt):
+                g = match.group()
+                val, hval = await get_val(syn.typ, syn.val)
+                val = syn.typ.name + (f'="{hval}"' if hval else "")
+                txt = re.sub(pattern, f"<code>{g}</code><tg-spoiler>[{val}]</tg-spoiler>", txt)
+    return txt
+
+
+async def cbanks(bnid: str) -> list[tuple[int, str]]:
+    beginning = to_latin(bnid[:2], lang_code="ru")
+    return await models.Pm.filter(norm__startswith=beginning, bank=True).values_list("id", "norm")
+
+
+async def cppo(txt: str) -> list[tuple[int, str]]:
+    opts = re.findall(r"\d+", txt) or [1, 2, 3, 5, 10]
+    return [(o, str(o)) for o in opts]
+
+
+synopts: dict[SynonymType, list[str] | type(IntEnum) | None | Coroutine] = {
+    SynonymType.name: ["not_slavic", "slavic"],
+    SynonymType.ppo: cppo,
+    SynonymType.from_party: Party,
+    SynonymType.to_party: Party,
+    SynonymType.slip_req: Slip,
+    SynonymType.slip_send: Slip,
+    SynonymType.abuser: AbuserType,
+    SynonymType.scale: ["1", "10", "100", "1000"],
+    SynonymType.slavic: NameType,
+    SynonymType.mtl_like: None,
+    SynonymType.bank: cbanks,
+    SynonymType.bank_side: ["except", "only"],
+}
+rkm = ReplyKeyboardMarkup(
+    keyboard=[
+        [KeyboardButton(text="ppo"), KeyboardButton(text="abuser")],
+        [KeyboardButton(text="from_party"), KeyboardButton(text="to_party")],
+        [KeyboardButton(text="slip_send"), KeyboardButton(text="slip_req")],
+        [KeyboardButton(text="name"), KeyboardButton(text="slavic")],
+        [KeyboardButton(text="scale"), KeyboardButton(text="mtl_like")],
+        [KeyboardButton(text="bank"), KeyboardButton(text="bank_side")],
+    ],
+    one_time_keyboard=True,
+)
+
+
+async def get_val(typ: SynonymType.__class__, val: str) -> tuple[SynonymType | int | bool, str]:
+    if isinstance(val, str) and val.isnumeric():
+        val = int(val)
+    if isclass(lst := synopts[typ]) and issubclass(lst, IntEnum):
+        return (v := lst(val)), v.name
+    elif isinstance(lst, list):
+        return val, lst[val]
+    elif typ == SynonymType.bank:
+        return val, (await models.Pm[val]).norm
+    return val, val
+
+
+async def btns(typ: SynonymType.__class__, txt: str = None) -> InlineKeyboardMarkup | None:
+    if lst := synopts[typ]:
+        if isinstance(lst, list):
+            kb = [[InlineKeyboardButton(text=n, callback_data=f"st:{typ.name}:{i}")] for i, n in enumerate(lst)]
+        elif isclass(lst) and issubclass(lst, IntEnum):
+            kb = [[InlineKeyboardButton(text=i.name, callback_data=f"st:{typ.name}:{i.value}")] for i in lst]
+        else:
+            kb = [[InlineKeyboardButton(text=n, callback_data=f"st:{typ.name}:{i}")] for i, n in await lst(txt)]
+        return InlineKeyboardMarkup(inline_keyboard=kb)
+    else:
+        return lst
+
+
+@r.message(CommandStart())
+async def start(msg: Message, state: FSMContext):
+    cond = await models.Cond.filter(parsed__isnull=True).order_by("-created_at").first().prefetch_related("parsed")
+    await state.set_data({"cid": cond.id, "cond_txt": cond.raw_txt})
+    await msg.reply(await wrap_cond(cond.raw_txt), reply_markup=rkm)
+
+
+@r.message(F.quote)
+async def got_synonym(msg: Message, state: FSMContext):
+    if not (msg.text in {st.name for st in SynonymType} and SynonymType[msg.text]):
+        return await msg.reply_text(
+            f'Нет раздела "{msg.text}", не пиши текст сам, выдели кусок из моего сообщения,'
+            f"ответь на него, выбери кнопку раздела"
+        )
+    if not msg.quote:
+        return await msg.reply_text(f"Вы забыли выделить кусок текста для {msg.text}")
+    if typ := SynonymType[msg.text]:
+        await state.update_data({"syntext": msg.quote.text, "cmsg": msg.reply_to_message})
+        await models.Synonym.update_or_create({"typ": typ}, txt=msg.quote.text)
+        if rm := await btns(typ, msg.quote.text):
+            return await msg.answer("Уточните", reply_markup=rm, reply_to_message_id=msg.message_id)
+        await syn_result(msg, f"st:{typ.name}:1", state)
+    return None
+
+
+@r.callback_query()
+async def got_synonym_val(cbq: CallbackQuery, state: FSMContext):
+    await syn_result(cbq.message, cbq.data, state)
+
+
+async def syn_result(msg: Message, data: str, state: FSMContext):
+    t, st, sv = data.split(":")
+    if t == "st":
+        typ = SynonymType[st]
+        val, hval = await get_val(typ, sv)
+        syntext = await state.get_value("syntext")
+        cid = await state.get_value("cid")
+        syn, _ = await models.Synonym.update_or_create({"val": val}, typ=typ, txt=syntext)
+        await models.CondParsed.update_or_create({typ.name: val}, cond_id=cid)
+        await msg.reply(
+            f'Текст "{syntext}" определен как синоним для `{typ.name}` со значением {hval}',
+            reply_markup=InlineKeyboardMarkup(
+                inline_keyboard=[
+                    [
+                        InlineKeyboardButton(text="Готово! Давай новый", callback_data=f"cond:complete:{cid}"),
+                        InlineKeyboardButton(text="Продолжить с этим текстом", callback_data=f"cond:process:{cid}"),
+                    ]
+                ]
+            ),
+        )
+        # await msg.reply_to_message.delete()
+        # await msg.delete()
+        # await cmsg.edit_text(wrapped_txt)
+
+    elif t == "cond":
+        if st == "complete":
+            await models.CondParsed.update_or_create({"parsed": True}, cond_id=int(sv))
+            await start(msg, state)
+        else:
+            wrapped_txt = await wrap_cond(await state.get_value("cond_txt"))
+            # cmsg: Message = await state.get_value('cmsg')
+            await msg.reply(wrapped_txt, reply_markup=rkm)
+    else:
+        await msg.reply("Где я?")
+
+
+@r.message(F.reply_to_message)
+async def clear(msg: Message):
+    await msg.reply_to_message.delete_reply_markup()
+    await msg.delete()
+
+
+@r.message()
+async def unknown(msg: Message):
+    # user = await User.get(username_id=msg.from_user.id)
+    await msg.delete()

xync_bot/{handlers → routers}/main.py

@@ -15,9 +15,9 @@ from aiogram.types import (
     WebAppInfo,
 )
 from aiogram.utils.deep_linking import create_start_link
-from xync_bot.handlers import user_upsert
 from xync_schema.models import User, Order, Msg, Forum
 
+from xync_bot.routers import user_upsert
 from xync_bot.shared import NavCallbackData
 
 main = Router()

{xync_bot-0.3.5.dist-info → xync_bot-0.3.7.dev1.dist-info}/METADATA

@@ -1,17 +1,19 @@
 Metadata-Version: 2.4
 Name: xync-bot
-Version: 0.3.5
+Version: 0.3.7.dev1
 Summary: Telegram bot with web app for xync net
 Author-email: Artemiev <mixartemev@gmail.com>
-License: EULA
+License-Expression: GPL-3.0-or-later
 Project-URL: Homepage, https://gitlab.com/xync/back/tg-bot
 Project-URL: Repository, https://gitlab.com/xync/back/tg-bot
-Keywords: aiogram,xync-net
+Keywords: aiogram,cyrtranslit,xync-net
 Requires-Python: >=3.12
-Description-Content-Type: text/markdown
 Requires-Dist: xync-schema
 Provides-Extra: dev
+Requires-Dist: PGram; extra == "dev"
+Requires-Dist: pre-commit; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: python-dotenv; extra == "dev"
+Requires-Dist: setuptools-scm; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"

xync_bot-0.3.7.dev1.dist-info/RECORD

@@ -0,0 +1,16 @@
+xync_bot/__init__.py,sha256=DxSf_OvNstlmK9_-z7Q-3ld1SC6Hd6gWxBlFffd7Cjo,490
+xync_bot/loader.py,sha256=wnhRRMJ6KSFAFZnDBGg9mbbktq34VHtcNSt89WuPn04,558
+xync_bot/shared.py,sha256=MlKkTrsT29l7fF6-qAN9FO14cSuXuOuYxbNY5F4S2w4,137
+xync_bot/typs.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xync_bot/routers/__init__.py,sha256=0CwHTSgAWzKRZ7PtsaLGA8PtjWQtZqsEXKyncNbCKLA,374
+xync_bot/routers/main.py,sha256=ZfgEX17_QaZ8CVEQqNZnNFt8RLfr6xfZ_A2ZGJoe-Oo,9520
+xync_bot/routers/order.py,sha256=ZKWDLyiWrXzcR-aHKLBTBCACwp-P0Vvnr22T-EuLHaM,274
+xync_bot/routers/photo.py,sha256=aq6ImIOoZQYTW-lEy26qjgj5TYAuk4bQjgiCv64mPJs,1203
+xync_bot/routers/vpn.py,sha256=qKK55UrjEZeDvu7ljWXNUFBFgXTPTIEaCT2OAmKWky4,2219
+xync_bot/routers/xicon.png,sha256=O57_kvzhVcCXSoGYZ61m0dW9pizY6gxR8Yj5aeCP0RQ,429283
+xync_bot/routers/cond/__init__.py,sha256=n9MLqu0tENaA3n2-WSh7XAelXxyFJtCQGdd6lqP7GmI,7248
+xync_bot/routers/cond/cond.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xync_bot-0.3.7.dev1.dist-info/METADATA,sha256=g2DghtBKWq9Igmbx6p_sKymDH473mWvq-fdogyuu980,700
+xync_bot-0.3.7.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xync_bot-0.3.7.dev1.dist-info/top_level.txt,sha256=O2IjMc1ryAf0rwIXWohSNT5Kzcs9johgKRDz8lCC0rs,9
+xync_bot-0.3.7.dev1.dist-info/RECORD,,

{xync_bot-0.3.5.dist-info → xync_bot-0.3.7.dev1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

xync_bot-0.3.5.dist-info/RECORD

@@ -1,13 +0,0 @@
-xync_bot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xync_bot/shared.py,sha256=MlKkTrsT29l7fF6-qAN9FO14cSuXuOuYxbNY5F4S2w4,137
-xync_bot/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xync_bot/handlers/__init__.py,sha256=0CwHTSgAWzKRZ7PtsaLGA8PtjWQtZqsEXKyncNbCKLA,374
-xync_bot/handlers/main.py,sha256=eIgorrJXzRJfsO7fx1XxVHFJDfMiJyWt7myS4qSqS8U,9521
-xync_bot/handlers/order.py,sha256=ZKWDLyiWrXzcR-aHKLBTBCACwp-P0Vvnr22T-EuLHaM,274
-xync_bot/handlers/photo.py,sha256=aq6ImIOoZQYTW-lEy26qjgj5TYAuk4bQjgiCv64mPJs,1203
-xync_bot/handlers/vpn.py,sha256=qKK55UrjEZeDvu7ljWXNUFBFgXTPTIEaCT2OAmKWky4,2219
-xync_bot/handlers/xicon.png,sha256=O57_kvzhVcCXSoGYZ61m0dW9pizY6gxR8Yj5aeCP0RQ,429283
-xync_bot-0.3.5.dist-info/METADATA,sha256=Ixje9jeQi2Xf6sJXWbag5W2Tbz3TrsfLv5-2QIpaaEk,575
-xync_bot-0.3.5.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-xync_bot-0.3.5.dist-info/top_level.txt,sha256=O2IjMc1ryAf0rwIXWohSNT5Kzcs9johgKRDz8lCC0rs,9
-xync_bot-0.3.5.dist-info/RECORD,,