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

Update BtcOffersBot (notifications in dollars and premiums)

Browse Source
This commit is contained in:
AlberLC
2025-04-16 20:45:23 +02:00
parent aac5fe951f
commit d977d5d882
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
flanabot/bots/btc_offers_bot.py
View File

@@ -186,8 +186,8 @@ class BtcOffersBot(MultiBot, ABC):
while True:
data = json.loads(await self._websocket.recv())
chat = await self.get_chat(data['chat_id'])
chat.btc_offers_max_eur = None
chat.save(pull_exclude_fields=('btc_offers_max_eur',))
chat.btc_offers_query = {}
chat.save(pull_exclude_fields=('btc_offers_query',))
await self._send_offers(data['offers'], chat, notifications_disabled=True)
# ---------------------------------------------- #
@@ -223,7 +223,7 @@ class BtcOffersBot(MultiBot, ABC):
response_text = f"✅ ¡Perfecto! Te avisaré cuando existan ofertas con una prima del {query['max_premium']:.2f} % o menor."
await self.send(response_text, message)
await self.start_btc_offers_notification(message.chat, max_price_eur)
await self.start_btc_offers_notification(message.chat, query)
async def _on_ready(self):
await super()._on_ready()
@@ -237,12 +237,12 @@ class BtcOffersBot(MultiBot, ABC):
await self.start_btc_offers_notification(chat, chat.btc_offers_max_eur)
async def _on_stop_btc_offers_notification(self, message: Message):
previous_btc_offers_max_eur = message.chat.btc_offers_max_eur
previous_btc_offers_query = message.chat.btc_offers_query
await self.stop_btc_offers_notification(message.chat)
if previous_btc_offers_max_eur:
await self.send('🛑 Los avisos de ofertas de BTC se han desactivado.', message)
if previous_btc_offers_query:
await self.send('🛑 Los avisos de ofertas de BTC se han eliminado.', message)
else:
await self.send('🤔 No existía ningún aviso de ofertas de BTC configurado.', message)
@@ -254,7 +254,7 @@ class BtcOffersBot(MultiBot, ABC):
self._websocket = await client.connect(f'ws://{self._api_endpoint}')
self._notification_task = asyncio.create_task(self._wait_btc_offers_notification())
chat.btc_offers_max_eur = max_price_eur
chat.btc_offers_query = query
chat.save()
await self._websocket.send(json.dumps({'action': 'start', 'chat_id': chat.id, 'max_price_eur': max_price_eur}))
@@ -263,5 +263,5 @@ class BtcOffersBot(MultiBot, ABC):
return
await self._websocket.send(json.dumps({'action': 'stop', 'chat_id': chat.id}))
chat.btc_offers_max_eur = None
chat.save(pull_exclude_fields=('btc_offers_max_eur',))
chat.btc_offers_query = {}
chat.save(pull_exclude_fields=('btc_offers_query',))

2
flanabot/models/chat.py
View File

@@ -16,7 +16,7 @@ class Chat(MultiBotChat):
'scraping_delete_original': True,
'ubereats': False
})
btc_offers_max_eur: float | None = None
btc_offers_query: dict[str, float] = field(default_factory=lambda: {})
ubereats: dict = field(default_factory=lambda: {
'cookies': [],
'last_codes': [],