fix dboper

app/db/site_oper.py

@@ -0,0 +1,76 @@
+from typing import Tuple, List
+
+from app.db import DbOper
+from app.db.models.site import Site
+
+
+class SiteOper(DbOper):
+    """
+    站点管理
+    """
+
+    def add(self, **kwargs) -> Tuple[bool, str]:
+        """
+        新增站点
+        """
+        site = Site(**kwargs)
+        if not site.get_by_domain(self._db, kwargs.get("domain")):
+            site.create(self._db)
+            return True, "新增站点成功"
+        return False, "站点已存在"
+
+    def get(self, sid: int):
+        """
+        查询单个站点
+        """
+        return Site.get(self._db, sid)
+
+    def list(self) -> List[Site]:
+        """
+        获取站点列表
+        """
+        return Site.list(self._db)
+
+    def list_active(self):
+        """
+        按状态获取站点列表
+        """
+        return Site.get_actives(self._db)
+
+    def delete(self, sid: int):
+        """
+        删除站点
+        """
+        return Site.delete(self._db, sid)
+
+    def update(self, sid: int, payload: dict):
+        """
+        更新站点
+        """
+        site = Site.get(self._db, sid)
+        site.update(self._db, payload)
+        return site
+
+    def get_by_domain(self, domain: str) -> Site:
+        """
+        按域名获取站点
+        """
+        return Site.get_by_domain(self._db, domain)
+
+    def exists(self, domain: str) -> bool:
+        """
+        判断站点是否存在
+        """
+        return Site.get_by_domain(self._db, domain) is not None
+
+    def update_cookie(self, domain: str, cookies: str) -> Tuple[bool, str]:
+        """
+        更新站点Cookie
+        """
+        site = Site.get_by_domain(self._db, domain)
+        if not site:
+            return False, "站点不存在"
+        site.update(self._db, {
+            "cookie": cookies
+        })
+        return True, "更新站点Cookie成功"