add maxkb toggle
This commit is contained in:
16
database.py
16
database.py
@@ -1,5 +1,5 @@
|
||||
from tortoise.contrib.fastapi import register_tortoise
|
||||
from models import PresetCode
|
||||
from models import PresetCode, Setting
|
||||
from schemas import PresetCodeCreate, PresetCodeResponse
|
||||
from typing import List, Optional
|
||||
|
||||
@@ -38,10 +38,20 @@ class DatabaseService:
|
||||
|
||||
@staticmethod
|
||||
async def delete_code(code_id: int) -> bool:
|
||||
"""删除指定 ID 的代码"""
|
||||
code = await PresetCode.get_or_none(id=code_id)
|
||||
if not code:
|
||||
return False
|
||||
|
||||
await code.delete()
|
||||
return True
|
||||
|
||||
@staticmethod
|
||||
async def get_maxkb_enabled() -> bool:
|
||||
setting, _ = await Setting.get_or_create(key="maxkb_enabled", defaults={"value": True})
|
||||
return setting.value
|
||||
|
||||
@staticmethod
|
||||
async def toggle_maxkb_enabled() -> bool:
|
||||
setting, _ = await Setting.get_or_create(key="maxkb_enabled", defaults={"value": True})
|
||||
setting.value = not setting.value
|
||||
await setting.save()
|
||||
return setting.value
|
||||
|
||||
Reference in New Issue
Block a user