分离repo层

nonebot_plugin_nagabus/data/naga_cookies.py

@@ -0,0 +1,11 @@
+from collections.abc import Mapping
+
+from ..datastore import plugin_data
+
+
+async def get_naga_cookies() -> dict:
+    return await plugin_data.config.get("naga_cookies", {})
+
+
+async def set_naga_cookies(cookies: Mapping[str, str]):
+    await plugin_data.config.set("naga_cookies", cookies)

nonebot_plugin_nagabus/naga/service.py

@@ -17,10 +17,10 @@
 from ..config import conf
 from ..utils.tz import TZ_TOKYO
 from .fake_api import FakeNagaApi
-from ..datastore import plugin_data
 from .utils import model_type_to_str
 from ..data.mjs import get_majsoul_paipu
 from .api import NagaApi, OrderReportList
+from ..data.naga_cookies import get_naga_cookies, set_naga_cookies
 from .errors import (
     OrderError,
     InvalidGameError,
@@ -120,14 +120,14 @@ def __init__(self):
         self._order_report = ObservableOrderReport(self.api)
 
     async def start(self):
-        cookies_obj = await plugin_data.config.get("naga_cookies", {})
+        cookies_obj = await get_naga_cookies()
         self.cookies = Cookies(cookies_obj)
 
     async def close(self):
         await self.api.close()
 
     async def set_cookies(self, cookies: Mapping[str, str]):
-        await plugin_data.config.set("naga_cookies", cookies)
+        await set_naga_cookies(cookies)
         self.cookies = Cookies(dict(cookies))
         logger.info(
             f"naga_cookies set to {'; '.join(f'{kv[0]}={kv[1]}' for kv in cookies.items())}"