add 站点图标缓存

app/db/models/site.py

@@ -35,3 +35,18 @@ class Site(Base):
     @staticmethod
     def get_actives(db: Session):
         return db.query(Site).filter(Site.is_active == 1).all()
+
+
+class SiteIcon(Base):
+    """
+    站点图标表
+    """
+    id = Column(Integer, Sequence('id'), primary_key=True, index=True)
+    name = Column(String, nullable=False)
+    domain = Column(String, index=True)
+    url = Column(String, nullable=False)
+    base64 = Column(String)
+
+    @staticmethod
+    def get_by_domain(db: Session, domain: str):
+        return db.query(SiteIcon).filter(SiteIcon.domain == domain).first()

app/db/sites.py

@@ -3,7 +3,7 @@ from typing import Tuple, List
 from sqlalchemy.orm import Session
 
 from app.db import SessionLocal
-from app.db.models.site import Site
+from app.db.models.site import Site, SiteIcon
 
 
 class Sites:
@@ -60,3 +60,17 @@ class Sites:
             "cookie": cookies
         })
         return True, "更新站点Cookie成功"
+
+    def update_icon(self, name: str, domain: str, icon_url: str, icon_base64: str) -> bool:
+        """
+        更新站点图标
+        """
+        siteicon = SiteIcon(name=name, domain=domain, url=icon_url, base64=icon_base64)
+        if not siteicon.get_by_domain(self._db, domain):
+            siteicon.create(self._db)
+        elif icon_base64:
+            siteicon.update(self._db, {
+                "url": icon_url,
+                "base64": f"data:image/ico;base64,{icon_base64}"
+            })
+        return True