xync-bot 0.3.7.dev1__py3-none-any.whl → 0.3.24.dev3__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of xync-bot might be problematic. Click here for more details.
- xync_bot/__init__.py +8 -2
- xync_bot/loader.py +2 -0
- xync_bot/routers/__init__.py +61 -5
- xync_bot/routers/cond/__init__.py +70 -152
- xync_bot/routers/cond/func.py +118 -0
- xync_bot/routers/main.py +35 -36
- xync_bot/routers/pay/cd.py +50 -0
- xync_bot/routers/pay/dep.py +70 -0
- xync_bot/routers/pay/handler.py +299 -0
- xync_bot/routers/pay/window.py +232 -0
- {xync_bot-0.3.7.dev1.dist-info → xync_bot-0.3.24.dev3.dist-info}/METADATA +3 -1
- xync_bot-0.3.24.dev3.dist-info/RECORD +20 -0
- xync_bot/routers/cond/cond.py +0 -0
- xync_bot-0.3.7.dev1.dist-info/RECORD +0 -16
- {xync_bot-0.3.7.dev1.dist-info → xync_bot-0.3.24.dev3.dist-info}/WHEEL +0 -0
- {xync_bot-0.3.7.dev1.dist-info → xync_bot-0.3.24.dev3.dist-info}/top_level.txt +0 -0
xync_bot/__init__.py
CHANGED
|
@@ -5,7 +5,11 @@ from PGram import Bot
|
|
|
5
5
|
from aiogram.client.default import DefaultBotProperties
|
|
6
6
|
from x_model import init_db
|
|
7
7
|
|
|
8
|
-
from xync_bot.routers
|
|
8
|
+
from xync_bot.routers import Store
|
|
9
|
+
from xync_bot.routers.cond import cr as cr
|
|
10
|
+
|
|
11
|
+
# from xync_bot.routers.main import mr
|
|
12
|
+
from xync_bot.routers.pay.handler import pay as pay
|
|
9
13
|
|
|
10
14
|
if __name__ == "__main__":
|
|
11
15
|
from xync_bot.loader import TOKEN, TORM
|
|
@@ -14,7 +18,9 @@ if __name__ == "__main__":
|
|
|
14
18
|
|
|
15
19
|
async def main() -> None:
|
|
16
20
|
cn = await init_db(TORM)
|
|
17
|
-
|
|
21
|
+
store = Store()
|
|
22
|
+
store.glob = await Store.Global()
|
|
23
|
+
bot = Bot(TOKEN, [pay], cn, default=DefaultBotProperties(parse_mode="HTML"), store=store)
|
|
18
24
|
await bot.start()
|
|
19
25
|
|
|
20
26
|
run(main())
|
xync_bot/loader.py
CHANGED
xync_bot/routers/__init__.py
CHANGED
|
@@ -1,8 +1,64 @@
|
|
|
1
|
-
from
|
|
2
|
-
from aiogram.utils.web_app import WebAppUser
|
|
1
|
+
from tortoise.functions import Min
|
|
3
2
|
from xync_schema import models
|
|
4
3
|
|
|
4
|
+
from xync_bot.routers.pay.dep import flags
|
|
5
5
|
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
6
|
+
|
|
7
|
+
class SingleStore(type):
|
|
8
|
+
_store = None
|
|
9
|
+
|
|
10
|
+
async def __call__(cls):
|
|
11
|
+
if not cls._store:
|
|
12
|
+
cls._store = super(SingleStore, cls).__call__()
|
|
13
|
+
cls._store.coins = {k: v for k, v in await models.Coin.all().order_by("ticker").values_list("id", "ticker")}
|
|
14
|
+
cls._store.curs = {
|
|
15
|
+
k: v
|
|
16
|
+
for k, v in await models.Cur.filter(ticker__in=flags.keys())
|
|
17
|
+
.order_by("ticker")
|
|
18
|
+
.values_list("id", "ticker")
|
|
19
|
+
}
|
|
20
|
+
cls._store.exs = {k: v for k, v in await models.Ex.all().values_list("id", "name")}
|
|
21
|
+
cls._store.pms = {
|
|
22
|
+
k: v
|
|
23
|
+
for k, v in await models.Pmex.filter(pm__pmcurs__id__in=cls._store.curs.keys())
|
|
24
|
+
.annotate(sname=Min("name"))
|
|
25
|
+
.group_by("pm_id")
|
|
26
|
+
.values_list("pm_id", "sname")
|
|
27
|
+
}
|
|
28
|
+
cls._store.coinexs = {
|
|
29
|
+
c.id: [ex.ex_id for ex in c.coinexs] for c in await models.Coin.all().prefetch_related("coinexs")
|
|
30
|
+
}
|
|
31
|
+
cls._store.curpms = {
|
|
32
|
+
c.id: [pmc.pm_id for pmc in c.pmcurs]
|
|
33
|
+
for c in await models.Cur.filter(id__in=cls._store.curs.keys()).prefetch_related("pmcurs")
|
|
34
|
+
}
|
|
35
|
+
|
|
36
|
+
return cls._store
|
|
37
|
+
|
|
38
|
+
|
|
39
|
+
class Store:
|
|
40
|
+
class Global(metaclass=SingleStore):
|
|
41
|
+
coins: dict[int, str] # id:ticker
|
|
42
|
+
curs: dict[int, str] # id:ticker
|
|
43
|
+
exs: dict[int, str] # id:name
|
|
44
|
+
pms: dict[int, models.Pm] # id:name
|
|
45
|
+
|
|
46
|
+
class Permanent:
|
|
47
|
+
user: models.User
|
|
48
|
+
actors: dict[int, models.Actor] # key=ex_id
|
|
49
|
+
|
|
50
|
+
class Current:
|
|
51
|
+
t_cur_id: int = None
|
|
52
|
+
s_cur_id: int = None
|
|
53
|
+
t_coin_id: int = None
|
|
54
|
+
s_coin_id: int = None
|
|
55
|
+
t_pm_id: int = None
|
|
56
|
+
s_pm_id: int = None
|
|
57
|
+
t_ex_id: int = None
|
|
58
|
+
s_ex_id: int = None
|
|
59
|
+
addr: models.Addr = None
|
|
60
|
+
cred: models.Cred = None
|
|
61
|
+
|
|
62
|
+
glob: Global
|
|
63
|
+
perm: Permanent = Permanent()
|
|
64
|
+
curr: Current = Current()
|
|
@@ -1,114 +1,33 @@
|
|
|
1
|
-
import re
|
|
2
|
-
from enum import IntEnum
|
|
3
|
-
from inspect import isclass
|
|
4
|
-
from typing import Coroutine
|
|
5
|
-
|
|
6
1
|
from aiogram import Router, F
|
|
7
|
-
from aiogram.
|
|
2
|
+
from aiogram.exceptions import TelegramBadRequest
|
|
3
|
+
from aiogram.filters import Command
|
|
8
4
|
from aiogram.fsm.context import FSMContext
|
|
9
|
-
from aiogram.types import
|
|
10
|
-
Message,
|
|
11
|
-
ReplyKeyboardMarkup,
|
|
12
|
-
KeyboardButton,
|
|
13
|
-
InlineKeyboardMarkup,
|
|
14
|
-
InlineKeyboardButton,
|
|
15
|
-
CallbackQuery,
|
|
16
|
-
# ReplyKeyboardRemove,
|
|
17
|
-
)
|
|
18
|
-
from cyrtranslit import to_latin
|
|
5
|
+
from aiogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton, CallbackQuery
|
|
19
6
|
from xync_schema import models
|
|
20
|
-
from xync_schema.enums import SynonymType
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
for syn in await models.Synonym.all():
|
|
29
|
-
lb, rb = "\\b" if syn.boundary & Boundary.left else "", "\\b" if syn.boundary & Boundary.right else ""
|
|
30
|
-
if syn.typ == SynonymType.bank_side:
|
|
31
|
-
syn.txt.replace("#banks#", f"({bnks})")
|
|
32
|
-
if syn.is_re or syn.txt in txt:
|
|
33
|
-
pattern = re.compile(lb + syn.txt + rb)
|
|
34
|
-
if match := re.search(pattern, txt):
|
|
35
|
-
g = match.group()
|
|
36
|
-
val, hval = await get_val(syn.typ, syn.val)
|
|
37
|
-
val = syn.typ.name + (f'="{hval}"' if hval else "")
|
|
38
|
-
txt = re.sub(pattern, f"<code>{g}</code><tg-spoiler>[{val}]</tg-spoiler>", txt)
|
|
39
|
-
return txt
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
async def cbanks(bnid: str) -> list[tuple[int, str]]:
|
|
43
|
-
beginning = to_latin(bnid[:2], lang_code="ru")
|
|
44
|
-
return await models.Pm.filter(norm__startswith=beginning, bank=True).values_list("id", "norm")
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
async def cppo(txt: str) -> list[tuple[int, str]]:
|
|
48
|
-
opts = re.findall(r"\d+", txt) or [1, 2, 3, 5, 10]
|
|
49
|
-
return [(o, str(o)) for o in opts]
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
synopts: dict[SynonymType, list[str] | type(IntEnum) | None | Coroutine] = {
|
|
53
|
-
SynonymType.name: ["not_slavic", "slavic"],
|
|
54
|
-
SynonymType.ppo: cppo,
|
|
55
|
-
SynonymType.from_party: Party,
|
|
56
|
-
SynonymType.to_party: Party,
|
|
57
|
-
SynonymType.slip_req: Slip,
|
|
58
|
-
SynonymType.slip_send: Slip,
|
|
59
|
-
SynonymType.abuser: AbuserType,
|
|
60
|
-
SynonymType.scale: ["1", "10", "100", "1000"],
|
|
61
|
-
SynonymType.slavic: NameType,
|
|
62
|
-
SynonymType.mtl_like: None,
|
|
63
|
-
SynonymType.bank: cbanks,
|
|
64
|
-
SynonymType.bank_side: ["except", "only"],
|
|
65
|
-
}
|
|
66
|
-
rkm = ReplyKeyboardMarkup(
|
|
67
|
-
keyboard=[
|
|
68
|
-
[KeyboardButton(text="ppo"), KeyboardButton(text="abuser")],
|
|
69
|
-
[KeyboardButton(text="from_party"), KeyboardButton(text="to_party")],
|
|
70
|
-
[KeyboardButton(text="slip_send"), KeyboardButton(text="slip_req")],
|
|
71
|
-
[KeyboardButton(text="name"), KeyboardButton(text="slavic")],
|
|
72
|
-
[KeyboardButton(text="scale"), KeyboardButton(text="mtl_like")],
|
|
73
|
-
[KeyboardButton(text="bank"), KeyboardButton(text="bank_side")],
|
|
74
|
-
],
|
|
75
|
-
one_time_keyboard=True,
|
|
76
|
-
)
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
async def get_val(typ: SynonymType.__class__, val: str) -> tuple[SynonymType | int | bool, str]:
|
|
80
|
-
if isinstance(val, str) and val.isnumeric():
|
|
81
|
-
val = int(val)
|
|
82
|
-
if isclass(lst := synopts[typ]) and issubclass(lst, IntEnum):
|
|
83
|
-
return (v := lst(val)), v.name
|
|
84
|
-
elif isinstance(lst, list):
|
|
85
|
-
return val, lst[val]
|
|
86
|
-
elif typ == SynonymType.bank:
|
|
87
|
-
return val, (await models.Pm[val]).norm
|
|
88
|
-
return val, val
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
async def btns(typ: SynonymType.__class__, txt: str = None) -> InlineKeyboardMarkup | None:
|
|
92
|
-
if lst := synopts[typ]:
|
|
93
|
-
if isinstance(lst, list):
|
|
94
|
-
kb = [[InlineKeyboardButton(text=n, callback_data=f"st:{typ.name}:{i}")] for i, n in enumerate(lst)]
|
|
95
|
-
elif isclass(lst) and issubclass(lst, IntEnum):
|
|
96
|
-
kb = [[InlineKeyboardButton(text=i.name, callback_data=f"st:{typ.name}:{i.value}")] for i in lst]
|
|
97
|
-
else:
|
|
98
|
-
kb = [[InlineKeyboardButton(text=n, callback_data=f"st:{typ.name}:{i}")] for i, n in await lst(txt)]
|
|
99
|
-
return InlineKeyboardMarkup(inline_keyboard=kb)
|
|
100
|
-
else:
|
|
101
|
-
return lst
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
@r.message(CommandStart())
|
|
7
|
+
from xync_schema.enums import SynonymType
|
|
8
|
+
|
|
9
|
+
from xync_bot.routers.cond.func import wrap_cond, get_val, btns, rkm, ikm, SynTypeCd, CondCd
|
|
10
|
+
|
|
11
|
+
cr = Router()
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
@cr.message(Command("cond"))
|
|
105
15
|
async def start(msg: Message, state: FSMContext):
|
|
16
|
+
await msg.reply('Заполняем синонимы признаков условий объявлений: "ссылка на инструкцию"', reply_markup=rkm)
|
|
17
|
+
await show_new_cond(msg, state)
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
async def show_new_cond(msg: Message, state: FSMContext):
|
|
106
21
|
cond = await models.Cond.filter(parsed__isnull=True).order_by("-created_at").first().prefetch_related("parsed")
|
|
107
|
-
await state.set_data({"
|
|
108
|
-
await msg
|
|
22
|
+
await state.set_data({"cond": cond})
|
|
23
|
+
await show_cond(msg, cond)
|
|
109
24
|
|
|
110
25
|
|
|
111
|
-
|
|
26
|
+
async def show_cond(msg: Message, cond: models.Cond):
|
|
27
|
+
await msg.answer(await wrap_cond(cond), reply_markup=ikm)
|
|
28
|
+
|
|
29
|
+
|
|
30
|
+
@cr.message(F.quote)
|
|
112
31
|
async def got_synonym(msg: Message, state: FSMContext):
|
|
113
32
|
if not (msg.text in {st.name for st in SynonymType} and SynonymType[msg.text]):
|
|
114
33
|
return await msg.reply_text(
|
|
@@ -122,58 +41,57 @@ async def got_synonym(msg: Message, state: FSMContext):
|
|
|
122
41
|
await models.Synonym.update_or_create({"typ": typ}, txt=msg.quote.text)
|
|
123
42
|
if rm := await btns(typ, msg.quote.text):
|
|
124
43
|
return await msg.answer("Уточните", reply_markup=rm, reply_to_message_id=msg.message_id)
|
|
125
|
-
await syn_result(msg,
|
|
44
|
+
await syn_result(msg, SynTypeCd(typ=typ.name, val=1), state)
|
|
126
45
|
return None
|
|
127
46
|
|
|
128
47
|
|
|
129
|
-
@
|
|
130
|
-
async def got_synonym_val(cbq: CallbackQuery, state: FSMContext):
|
|
131
|
-
await syn_result(cbq.message,
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
async def syn_result(msg: Message,
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
InlineKeyboardButton(text="Готово! Давай новый", callback_data=f"cond:complete:{cid}"),
|
|
149
|
-
InlineKeyboardButton(text="Продолжить с этим текстом", callback_data=f"cond:process:{cid}"),
|
|
150
|
-
]
|
|
48
|
+
@cr.callback_query(SynTypeCd.filter())
|
|
49
|
+
async def got_synonym_val(cbq: CallbackQuery, callback_data: SynTypeCd, state: FSMContext):
|
|
50
|
+
await syn_result(cbq.message, callback_data, state)
|
|
51
|
+
|
|
52
|
+
|
|
53
|
+
async def syn_result(msg: Message, cbd: SynTypeCd, state: FSMContext):
|
|
54
|
+
cond: models.Cond = await state.get_value("cond")
|
|
55
|
+
typ = SynonymType[cbd.typ]
|
|
56
|
+
val, hval = await get_val(typ, cbd.val)
|
|
57
|
+
syntext = await state.get_value("syntext")
|
|
58
|
+
syn, _ = await models.Synonym.update_or_create({"val": val}, typ=typ, txt=syntext)
|
|
59
|
+
await models.CondParsed.update_or_create({typ.name: val}, cond_id=cond.id)
|
|
60
|
+
await msg.answer(
|
|
61
|
+
f'Текст "{syntext}" определен как синоним для `{typ.name}` со значением {hval}',
|
|
62
|
+
reply_markup=InlineKeyboardMarkup(
|
|
63
|
+
inline_keyboard=[
|
|
64
|
+
[
|
|
65
|
+
InlineKeyboardButton(text="Готово! Давай новый", callback_data="cond:complete"),
|
|
66
|
+
InlineKeyboardButton(text="Продолжить с этим текстом", callback_data="cond:continue"),
|
|
151
67
|
]
|
|
152
|
-
|
|
153
|
-
)
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
68
|
+
]
|
|
69
|
+
),
|
|
70
|
+
)
|
|
71
|
+
|
|
72
|
+
|
|
73
|
+
@cr.callback_query(CondCd.filter())
|
|
74
|
+
async def got_action(cbq: CallbackQuery, callback_data: CondCd, state: FSMContext):
|
|
75
|
+
cond: models.Cond = await state.get_value("cond")
|
|
76
|
+
if callback_data.act == "complete":
|
|
77
|
+
await models.CondParsed.update_or_create({"parsed": True}, cond=cond)
|
|
78
|
+
await show_new_cond(cbq.message, state)
|
|
79
|
+
elif callback_data.act == "pass":
|
|
80
|
+
await show_new_cond(cbq.message, state)
|
|
81
|
+
elif callback_data.act == "refresh":
|
|
82
|
+
try:
|
|
83
|
+
await cbq.message.edit_text(await wrap_cond(cond), reply_markup=ikm)
|
|
84
|
+
except TelegramBadRequest as e:
|
|
85
|
+
if "message is not modified:" in e.message:
|
|
86
|
+
return await cbq.answer("Текст не изменился")
|
|
87
|
+
raise e
|
|
88
|
+
elif callback_data.act == "continue":
|
|
89
|
+
await (await state.get_value("cmsg")).delete()
|
|
90
|
+
await show_cond(cbq.message, cond)
|
|
91
|
+
return await cbq.answer(callback_data.act)
|
|
92
|
+
|
|
93
|
+
|
|
94
|
+
@cr.message()
|
|
177
95
|
async def unknown(msg: Message):
|
|
178
96
|
# user = await User.get(username_id=msg.from_user.id)
|
|
179
97
|
await msg.delete()
|
|
@@ -0,0 +1,118 @@
|
|
|
1
|
+
import re
|
|
2
|
+
from enum import IntEnum
|
|
3
|
+
from inspect import isclass
|
|
4
|
+
from typing import Coroutine
|
|
5
|
+
|
|
6
|
+
from aiogram.filters.callback_data import CallbackData
|
|
7
|
+
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton, ReplyKeyboardMarkup, KeyboardButton
|
|
8
|
+
from cyrtranslit import to_latin
|
|
9
|
+
from xync_schema import models
|
|
10
|
+
from xync_schema.enums import SynonymType, Boundary, Party, Slip, AbuserType, NameType
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
class SynTypeCd(CallbackData, prefix="st"):
|
|
14
|
+
typ: str
|
|
15
|
+
val: int
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
class CondCd(CallbackData, prefix="cond"):
|
|
19
|
+
act: str
|
|
20
|
+
|
|
21
|
+
|
|
22
|
+
async def wrap_cond(cond: models.Cond):
|
|
23
|
+
bnks = await models.Synonym.filter(typ=SynonymType.bank)
|
|
24
|
+
bnks = "|".join("\\b" + b.txt + ("\\b" if b.boundary & Boundary.right else "") for b in bnks)
|
|
25
|
+
ad = await models.Ad.filter(cond=cond).order_by("-updated_at").prefetch_related("ex", "pair_side").first()
|
|
26
|
+
hdr = ("SELL" if ad.pair_side.is_sell else "BUY") + f":{ad.price} [{ad.min_fiat}]"
|
|
27
|
+
if (mx := (ad.max_fiat or ad.amount)) > ad.min_fiat:
|
|
28
|
+
hdr += f"->[{mx}]"
|
|
29
|
+
txt = cond.raw_txt
|
|
30
|
+
for syn in await models.Synonym.all():
|
|
31
|
+
lb, rb = "\\b" if syn.boundary & Boundary.left else "", "\\b" if syn.boundary & Boundary.right else ""
|
|
32
|
+
if syn.typ == SynonymType.bank_side:
|
|
33
|
+
syn.txt.replace("#banks#", f"({bnks})")
|
|
34
|
+
if syn.is_re or syn.txt in txt:
|
|
35
|
+
pattern = re.compile(lb + syn.txt + rb)
|
|
36
|
+
if match := re.search(pattern, txt):
|
|
37
|
+
g = match.group()
|
|
38
|
+
val, hval = await get_val(syn.typ, syn.val)
|
|
39
|
+
val = syn.typ.name + (f'="{hval}"' if hval else "")
|
|
40
|
+
txt = re.sub(pattern, f"<code>{g}</code><tg-spoiler>[{val}]</tg-spoiler>", txt)
|
|
41
|
+
return f"<blockquote>{hdr} {ad.ex.name}</blockquote>{txt}"
|
|
42
|
+
|
|
43
|
+
|
|
44
|
+
async def cbanks(bnid: str) -> list[tuple[int, str]]:
|
|
45
|
+
beginning = to_latin(bnid[:2], lang_code="ru")
|
|
46
|
+
return await models.Pm.filter(norm__startswith=beginning, bank=True).values_list("id", "norm")
|
|
47
|
+
|
|
48
|
+
|
|
49
|
+
async def cppo(txt: str) -> list[tuple[int, str]]:
|
|
50
|
+
opts = re.findall(r"\d+", txt) or [1, 1000, 5000]
|
|
51
|
+
return [(o, str(o)) for o in opts]
|
|
52
|
+
|
|
53
|
+
|
|
54
|
+
async def contact(txt: str) -> list[tuple[int, str]]: ...
|
|
55
|
+
|
|
56
|
+
|
|
57
|
+
synopts: dict[SynonymType, list[str] | type(IntEnum) | None | Coroutine] = {
|
|
58
|
+
SynonymType.name: ["not_slavic", "slavic"],
|
|
59
|
+
SynonymType.ppo: cppo,
|
|
60
|
+
SynonymType.from_party: Party,
|
|
61
|
+
SynonymType.to_party: Party,
|
|
62
|
+
SynonymType.slip_req: Slip,
|
|
63
|
+
SynonymType.slip_send: Slip,
|
|
64
|
+
SynonymType.abuser: AbuserType,
|
|
65
|
+
SynonymType.scale: ["1", "100", "1000", "5000"],
|
|
66
|
+
SynonymType.slavic: NameType,
|
|
67
|
+
SynonymType.mtl_like: None,
|
|
68
|
+
SynonymType.bank: cbanks,
|
|
69
|
+
SynonymType.bank_side: ["except", "only"],
|
|
70
|
+
SynonymType.sbp_strict: ["no", "sbp", "card"],
|
|
71
|
+
SynonymType.contact: contact,
|
|
72
|
+
}
|
|
73
|
+
|
|
74
|
+
rkm = ReplyKeyboardMarkup(
|
|
75
|
+
keyboard=[
|
|
76
|
+
[KeyboardButton(text="ppo"), KeyboardButton(text="abuser")],
|
|
77
|
+
[KeyboardButton(text="from_party"), KeyboardButton(text="to_party")],
|
|
78
|
+
[KeyboardButton(text="slip_send"), KeyboardButton(text="slip_req")],
|
|
79
|
+
[KeyboardButton(text="name"), KeyboardButton(text="slavic")],
|
|
80
|
+
[KeyboardButton(text="scale"), KeyboardButton(text="mtl_like")],
|
|
81
|
+
[KeyboardButton(text="bank"), KeyboardButton(text="bank_side")],
|
|
82
|
+
[KeyboardButton(text="sbp_strict"), KeyboardButton(text="contact")],
|
|
83
|
+
],
|
|
84
|
+
# one_time_keyboard=True,
|
|
85
|
+
)
|
|
86
|
+
ikm = InlineKeyboardMarkup(
|
|
87
|
+
inline_keyboard=[
|
|
88
|
+
[
|
|
89
|
+
InlineKeyboardButton(text="Обновить", callback_data="cond:refresh"),
|
|
90
|
+
InlineKeyboardButton(text="Пропустить", callback_data="cond:pass"),
|
|
91
|
+
]
|
|
92
|
+
]
|
|
93
|
+
)
|
|
94
|
+
|
|
95
|
+
|
|
96
|
+
async def get_val(typ: SynonymType.__class__, val: str | int) -> tuple[SynonymType | int | bool, str]:
|
|
97
|
+
if isinstance(val, str) and val.isnumeric():
|
|
98
|
+
val = int(val)
|
|
99
|
+
if isclass(lst := synopts[typ]) and issubclass(lst, IntEnum):
|
|
100
|
+
return (v := lst(val)), v.name
|
|
101
|
+
elif isinstance(lst, list):
|
|
102
|
+
return val, lst[val]
|
|
103
|
+
elif typ == SynonymType.bank:
|
|
104
|
+
return val, (await models.Pm[val]).norm
|
|
105
|
+
return val, val
|
|
106
|
+
|
|
107
|
+
|
|
108
|
+
async def btns(typ: SynonymType.__class__, txt: str = None) -> InlineKeyboardMarkup | None:
|
|
109
|
+
if lst := synopts[typ]:
|
|
110
|
+
if isinstance(lst, list):
|
|
111
|
+
kb = [[InlineKeyboardButton(text=n, callback_data=f"st:{typ.name}:{i}")] for i, n in enumerate(lst)]
|
|
112
|
+
elif isclass(lst) and issubclass(lst, IntEnum):
|
|
113
|
+
kb = [[InlineKeyboardButton(text=i.name, callback_data=f"st:{typ.name}:{i.value}")] for i in lst]
|
|
114
|
+
else:
|
|
115
|
+
kb = [[InlineKeyboardButton(text=n, callback_data=f"st:{typ.name}:{i}")] for i, n in await lst(txt)]
|
|
116
|
+
return InlineKeyboardMarkup(inline_keyboard=kb)
|
|
117
|
+
else:
|
|
118
|
+
return lst
|
xync_bot/routers/main.py
CHANGED
|
@@ -6,7 +6,7 @@ from aiogram.exceptions import TelegramBadRequest
|
|
|
6
6
|
from aiogram.filters import CommandStart, CommandObject, ChatMemberUpdatedFilter, JOIN_TRANSITION, LEAVE_TRANSITION
|
|
7
7
|
from aiogram.filters.callback_data import CallbackData
|
|
8
8
|
from aiogram.types import (
|
|
9
|
-
User
|
|
9
|
+
User,
|
|
10
10
|
ChatMemberUpdated,
|
|
11
11
|
Message,
|
|
12
12
|
InlineKeyboardMarkup,
|
|
@@ -15,12 +15,11 @@ from aiogram.types import (
|
|
|
15
15
|
WebAppInfo,
|
|
16
16
|
)
|
|
17
17
|
from aiogram.utils.deep_linking import create_start_link
|
|
18
|
-
from xync_schema
|
|
18
|
+
from xync_schema import models
|
|
19
19
|
|
|
20
|
-
from xync_bot.routers import user_upsert
|
|
21
20
|
from xync_bot.shared import NavCallbackData
|
|
22
21
|
|
|
23
|
-
|
|
22
|
+
mr = Router()
|
|
24
23
|
|
|
25
24
|
|
|
26
25
|
class RrCallbackData(CallbackData, prefix="reg_res"): # registration response
|
|
@@ -38,19 +37,19 @@ home_btns = InlineKeyboardMarkup(
|
|
|
38
37
|
)
|
|
39
38
|
|
|
40
39
|
|
|
41
|
-
@
|
|
40
|
+
@mr.message(CommandStart(deep_link=True, deep_link_encoded=True))
|
|
42
41
|
async def start_handler(msg: Message, command: CommandObject):
|
|
43
|
-
me:
|
|
42
|
+
me: User = msg.from_user
|
|
44
43
|
ref_id: int = command.args.isnumeric() and int(command.args)
|
|
45
|
-
user = await User.get_or_none(id=me.id, blocked=False)
|
|
44
|
+
user = await models.User.get_or_none(id=me.id, blocked=False)
|
|
46
45
|
rm = None
|
|
47
46
|
logging.info(msg, {"src": "start"})
|
|
48
47
|
if user:
|
|
49
48
|
rs, rm = f"{me.full_name}, you have registered already😉", home_btns
|
|
50
|
-
elif not (ref := await User.get_or_none(id=ref_id)):
|
|
49
|
+
elif not (ref := await models.User.get_or_none(id=ref_id)):
|
|
51
50
|
rs = f"No registered user #{ref_id}😬"
|
|
52
51
|
else: # new user created
|
|
53
|
-
user, cr = await
|
|
52
|
+
user, cr = await models.User.tg2in(me, False)
|
|
54
53
|
await user.update_from_dict({"ref": ref}).save()
|
|
55
54
|
approve_btns = InlineKeyboardMarkup(
|
|
56
55
|
inline_keyboard=[
|
|
@@ -67,9 +66,9 @@ async def start_handler(msg: Message, command: CommandObject):
|
|
|
67
66
|
return await msg.answer(rs, reply_markup=rm)
|
|
68
67
|
|
|
69
68
|
|
|
70
|
-
@
|
|
69
|
+
@mr.callback_query(RrCallbackData.filter())
|
|
71
70
|
async def phrases_input_request(cb: CallbackQuery, callback_data: RrCallbackData) -> None:
|
|
72
|
-
protege = await User[callback_data.to]
|
|
71
|
+
protege = await models.User[callback_data.to]
|
|
73
72
|
if callback_data.res:
|
|
74
73
|
# protege.status = UserStatus.RESTRICTED
|
|
75
74
|
await protege.save()
|
|
@@ -84,17 +83,17 @@ async def phrases_input_request(cb: CallbackQuery, callback_data: RrCallbackData
|
|
|
84
83
|
await cb.message.edit_text(rs)
|
|
85
84
|
|
|
86
85
|
|
|
87
|
-
@
|
|
86
|
+
@mr.message(CommandStart(deep_link=True)) # attempt to reg by fake link
|
|
88
87
|
async def fraud_handler(msg: Message):
|
|
89
88
|
logging.warning(f"Start: {msg.from_user.id}. Msg: {msg}")
|
|
90
89
|
# todo: alert to admins! Fraud attempt!
|
|
91
90
|
await msg.answer("🤔")
|
|
92
91
|
|
|
93
92
|
|
|
94
|
-
@
|
|
93
|
+
@mr.message(CommandStart())
|
|
95
94
|
async def start_no_ref_handler(msg: Message):
|
|
96
95
|
me = msg.from_user
|
|
97
|
-
user, cr = await
|
|
96
|
+
user, cr = await models.User.tg2in(me, False)
|
|
98
97
|
rr = "сначала вы должны найти поручителя, и перейти по его реферальной ссылке.\nhttps://telegra.ph/XyncNet-02-13"
|
|
99
98
|
if cr: # has ref and created now
|
|
100
99
|
await msg.answer(f"Здравствуйте {me.full_name}, что бы использовать возможности нашей сети, {rr}")
|
|
@@ -104,10 +103,10 @@ async def start_no_ref_handler(msg: Message):
|
|
|
104
103
|
await msg.answer(f"{me.full_name}, не балуйтесь, вы и так уже активный участник👌🏼", reply_markup=home_btns)
|
|
105
104
|
|
|
106
105
|
|
|
107
|
-
@
|
|
106
|
+
@mr.callback_query(NavCallbackData.filter(F.to == "ref_link"))
|
|
108
107
|
async def ref_link_handler(cbq: CallbackQuery):
|
|
109
108
|
me = cbq.from_user
|
|
110
|
-
if not (u := await User.get_or_none(id=me.id, blocked=False).prefetch_related("ref")):
|
|
109
|
+
if not (u := await models.User.get_or_none(id=me.id, blocked=False).prefetch_related("ref")):
|
|
111
110
|
return await cbq.answer(f"{me.full_name}, сначала сами получите одобрение поручителя😉")
|
|
112
111
|
link = await create_start_link(cbq.bot, str(u.id), encode=True)
|
|
113
112
|
logging.info(f"Start: {me.id}. Msg: {cbq}")
|
|
@@ -119,36 +118,36 @@ async def ref_link_handler(cbq: CallbackQuery):
|
|
|
119
118
|
await cbq.answer("Wait for your protege request..")
|
|
120
119
|
|
|
121
120
|
|
|
122
|
-
@
|
|
121
|
+
@mr.my_chat_member(F.chat.type == "private") # my_chat_member is fired on add bot to any chat. filter for preventing
|
|
123
122
|
async def my_user_set_status(my_chat_member: ChatMemberUpdated):
|
|
124
123
|
logging.info({"my_chat_member": my_chat_member.model_dump(exclude_none=True)})
|
|
125
|
-
u:
|
|
124
|
+
u: User = my_chat_member.from_user
|
|
126
125
|
blocked = my_chat_member.new_chat_member.status in ("left", "kicked")
|
|
127
|
-
await
|
|
126
|
+
await models.User.tg2in(u, blocked)
|
|
128
127
|
|
|
129
128
|
|
|
130
|
-
@
|
|
129
|
+
@mr.my_chat_member()
|
|
131
130
|
async def user_set_status(my_chat_member: ChatMemberUpdated):
|
|
132
131
|
if my_chat_member.new_chat_member.user.username == "XyncNetBot": # удалена группа где бот был добавлен админом
|
|
133
|
-
if forum := await Forum.get_or_none(id=my_chat_member.chat.id):
|
|
132
|
+
if forum := await models.Forum.get_or_none(id=my_chat_member.chat.id):
|
|
134
133
|
await forum.delete()
|
|
135
134
|
res = f"I {my_chat_member.new_chat_member.status} from {my_chat_member.chat.id}:{my_chat_member.chat.title}"
|
|
136
135
|
return logging.info(res)
|
|
137
136
|
logging.info({"my_chat_member": my_chat_member.model_dump(exclude_none=True)})
|
|
138
|
-
u:
|
|
137
|
+
u: User = my_chat_member.from_user
|
|
139
138
|
blocked = my_chat_member.new_chat_member.status in ("left", "kicked")
|
|
140
139
|
if blocked:
|
|
141
|
-
if forum := await Forum.get_or_none(id=my_chat_member.chat.id, user_id=u.id):
|
|
140
|
+
if forum := await models.Forum.get_or_none(id=my_chat_member.chat.id, user_id=u.id):
|
|
142
141
|
if forum.joined:
|
|
143
142
|
forum.joined = False
|
|
144
143
|
await forum.save()
|
|
145
|
-
await
|
|
144
|
+
return await models.User.tg2in(u, blocked)
|
|
146
145
|
|
|
147
146
|
|
|
148
|
-
@
|
|
147
|
+
@mr.chat_member(ChatMemberUpdatedFilter(LEAVE_TRANSITION)) # юзер покинул группу Ордеров
|
|
149
148
|
async def on_user_leave(member: ChatMemberUpdated):
|
|
150
149
|
logging.info({"user_leave": member.model_dump(exclude_none=True)})
|
|
151
|
-
if forum := await Forum[member.chat.id]:
|
|
150
|
+
if forum := await models.Forum[member.chat.id]:
|
|
152
151
|
if forum.joined:
|
|
153
152
|
forum.joined = False
|
|
154
153
|
await forum.save()
|
|
@@ -156,10 +155,10 @@ async def on_user_leave(member: ChatMemberUpdated):
|
|
|
156
155
|
return await member.bot.send_message(member.new_chat_member.user.id, resp)
|
|
157
156
|
|
|
158
157
|
|
|
159
|
-
@
|
|
158
|
+
@mr.chat_member(ChatMemberUpdatedFilter(JOIN_TRANSITION)) # Юзер добавился в группу Ордеров
|
|
160
159
|
async def on_user_join(member: ChatMemberUpdated, app_url: str):
|
|
161
160
|
logging.info({"user_join": member.model_dump(exclude_none=True)})
|
|
162
|
-
if forum := await Forum.get_or_none(id=member.chat.id):
|
|
161
|
+
if forum := await models.Forum.get_or_none(id=member.chat.id):
|
|
163
162
|
if not forum.joined:
|
|
164
163
|
forum.joined = True
|
|
165
164
|
await forum.save()
|
|
@@ -170,26 +169,26 @@ async def on_user_join(member: ChatMemberUpdated, app_url: str):
|
|
|
170
169
|
return await member.bot.send_message(member.new_chat_member.user.id, resp, reply_markup=rm)
|
|
171
170
|
|
|
172
171
|
|
|
173
|
-
@
|
|
172
|
+
@mr.message(F.is_topic_message)
|
|
174
173
|
async def order_msg(msg: Message):
|
|
175
|
-
sender = await User[msg.from_user.id]
|
|
174
|
+
sender = await models.User[msg.from_user.id]
|
|
176
175
|
cid = msg.chat.shifted_id
|
|
177
176
|
assert sender.forum == cid, "sender is not client"
|
|
178
|
-
if order := await Order.get_or_none(taker__user_id=sender.id, taker_topic=msg.message_thread_id):
|
|
177
|
+
if order := await models.Order.get_or_none(taker__user_id=sender.id, taker_topic=msg.message_thread_id):
|
|
179
178
|
is_taker = True
|
|
180
|
-
elif order := await Order.get_or_none(ad__agent__user_id=sender.id, maker_topic=msg.message_thread_id):
|
|
179
|
+
elif order := await models.Order.get_or_none(ad__agent__user_id=sender.id, maker_topic=msg.message_thread_id):
|
|
181
180
|
is_taker = False
|
|
182
181
|
else:
|
|
183
182
|
return await msg.answer("No such order")
|
|
184
183
|
# raise Exception("No such order")
|
|
185
|
-
receiver: User = await (order.ad.maker.user if is_taker else order.taker.user)
|
|
184
|
+
receiver: models.User = await (order.ad.maker.user if is_taker else order.taker.user)
|
|
186
185
|
rcv_topic = order.taker_topic if is_taker else order.maker_topic
|
|
187
|
-
await Msg.create(tgid=msg.message_id, txt=msg.text, order_id=order.id, receiver=receiver)
|
|
186
|
+
await models.Msg.create(tgid=msg.message_id, txt=msg.text, order_id=order.id, receiver=receiver)
|
|
188
187
|
logging.info(msg, {"src": "order_msg"})
|
|
189
188
|
return await msg.send_copy(receiver.forum, message_thread_id=rcv_topic)
|
|
190
189
|
|
|
191
190
|
|
|
192
|
-
@
|
|
191
|
+
@mr.message(
|
|
193
192
|
F.content_type.not_in(
|
|
194
193
|
{
|
|
195
194
|
ContentType.NEW_CHAT_MEMBERS,
|
|
@@ -211,7 +210,7 @@ async def del_cbq(msg: Message):
|
|
|
211
210
|
logging.error({"NOT_DELETED": msg.model_dump(exclude_none=True)})
|
|
212
211
|
|
|
213
212
|
|
|
214
|
-
@
|
|
213
|
+
@mr.message()
|
|
215
214
|
async def all_rest(msg: Message):
|
|
216
215
|
logging.warning(
|
|
217
216
|
{
|