diff --git a/nonebot_plugin_localstore/__init__.py b/nonebot_plugin_localstore/__init__.py index 9f08b76..f264dd2 100644 --- a/nonebot_plugin_localstore/__init__.py +++ b/nonebot_plugin_localstore/__init__.py @@ -34,17 +34,29 @@ APP_NAME = "nonebot2" BASE_CACHE_DIR = ( - user_cache_dir(APP_NAME).resolve() + ( + (Path.cwd() / "cache") + if plugin_config.localstore_use_cwd + else user_cache_dir(APP_NAME).resolve() + ) if plugin_config.localstore_cache_dir is None else plugin_config.localstore_cache_dir.resolve() ) BASE_CONFIG_DIR = ( - user_config_dir(APP_NAME).resolve() + ( + (Path.cwd() / "config") + if plugin_config.localstore_use_cwd + else user_config_dir(APP_NAME).resolve() + ) if plugin_config.localstore_config_dir is None else plugin_config.localstore_config_dir.resolve() ) BASE_DATA_DIR = ( - user_data_dir(APP_NAME).resolve() + ( + (Path.cwd() / "data") + if plugin_config.localstore_use_cwd + else user_data_dir(APP_NAME).resolve() + ) if plugin_config.localstore_data_dir is None else plugin_config.localstore_data_dir.resolve() ) diff --git a/nonebot_plugin_localstore/config.py b/nonebot_plugin_localstore/config.py index 59e979a..3f5ab69 100644 --- a/nonebot_plugin_localstore/config.py +++ b/nonebot_plugin_localstore/config.py @@ -5,6 +5,7 @@ class Config(BaseModel): + localstore_use_cwd: bool = False localstore_cache_dir: Optional[Path] = None localstore_config_dir: Optional[Path] = None localstore_data_dir: Optional[Path] = None