sigonasr2/miyu-bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add relative event id search

Browse Source
This commit is contained in:
qwewqa 2021-01-14 18:38:27 -05:00
parent f85640569b
commit eb88a52bb5
2 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
miyu_bot/commands/cogs/event.py
View File

@ -34,6 +34,13 @@ class Event(commands.Cog):
event: EventMaster event: EventMaster
if arg: if arg:
if arg[0] in ['-', '+']:
try:
latest = self.get_latest_event(ctx)
event = masters.events.get(str(latest.id + int(arg)), ctx)
except ValueError:
event = masters.events.get(arg, ctx)
else:
event = masters.events.get(arg, ctx) event = masters.events.get(arg, ctx)
else: else:
event = self.get_latest_event(ctx) event = self.get_latest_event(ctx)

8
miyu_bot/commands/common/master_asset_manager.py
View File

@ -1,4 +1,4 @@
from functools import cached_property from functools import lru_cache
from typing import Callable, Any, Optional from typing import Callable, Any, Optional
from d4dj_utils.manager.asset_manager import AssetManager from d4dj_utils.manager.asset_manager import AssetManager
@ -14,7 +14,8 @@ class MasterFilterManager:
def __init__(self, manager: AssetManager): def __init__(self, manager: AssetManager):
self.manager = manager self.manager = manager
@cached_property @property
@lru_cache(None)
def music(self): def music(self):
return MasterFilter( return MasterFilter(
self.manager.music_master, self.manager.music_master,
@ -23,7 +24,8 @@ class MasterFilterManager:
fallback_naming_function=lambda m: m.id, fallback_naming_function=lambda m: m.id,
) )
@cached_property @property
@lru_cache(None)
def events(self): def events(self):
return MasterFilter( return MasterFilter(
self.manager.event_master, self.manager.event_master,