codey/discord-havoc
1
0
Fork 0
Code Issues Pull Requests Actions Projects Wiki Activity

misc/migration related

Browse Source
This commit is contained in:
codey
2025-06-08 08:53:27 -04:00
parent 046ad4f94a
commit 75addd629c
6 changed files with 113 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
util/misc_util.py
View File

@ -21,14 +21,14 @@ class Util:
self.URL_INSULTAPI: str = "https://insult.mattbas.org/api/insult"
self.COMPLIMENT_GENERATOR = ComplimentGenerator()
self.dbs: dict[str, str | LiteralString] = {
"whisky": os.path.join("/mnt/data/share", "sqlite_dbs", "whiskey.db"),
"drinks": os.path.join("/mnt/data/share", "sqlite_dbs", "cocktails.db"),
"strains": os.path.join("/mnt/data/share", "sqlite_dbs", "strains.db"),
"qajoke": os.path.join("/mnt/data/share", "sqlite_dbs", "qajoke.db"),
"rjokes": os.path.join("/mnt/data/share", "sqlite_dbs", "rjokes.db"),
"randmsg": os.path.join("/mnt/data/share", "sqlite_dbs", "randmsg.db"),
"stats": os.path.join("/mnt/data/share", "sqlite_dbs", "havoc_stats.db"),
"cookies": os.path.join("/mnt/data/share", "sqlite_dbs", "cookies.db"),
"whisky": os.path.join("/usr/local/share", "sqlite_dbs", "whiskey.db"),
"drinks": os.path.join("/usr/local/share", "sqlite_dbs", "cocktails.db"),
"strains": os.path.join("/usr/local/share", "sqlite_dbs", "strains.db"),
"qajoke": os.path.join("/usr/local/share", "sqlite_dbs", "qajoke.db"),
"rjokes": os.path.join("/usr/local/share", "sqlite_dbs", "rjokes.db"),
"randmsg": os.path.join("/usr/local/share", "sqlite_dbs", "randmsg.db"),
"stats": os.path.join("/usr/local/share", "sqlite_dbs", "havoc_stats.db"),
"cookies": os.path.join("/usr/local/share", "sqlite_dbs", "cookies.db"),
}
self.COFFEES: list = [
"a cup of french-pressed coffee",
@ -165,7 +165,7 @@ class Util:
query, (counter,) if counter else None
) as db_cursor:
result = await db_cursor.fetchone()
return result
return dict(result)
async def get_stats_embed(self) -> Optional[Embed]:
"""
@ -207,9 +207,7 @@ class Util:
f"UPDATE stats SET {counter} = {counter} + 1"
) as db_cursor:
if db_cursor.rowcount < 0:
logging.critical(
"[increment_counter] Fail! %s", db_cursor.rowcount
)
logging.critical("[increment_counter] Fail! %s", db_cursor.rowcount)
return False
await db_conn.commit()
return True
@ -547,6 +545,26 @@ class Util:
return (days, hours, minutes, seconds, ms, us)
def get_days_to_halloween(self) -> Optional[tuple]:
"""
Get # of Days until Halloween
Returns:
Optional[tuple]
"""
today: datetime.datetime = datetime.datetime.now(tz=pytz.UTC)
halloween: datetime.datetime = datetime.datetime(
year=today.year,
month=10,
day=31,
tzinfo=pytz.UTC,
)
td: datetime.timedelta = halloween - today
days, hours, minutes, seconds, us, ms = self.tdTuple(td)
return (days, hours, minutes, seconds, ms, us)
async def get_randmsg(self) -> Optional[str]:
"""
Get Random Message from randmsg.db