Compare commits
17 Commits
Author | SHA1 | Date | |
---|---|---|---|
55b174927b
|
|||
5c88d08c6f
|
|||
79c1f5b147
|
|||
a468824d30
|
|||
efc17fba23
|
|||
148f2901e4
|
|||
2853d0d66e
|
|||
cdae81a6ee | |||
ad3eda601a
|
|||
5efdee2752 | |||
ab514cab11
|
|||
6063452c64 | |||
534c652bb5
|
|||
dda548b5fc | |||
42cc6aa29e | |||
c745405064 | |||
ecfde46f94 |
@ -1,5 +1,6 @@
|
||||
import logging
|
||||
from logging import Logger
|
||||
from typing import Literal
|
||||
|
||||
from libbot.cache.classes import CacheMemcached, CacheRedis
|
||||
from libbot.cache.manager import create_cache_client
|
||||
@ -16,5 +17,12 @@ class HoloBot(PycordBot):
|
||||
self._set_cache_engine()
|
||||
|
||||
def _set_cache_engine(self) -> None:
|
||||
if "cache" in self.config and self.config["cache"]["type"] is not None:
|
||||
self.cache = create_cache_client(self.config, self.config["cache"]["type"])
|
||||
cache_type: Literal["redis", "memcached"] | None = self.config["cache"]["type"]
|
||||
|
||||
if "cache" in self.config and cache_type is not None:
|
||||
self.cache = create_cache_client(
|
||||
self.config,
|
||||
cache_type,
|
||||
prefix=self.config["cache"][cache_type]["prefix"],
|
||||
default_ttl_seconds=3600,
|
||||
)
|
||||
|
@ -25,10 +25,12 @@
|
||||
"cache": {
|
||||
"type": null,
|
||||
"memcached": {
|
||||
"uri": "127.0.0.1:11211"
|
||||
"uri": "127.0.0.1:11211",
|
||||
"prefix": null
|
||||
},
|
||||
"redis": {
|
||||
"uri": "redis://127.0.0.1:6379/0"
|
||||
"uri": "redis://127.0.0.1:6379/0",
|
||||
"prefix": null
|
||||
}
|
||||
},
|
||||
"logging": {
|
||||
|
31
migrations/202507091444.py
Normal file
31
migrations/202507091444.py
Normal file
@ -0,0 +1,31 @@
|
||||
import logging
|
||||
from logging import Logger
|
||||
|
||||
from libbot.utils import config_set, config_delete
|
||||
from mongodb_migrations.base import BaseMigration
|
||||
|
||||
logger: Logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class Migration(BaseMigration):
|
||||
def upgrade(self):
|
||||
try:
|
||||
config_set("prefix", None, "cache", "memcached")
|
||||
config_set("prefix", None, "cache", "redis")
|
||||
except Exception as exc:
|
||||
logger.error(
|
||||
"Could not upgrade the config during migration '%s' due to: %s",
|
||||
__name__,
|
||||
exc,
|
||||
)
|
||||
|
||||
def downgrade(self):
|
||||
try:
|
||||
config_delete("prefix", "cache", "redis")
|
||||
config_delete("prefix", "cache", "memcached")
|
||||
except Exception as exc:
|
||||
logger.error(
|
||||
"Could not downgrade the config during migration '%s' due to: %s",
|
||||
__name__,
|
||||
exc,
|
||||
)
|
@ -5,9 +5,9 @@ requests>=2.32.2
|
||||
aiofiles~=24.1.0
|
||||
apscheduler>=3.10.0
|
||||
async_pymongo==0.1.11
|
||||
libbot[speed,pycord,cache]==4.1.0
|
||||
libbot[speed,pycord,cache]==4.4.0
|
||||
mongodb-migrations==1.3.1
|
||||
pymemcache~=4.0.0
|
||||
redis~=5.2.1
|
||||
redis~=6.2.0
|
||||
ujson~=5.10.0
|
||||
WaifuPicsPython==0.2.0
|
Reference in New Issue
Block a user