refactor: move sector phase persistence to heaven repository
This commit is contained in:
+1
-31
@@ -2,7 +2,7 @@ from __future__ import annotations
|
||||
|
||||
import json
|
||||
import sqlite3
|
||||
from datetime import datetime, timezone
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
@@ -665,36 +665,6 @@ class ReviewDatabase(
|
||||
)
|
||||
MigrationRunner().apply(connection, MIGRATIONS)
|
||||
|
||||
|
||||
|
||||
def list_sector_phase_overrides(self) -> dict[str, str]:
|
||||
with self.connect() as connection:
|
||||
rows = connection.execute(
|
||||
"SELECT name, element FROM sector_phase_overrides ORDER BY updated_at DESC, name"
|
||||
).fetchall()
|
||||
return {row["name"]: row["element"] for row in rows}
|
||||
|
||||
def save_sector_phase_override(self, name: str, element: str) -> None:
|
||||
now = datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
with self.connect() as connection:
|
||||
connection.execute(
|
||||
"""
|
||||
INSERT INTO sector_phase_overrides (name, element, updated_at)
|
||||
VALUES (?, ?, ?)
|
||||
ON CONFLICT(name) DO UPDATE SET
|
||||
element = excluded.element,
|
||||
updated_at = excluded.updated_at
|
||||
""",
|
||||
(name, element, now),
|
||||
)
|
||||
|
||||
def delete_sector_phase_override(self, name: str) -> bool:
|
||||
with self.connect() as connection:
|
||||
cursor = connection.execute(
|
||||
"DELETE FROM sector_phase_overrides WHERE name = ?",
|
||||
(name,),
|
||||
)
|
||||
return cursor.rowcount > 0
|
||||
def list_wencai_saved_queries(
|
||||
self, user_id: int, limit: int = 30
|
||||
) -> list[dict[str, Any]]:
|
||||
|
||||
Reference in New Issue
Block a user