diff options
| -rw-r--r-- | bot/exts/info/doc/__init__.py | 3 | ||||
| -rw-r--r-- | bot/exts/info/doc/_batch_parser.py | 4 | ||||
| -rw-r--r-- | bot/exts/info/doc/_cog.py | 10 |
3 files changed, 9 insertions, 8 deletions
diff --git a/bot/exts/info/doc/__init__.py b/bot/exts/info/doc/__init__.py index 2bb43a950..38a8975c0 100644 --- a/bot/exts/info/doc/__init__.py +++ b/bot/exts/info/doc/__init__.py @@ -5,8 +5,9 @@ MAX_SIGNATURE_AMOUNT = 3 PRIORITY_PACKAGES = ( "python", ) +NAMESPACE = "doc" -doc_cache = DocRedisCache(namespace="Docs") +doc_cache = DocRedisCache(namespace=NAMESPACE) def setup(bot: Bot) -> None: diff --git a/bot/exts/info/doc/_batch_parser.py b/bot/exts/info/doc/_batch_parser.py index 42d81e98c..872f08ea9 100644 --- a/bot/exts/info/doc/_batch_parser.py +++ b/bot/exts/info/doc/_batch_parser.py @@ -15,7 +15,7 @@ from bs4 import BeautifulSoup import bot from bot.constants import Channels from bot.utils.lock import lock_arg -from . import doc_cache +from . import NAMESPACE, doc_cache from ._parsing import get_symbol_markdown if TYPE_CHECKING: from ._cog import DocItem @@ -100,7 +100,7 @@ class BatchParser: self.stale_inventory_notifier = StaleInventoryNotifier() - @lock_arg("doc.get_markdown", "doc_item", attrgetter("url"), wait=True) + @lock_arg(NAMESPACE, "doc_item", attrgetter("url"), wait=True) async def get_markdown(self, doc_item: DocItem) -> str: """ Get the result Markdown of `doc_item`. diff --git a/bot/exts/info/doc/_cog.py b/bot/exts/info/doc/_cog.py index c3458d776..430e8ebcb 100644 --- a/bot/exts/info/doc/_cog.py +++ b/bot/exts/info/doc/_cog.py @@ -19,7 +19,7 @@ from bot.pagination import LinePaginator from bot.utils.lock import SharedEvent, lock from bot.utils.messages import send_denial, wait_for_deletion from bot.utils.scheduling import Scheduler -from . import PRIORITY_PACKAGES, doc_cache +from . import NAMESPACE, PRIORITY_PACKAGES, doc_cache from ._batch_parser import BatchParser from ._inventory_parser import InventoryDict, fetch_inventory @@ -74,7 +74,7 @@ class DocCog(commands.Cog): self.init_refresh_task = self.bot.loop.create_task(self.init_refresh_inventory()) - @lock("doc", COMMAND_LOCK_SINGLETON, raise_error=True) + @lock(NAMESPACE, COMMAND_LOCK_SINGLETON, raise_error=True) async def init_refresh_inventory(self) -> None: """Refresh documentation inventory on cog initialization.""" await self.bot.wait_until_guild_available() @@ -330,7 +330,7 @@ class DocCog(commands.Cog): @docs_group.command(name='setdoc', aliases=('s',)) @commands.has_any_role(*MODERATION_ROLES) - @lock("doc", COMMAND_LOCK_SINGLETON, raise_error=True) + @lock(NAMESPACE, COMMAND_LOCK_SINGLETON, raise_error=True) async def set_command( self, ctx: commands.Context, @@ -367,7 +367,7 @@ class DocCog(commands.Cog): @docs_group.command(name='deletedoc', aliases=('removedoc', 'rm', 'd')) @commands.has_any_role(*MODERATION_ROLES) - @lock("doc", COMMAND_LOCK_SINGLETON, raise_error=True) + @lock(NAMESPACE, COMMAND_LOCK_SINGLETON, raise_error=True) async def delete_command(self, ctx: commands.Context, package_name: PackageName) -> None: """ Removes the specified package from the database. @@ -386,7 +386,7 @@ class DocCog(commands.Cog): @docs_group.command(name="refreshdoc", aliases=("rfsh", "r")) @commands.has_any_role(*MODERATION_ROLES) - @lock("doc", COMMAND_LOCK_SINGLETON, raise_error=True) + @lock(NAMESPACE, COMMAND_LOCK_SINGLETON, raise_error=True) async def refresh_command(self, ctx: commands.Context) -> None: """Refresh inventories and show the difference.""" old_inventories = set(self.base_urls) |