dev@locazia: fix wrappers
This commit is contained in:
1 parent
fafaef8915
commit
dfb2ac7106
5 files changed
+40
-25
No files matched your search
+1
-1
@@ -28,7 +28,7 @@ handler3.setFormatter(
|
||||
)
|
||||
logger.addHandler(handler3)
|
||||
|
||||
IGNORED_ISSUERS = os.getenv('IGNORED_ISSUERS', '').split(',')
|
||||
IGNORED_ISSUERS = [v for v in os.getenv('IGNORED_ISSUERS', '').split(',') if v]
|
||||
|
||||
|
||||
def make_log(issuer, message, *args, level='INFO', **kwargs):
|
||||
|
||||
@@ -27,35 +27,37 @@ class Wrapped_CBotChat:
|
||||
return "Bot"
|
||||
|
||||
async def return_result(self, result, message_type='common'):
|
||||
if message_type == 'common':
|
||||
ci = 0
|
||||
for oc_msg in self.db_session.query(KnownTelegramMessage).filter(
|
||||
KnownTelegramMessage.type == 'common',
|
||||
KnownTelegramMessage.chat_id == self._chat_id,
|
||||
).all():
|
||||
await self.delete_message(oc_msg.message_id)
|
||||
ci += 1
|
||||
if self.db_session:
|
||||
if message_type == 'common':
|
||||
ci = 0
|
||||
for oc_msg in self.db_session.query(KnownTelegramMessage).filter(
|
||||
KnownTelegramMessage.type == 'common',
|
||||
KnownTelegramMessage.chat_id == self._chat_id,
|
||||
).all():
|
||||
await self.delete_message(oc_msg.message_id)
|
||||
ci += 1
|
||||
|
||||
make_log(self, f"Deleted {ci} old messages", level='debug')
|
||||
make_log(self, f"Deleted {ci} old messages", level='debug')
|
||||
|
||||
if isinstance(result, types.Message):
|
||||
message_id = getattr(result, 'message_id', None)
|
||||
assert message_id, "No message_id"
|
||||
self.db_session.add(
|
||||
KnownTelegramMessage(
|
||||
type=message_type,
|
||||
chat_id=self._chat_id,
|
||||
message_id=message_id,
|
||||
created=datetime.now()
|
||||
if isinstance(result, types.Message):
|
||||
message_id = getattr(result, 'message_id', None)
|
||||
assert message_id, "No message_id"
|
||||
self.db_session.add(
|
||||
KnownTelegramMessage(
|
||||
type=message_type,
|
||||
chat_id=self._chat_id,
|
||||
message_id=message_id,
|
||||
created=datetime.now()
|
||||
)
|
||||
)
|
||||
)
|
||||
self.db_session.commit()
|
||||
else:
|
||||
make_log(self, f"Unknown result type: {type(result)}", level='warning')
|
||||
self.db_session.commit()
|
||||
else:
|
||||
make_log(self, f"Unknown result type: {type(result)}", level='warning')
|
||||
|
||||
return result
|
||||
|
||||
async def send_message(self, text: str, message_type='common', **kwargs):
|
||||
assert self._chat_id, "No chat_id"
|
||||
try:
|
||||
make_log(self, f"Send message to {self._chat_id}. Text len: {len(text)}", level='debug')
|
||||
r = await self._bot.send_message(
|
||||
@@ -71,6 +73,7 @@ class Wrapped_CBotChat:
|
||||
return None
|
||||
|
||||
async def edit_message(self, message_id, text, **kwargs):
|
||||
assert self._chat_id, "No chat_id"
|
||||
try:
|
||||
make_log(self, f"Edit message {self._chat_id}/{message_id}. Text len: {len(text)}", level='debug')
|
||||
return await self._bot.edit_message_text(
|
||||
@@ -90,6 +93,7 @@ class Wrapped_CBotChat:
|
||||
return None
|
||||
|
||||
async def delete_message(self, message_id):
|
||||
assert self._chat_id, "No chat_id"
|
||||
try:
|
||||
make_log(self, f"Delete message {self._chat_id}/{message_id}", level='debug')
|
||||
return await self._bot.delete_message(
|
||||
@@ -101,6 +105,7 @@ class Wrapped_CBotChat:
|
||||
return None
|
||||
|
||||
async def send_photo(self, file_id, message_type='common', **kwargs):
|
||||
assert self._chat_id, "No chat_id"
|
||||
try:
|
||||
make_log(self, f"Send photo to {self._chat_id}. File: {file_id}", level='debug')
|
||||
r = await self._bot.send_photo(
|
||||
@@ -114,6 +119,7 @@ class Wrapped_CBotChat:
|
||||
return None
|
||||
|
||||
async def send_document(self, file_id, message_type='common', **kwargs):
|
||||
assert self._chat_id, "No chat_id"
|
||||
try:
|
||||
make_log(self, f"Send document to {self._chat_id}. File: {file_id}", level='debug')
|
||||
r = await self._bot.send_document(
|
||||
@@ -127,6 +133,7 @@ class Wrapped_CBotChat:
|
||||
return None
|
||||
|
||||
async def send_video(self, file_id, message_type='common', **kwargs):
|
||||
assert self._chat_id, "No chat_id"
|
||||
try:
|
||||
make_log(self, f"Send video to {self._chat_id}. File: {file_id}", level='debug')
|
||||
r = await self._bot.send_video(
|
||||
@@ -140,6 +147,7 @@ class Wrapped_CBotChat:
|
||||
return None
|
||||
|
||||
async def copy_message(self, from_chat_id, message_id, message_type='common', **kwargs):
|
||||
assert self._chat_id, "No chat_id"
|
||||
try:
|
||||
make_log(self, f"Copy message from {from_chat_id}/{message_id} to {self._chat_id}", level='debug')
|
||||
r = await self._bot.copy_message(
|
||||
@@ -154,6 +162,7 @@ class Wrapped_CBotChat:
|
||||
return None
|
||||
|
||||
async def forward_message(self, from_chat_id, message_id, message_type='common', **kwargs):
|
||||
assert self._chat_id, "No chat_id"
|
||||
try:
|
||||
make_log(self, f"Forward message from {from_chat_id}/{message_id} to {self._chat_id}", level='debug')
|
||||
r = await self._bot.forward_message(
|
||||
|
||||
Reference in new issue
Block a user