add maxkb toggle

This commit is contained in:
2026-04-27 08:57:18 -06:00
parent c2549cf3cc
commit e1af34a986
3 changed files with 32 additions and 6 deletions

View File

@@ -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