spawnrys/flanabot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add BtcOffersBot.stop_all_btc_offers_notification

Browse Source
This commit is contained in:
AlberLC
2025-04-17 23:46:51 +02:00
parent ad2434ab67
commit 647b9288aa
1 changed files with 24 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
flanabot/bots/btc_offers_bot.py
View File

@@ -125,6 +125,12 @@ class BtcOffersBot(MultiBot, ABC):
self.register(self._on_stop_btc_offers_notification, keywords=(multibot_constants.KEYWORDS['stop'], constants.KEYWORDS['money']))
self.register(self._on_stop_btc_offers_notification, keywords=(multibot_constants.KEYWORDS['stop'], constants.KEYWORDS['notify']))
def _find_chats_to_notify(self) -> list[Chat]:
return self.Chat.find({'platform': self.platform.value, 'btc_offers_query': {'$exists': True, '$ne': {}}})
def _is_websocket_connected(self) -> bool:
return self._websocket and self._websocket.state in {websockets.State.CONNECTING, websockets.State.OPEN}
async def _send_offers(self, offers: list[dict], chat: Chat, notifications_disabled=False):
offers_parts = []
for i, offer in enumerate(offers, start=1):
@@ -195,7 +201,7 @@ class BtcOffersBot(MultiBot, ABC):
try:
data = json.loads(await self._websocket.recv())
except websockets.ConnectionClosed:
await self.start_saved_btc_offers_notifications()
await self.start_all_btc_offers_notifications()
else:
break
@@ -245,7 +251,7 @@ class BtcOffersBot(MultiBot, ABC):
async def _on_ready(self):
await super()._on_ready()
asyncio.create_task(self.start_saved_btc_offers_notifications())
asyncio.create_task(self.start_all_btc_offers_notifications())
async def _on_stop_btc_offers_notification(self, message: Message):
previous_btc_offers_query = message.chat.btc_offers_query
@@ -260,9 +266,19 @@ class BtcOffersBot(MultiBot, ABC):
# -------------------------------------------------------- #
# -------------------- PUBLIC METHODS -------------------- #
# -------------------------------------------------------- #
async def start_all_btc_offers_notifications(self):
if chats := self._find_chats_to_notify():
for chat in chats:
chat = await self.get_chat(chat.id)
chat.pull_from_database(overwrite_fields=('_id', 'btc_offers_query'))
await self.start_btc_offers_notification(chat, chat.btc_offers_query)
elif self._notification_task and not self._notification_task.done():
self._notification_task.cancel()
await asyncio.sleep(0)
async def start_btc_offers_notification(self, chat: Chat, query: dict[str, float]):
async with self._websocket_lock:
if not self._websocket or self._websocket.state in {websockets.State.CLOSING, websockets.State.CLOSED}:
if not self._is_websocket_connected():
while True:
try:
self._websocket = await websockets.connect(f'ws://{self._api_endpoint}')
@@ -278,23 +294,13 @@ class BtcOffersBot(MultiBot, ABC):
chat.save()
await self._websocket.send(json.dumps({'action': 'start', 'chat_id': chat.id, 'query': query}))
async def start_saved_btc_offers_notifications(self):
if chats := self.Chat.find({
'platform': self.platform.value,
'btc_offers_query': {'$exists': True, '$ne': {}}
}):
for chat in chats:
chat = await self.get_chat(chat.id)
chat.pull_from_database(overwrite_fields=('_id', 'btc_offers_query'))
await self.start_btc_offers_notification(chat, chat.btc_offers_query)
elif self._notification_task and not self._notification_task.done():
self._notification_task.cancel()
await asyncio.sleep(0)
async def stop_all_btc_offers_notification(self):
for chat in self._find_chats_to_notify():
await self.stop_btc_offers_notification(chat)
async def stop_btc_offers_notification(self, chat: Chat):
if not self._websocket:
return
if self._is_websocket_connected():
await self._websocket.send(json.dumps({'action': 'stop', 'chat_id': chat.id}))
chat.btc_offers_query = {}
chat.save(pull_exclude_fields=('btc_offers_query',))