mirror of
https://github.com/insertapp/mmolbbot.git
synced 2025-08-01 18:53:33 +00:00
feat: statistical live update
This commit is contained in:
parent
3d5f2e3de4
commit
c138a608d6
GPG key ID:
A70775C389ACF105
1 changed files with 104 additions and 3 deletions
|
|
@ -10,6 +10,7 @@ import itertools
|
|||
import aiosqlite as sqlite3
|
||||
from nextcord.ext import commands, application_checks, tasks
|
||||
from nextcord import TextInputStyle, IntegrationType
|
||||
import re
|
||||
import timeit
|
||||
|
||||
timings = []
|
||||
|
|
@ -145,6 +146,65 @@ async def classiclivegameworker(self,serverid,userid,channelid,gameid,offset):
|
|||
await self.bot.db.commit()
|
||||
await warning.send(f"Deleted {lastserverid} from the database due to 403 error")
|
||||
|
||||
async def statisticallivegameworker(self,serverid,userid,channelid,gameid,offset):
|
||||
begin = timeit.default_timer()
|
||||
lastserverid = serverid
|
||||
try:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
channel = self.bot.get_channel(channelid)
|
||||
data = await session.get(f"https://mmolb.com/api/game/{gameid}/live?after={offset}")
|
||||
data = await data.json()
|
||||
basedata = await session.get(f"https://mmolb.com/api/game/{gameid}")
|
||||
basedata = await basedata.json()
|
||||
if basedata["State"] == "Complete":
|
||||
await self.bot.db.execute(f"""
|
||||
DELETE from liveupdate WHERE channelid = {channelid} AND gameid = '{gameid}'
|
||||
""")
|
||||
if len(data["entries"]) > 0:
|
||||
finalstr = ""
|
||||
offsetadd = 0
|
||||
maysend = False
|
||||
for i in data["entries"]:
|
||||
offsetadd += 1
|
||||
if i["event"] == "InningEnd":
|
||||
finalstr += f"\n{i['message']} | {basedata["AwayTeamName"]} {basedata["AwayTeamEmoji"]} {data["entries"][-1]["away_score"]} vs {basedata["HomeTeamName"]} {basedata["HomeTeamEmoji"]} {data["entries"][-1]["home_score"]}"
|
||||
maysend = True
|
||||
for r in re.findall(r'[A-Za-z éö1-9]*out[A-Za-z éö1-9]*',i['message'],re.IGNORECASE):
|
||||
finalstr += f"\n🔴 {r}"
|
||||
maysend = True
|
||||
|
||||
if "scores" in i['message'] or "homers" in i['message']:
|
||||
if data["entries"][-1]["inning_side"] == 1:
|
||||
finalstr += f"\n{basedata["AwayTeamName"]} {basedata["AwayTeamEmoji"]} {data["entries"][-1]["away_score"]} vs {basedata["HomeTeamName"]} {basedata["HomeTeamEmoji"]} **{data["entries"][-1]["home_score"]}**"
|
||||
else:
|
||||
finalstr += f"\n{basedata["AwayTeamName"]} {basedata["AwayTeamEmoji"]} **{data["entries"][-1]["away_score"]}** vs {basedata["HomeTeamName"]} {basedata["HomeTeamEmoji"]} {data["entries"][-1]["home_score"]}"
|
||||
for r in re.findall(r'(?<=<strong>)([A-Za-z, !.]+)(?=<\/strong>)',i['message'],re.IGNORECASE):
|
||||
finalstr += f"\n🟢 {r}"
|
||||
maysend = True
|
||||
for r in re.findall(r'(?:[A-Za-z éö]*)(?:walks?|steals?|error|signles?|doubles?|triples?|to (?:first|second|third) base)(?:[A-Za-z éö]*)',i['message'],re.IGNORECASE):
|
||||
finalstr += f"\n👟 {r}"
|
||||
maysend = True
|
||||
if i["event"] == "Recordkeeping":
|
||||
await self.bot.db.execute(f"""
|
||||
DELETE from liveupdate WHERE channelid = {channelid} AND gameid = '{gameid}'
|
||||
""")
|
||||
maysend = True
|
||||
finalstr += f"{i['message'].replace("<strong>", "**").replace("</strong>", "**")}\n"
|
||||
if maysend:
|
||||
await channel.send(finalstr)
|
||||
await self.bot.db.execute(f"""
|
||||
UPDATE liveupdate set offset = {offset+offsetadd} WHERE channelid = '{channelid}' AND gameid = '{gameid}' AND classic = 2
|
||||
""")
|
||||
except Exception as e:
|
||||
warning = self.bot.get_channel(1365478368555827270)
|
||||
await warning.send(e)
|
||||
except nextcord.Forbidden:
|
||||
warning = self.bot.get_channel(1365478368555827270)
|
||||
await self.bot.db.execute("DELETE from teamsubscriptions WHERE serverid = ?", (lastserverid,))
|
||||
await self.bot.db.execute("DELETE from liveupdate WHERE serverid = ?", (lastserverid,))
|
||||
await self.bot.db.execute("DELETE from spotlightsubscriptions WHERE serverid = ?", (lastserverid,))
|
||||
await self.bot.db.commit()
|
||||
await warning.send(f"Deleted {lastserverid} from the database due to 403 error")
|
||||
|
||||
async def spotlightsubscriptionworker(self,serverid,channelid,classic,gameid,data):
|
||||
try:
|
||||
|
|
@ -152,6 +212,8 @@ async def spotlightsubscriptionworker(self,serverid,channelid,classic,gameid,dat
|
|||
test = await check.fetchone()
|
||||
if test is None:
|
||||
channel = self.bot.get_channel(channelid)
|
||||
if channel is None:
|
||||
return
|
||||
if data["State"] == "Complete":
|
||||
return
|
||||
check = await self.bot.db.execute("SELECT serverid,userid,channelid,messageid,gameid,offset FROM liveupdate WHERE channelid = ? AND gameid = ? AND classic = ?", (channelid,gameid,classic))
|
||||
|
|
@ -164,6 +226,7 @@ async def spotlightsubscriptionworker(self,serverid,channelid,classic,gameid,dat
|
|||
({channel.guild.id}, {self.bot.application_id}, {channelid}, {message.id}, "{gameid}", {len(data["EventLog"])}, 0)
|
||||
""")
|
||||
else:
|
||||
await channel.send(content=f"{data["AwayTeamName"]} {data["AwayTeamEmoji"]} vs {data["HomeTeamName"]} {data["HomeTeamEmoji"]}\n> PLAY BALL")
|
||||
await self.bot.db.execute(f"""
|
||||
INSERT INTO liveupdate VALUES
|
||||
({channel.guild.id}, {self.bot.application_id}, {channelid}, NULL, "{gameid}", {len(data["EventLog"])}, 1)
|
||||
|
|
@ -205,11 +268,18 @@ async def teamsubscriptionworker(self,serverid,channelid,teamid,classic):
|
|||
INSERT INTO liveupdate VALUES
|
||||
({channel.guild.id}, {self.bot.application_id}, {channelid}, {message.id}, "{gameid}", {len(data["EventLog"])}, 0)
|
||||
""")
|
||||
else:
|
||||
elif classic == 1:
|
||||
await channel.send(content=f"{data["AwayTeamName"]} {data["AwayTeamEmoji"]} vs {data["HomeTeamName"]} {data["HomeTeamEmoji"]}\n> PLAY BALL")
|
||||
await self.bot.db.execute(f"""
|
||||
INSERT INTO liveupdate VALUES
|
||||
({channel.guild.id}, {self.bot.application_id}, {channelid}, NULL, "{gameid}", {len(data["EventLog"])}, 1)
|
||||
""")
|
||||
else:
|
||||
await channel.send(content=f"{data["AwayTeamName"]} {data["AwayTeamEmoji"]} vs {data["HomeTeamName"]} {data["HomeTeamEmoji"]}\n🤖 PLAY BALL")
|
||||
await self.bot.db.execute(f"""
|
||||
INSERT INTO liveupdate VALUES
|
||||
({channel.guild.id}, {self.bot.application_id}, {channelid}, NULL, "{gameid}", {len(data["EventLog"])}, 2)
|
||||
""")
|
||||
except KeyError:
|
||||
return
|
||||
except sqlite3.OperationalError:
|
||||
|
|
@ -239,10 +309,10 @@ async def feedsubscriptionworker(self,serverid,channelid,teamid,offset):
|
|||
finalstr = ""
|
||||
for index in itertools.islice(data, offset, len(data)):
|
||||
finalstr += f"\n{index['emoji']} Season {index['season']} Day {index['day']}: {index['text']}"
|
||||
await channel.send(finalstr)
|
||||
await self.bot.db.execute(f"""
|
||||
UPDATE feedsubscriptions set offset = {len(data)} WHERE channelid = '{channelid}' AND teamid = '{teamid}'
|
||||
""")
|
||||
await channel.send(finalstr)
|
||||
except Exception as e:
|
||||
warning = self.bot.get_channel(1365478368555827270)
|
||||
await warning.send(e)
|
||||
|
|
@ -425,6 +495,32 @@ class liveupdate(commands.Cog):
|
|||
thanksdiscord = closestteam[:20]
|
||||
await interaction.response.send_autocomplete(thanksdiscord)
|
||||
|
||||
@subscribe.subcommand(
|
||||
name="statistical",
|
||||
description="Subscribe to a team using a modified version of classic, will spam the channel",
|
||||
)
|
||||
async def statisitcalteamsubscribe(self, interaction: nextcord.Interaction, team: str):
|
||||
if team not in self.bot.teams_dict:
|
||||
await interaction.response.send_message("Invalid Team!", ephemeral=True)
|
||||
await interaction.response.defer()
|
||||
teamid = self.bot.teams_dict[team]
|
||||
await interaction.edit_original_message(content="This channel is now subscribed to updates")
|
||||
await self.bot.db.execute(f"""
|
||||
INSERT INTO teamsubscriptions VALUES
|
||||
({interaction.guild_id}, {interaction.channel_id}, "{teamid}", 2)
|
||||
""")
|
||||
await self.bot.db.commit()
|
||||
|
||||
@statisitcalteamsubscribe.on_autocomplete("team")
|
||||
async def statisitcalteamsubscribeac(self, interaction: nextcord.Interaction, team: str):
|
||||
if not team:
|
||||
thanksdiscord = self.bot.teams_list[:20]
|
||||
await interaction.response.send_autocomplete(thanksdiscord)
|
||||
return
|
||||
closestteam = [name for name in self.bot.teams_list if name.lower().startswith(team.lower())]
|
||||
thanksdiscord = closestteam[:20]
|
||||
await interaction.response.send_autocomplete(thanksdiscord)
|
||||
|
||||
@subscribe.subcommand(
|
||||
name="feed",
|
||||
description="Subscribe to a team's feed",
|
||||
|
|
@ -599,6 +695,9 @@ class liveupdate(commands.Cog):
|
|||
res = await self.bot.db.execute("SELECT serverid,userid,channelid,gameid,offset FROM liveupdate WHERE classic = 1")
|
||||
res = await res.fetchall()
|
||||
worklist = worklist + [classiclivegameworker(self,serverid,userid,channelid,gameid,offset) for [serverid,userid,channelid,gameid,offset] in res]
|
||||
res = await self.bot.db.execute("SELECT serverid,userid,channelid,gameid,offset FROM liveupdate WHERE classic = 2")
|
||||
res = await res.fetchall()
|
||||
worklist = worklist + [statisticallivegameworker(self,serverid,userid,channelid,gameid,offset) for [serverid,userid,channelid,gameid,offset] in res]
|
||||
await asyncio.gather(*worklist,return_exceptions=True)
|
||||
await self.bot.db.commit()
|
||||
game = nextcord.Game(f"Broadcasting {len(worklist)} games")
|
||||
|
|
@ -633,8 +732,10 @@ class liveupdate(commands.Cog):
|
|||
game = await game.json()
|
||||
gameid = game["game_id"]
|
||||
except KeyError:
|
||||
game = await session.get("https://mmolb.com/api/event-games")
|
||||
game = await session.get("https://mmolb.com/api/superstar-games")
|
||||
game = await game.json()
|
||||
if game["games"][-1]['status'] == "Scheduled": #The game has been posted to the api but doesn't exist
|
||||
return
|
||||
gameid = game["games"][-1]["game_id"]
|
||||
data = await session.get(f"https://mmolb.com/api/game/{gameid}")
|
||||
data = await data.json()
|
||||
|
|
|
|||
Loading…
Reference in a new issue