'commit'

dsLightRag/WxGzh/T2_CollectArticle.py

@@ -40,6 +40,7 @@ if not logger.handlers:
     handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
     logger.addHandler(handler)
 
+
 async def get_wechat_sources():
     """从t_wechat_source表获取微信公众号列表"""
     try:
@@ -51,13 +52,14 @@ async def get_wechat_sources():
         await pool.close()
 
 
-
 async def is_article_exist(pool, article_url):
     """检查文章URL是否已存在数据库中"""
     try:
         async with pool.acquire() as conn:
             row = await conn.fetchrow('''
-                                      SELECT 1 FROM t_wechat_articles WHERE url = $1 LIMIT 1
+                                      SELECT 1
+                                      FROM t_wechat_articles
+                                      WHERE url = $1 LIMIT 1
                                       ''', article_url)
             return row is not None
     except Exception as e:
@@ -203,8 +205,7 @@ if __name__ == '__main__':
             try:
                 pool = loop.run_until_complete(init_postgres_pool())
                 loop.run_until_complete(
-                    save_article_to_db(pool, article_title, account_name, article_url, publish_time, content,
-                                       id))
+                    save_article_to_db(pool, article_title, account_name, article_url, publish_time, content, id))
             finally:
                 loop.run_until_complete(pool.close())
                 loop.close()