add 站点流控

app/chain/search.py

@@ -36,6 +36,11 @@ class SearchChain(ChainBase):
         for indexer in self.siteshelper.get_indexers():
             if not settings.INDEXER_SITES \
                     or any([s in indexer.get("domain") for s in settings.INDEXER_SITES.split(',')]):
+                # 站点流控
+                state, msg = self.siteshelper.check(indexer.get("domain"))
+                if not state:
+                    logger.warn(msg)
+                    continue
                 indexer_sites.append(indexer)
         if not indexer_sites:
             logger.warn('未开启任何有效站点，无法搜索资源')

app/db/models/site.py

@@ -22,9 +22,9 @@ class Site(Base):
     filter = Column(String)
     render = Column(Integer)
     note = Column(String)
-    limit_interval = Column(Integer)
+    limit_interval = Column(Integer, default=0)
-    limit_count = Column(Integer)
+    limit_count = Column(Integer, default=0)
-    limit_seconds = Column(Integer)
+    limit_seconds = Column(Integer, default=0)
     is_active = Column(Boolean(), default=True)
     lst_mod_date = Column(String, default=datetime.now().strftime("%Y-%m-%d %H:%M:%S"))