eriks/erepublik-discord_bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Eriks K 2021-07-17 12:04:47 +03:00
parent 9f23253232
commit 00c3959273
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
discord_bot.py
View File

@ -1,5 +1,6 @@
import asyncio import asyncio
import datetime import datetime
import json
import logging import logging
import os import os
import sys import sys
@ -27,6 +28,8 @@ fh.setLevel(logging.DEBUG)
logger.addHandler(fh) logger.addHandler(fh)
keep_fds = [fh.stream.fileno()] keep_fds = [fh.stream.fileno()]
os.makedirs('debug', exist_ok=True)
pidfile = "pid" pidfile = "pid"
with open(pidfile, 'w') as f: with open(pidfile, 'w') as f:
f.write(str(os.getpid())) f.write(str(os.getpid()))
@ -49,7 +52,7 @@ FLAGS = {1: 'flag_ro', 9: 'flag_br', 10: 'flag_it', 11: 'flag_fr', 12: 'flag_de'
82: 'flag_cy', 83: 'flag_by', 84: 'flag_nz', 164: 'flag_sa', 165: 'flag_eg', 166: 'flag_ae', 167: 'flag_al', 82: 'flag_cy', 83: 'flag_by', 84: 'flag_nz', 164: 'flag_sa', 165: 'flag_eg', 166: 'flag_ae', 167: 'flag_al',
168: 'flag_ge', 169: 'flag_am', 170: 'flag_ng', 171: 'flag_cu'} 168: 'flag_ge', 169: 'flag_am', 170: 'flag_ng', 171: 'flag_cu'}
MENTION_MAPPING = {1: "@D1", 2: "@D2", 3: "@D3", 4: "@D4", 11: "@AIR"} MENTION_MAPPING = {1: "D1", 2: "D2", 3: "D3", 4: "D4", 11: "Air"}
__last_battle_response = None __last_battle_response = None
__last_battle_update_timestamp = 0 __last_battle_update_timestamp = 0
@ -100,6 +103,8 @@ class MyClient(discord.Client):
async def report_epics(self): async def report_epics(self):
await self.wait_until_ready() await self.wait_until_ready()
roles = [role for role in self.get_guild(300297668553605131).roles if role.name in MENTION_MAPPING.values()]
role_mapping = {role.name: role.mention for role in roles}
while not self.is_closed(): while not self.is_closed():
try: try:
r = get_battle_page() r = get_battle_page()
@ -109,11 +114,13 @@ class MyClient(discord.Client):
continue continue
for bid, battle in r.get('battles', {}).items(): for bid, battle in r.get('battles', {}).items():
for div in battle.get('div', {}).values(): for div in battle.get('div', {}).values():
if div.get('epic') and not DB.get_epic(div.get('div')): if div.get('epic') and not DB.get_epic(div.get('id')):
with open(f'debug/{self.timestamp}.json', 'wb') as f:
json.dump(r, f)
await self.get_channel(DEFAULT_CHANNEL_ID).send( await self.get_channel(DEFAULT_CHANNEL_ID).send(
f"<@{MENTION_MAPPING[div['div']]}> Epic battle! Round time {s_to_human(self.timestamp - battle['start'])}\n" f"{role_mapping[MENTION_MAPPING[div['div']]]} Epic battle! Round time {s_to_human(self.timestamp - battle['start'])}\n"
f"https://www.erepublik.com/en/military/battlefield/{battle['id']}") f"https://www.erepublik.com/en/military/battlefield/{battle['id']}")
DB.add_epic(div.get('div')) DB.add_epic(div.get('id'))
sleep_seconds = r.get('last_updated') + 60 - self.timestamp sleep_seconds = r.get('last_updated') + 60 - self.timestamp
await asyncio.sleep(sleep_seconds if sleep_seconds > 0 else 0) await asyncio.sleep(sleep_seconds if sleep_seconds > 0 else 0)