++

base/update_feed.py

@@ -81,8 +81,63 @@ def check_data_count():
             conn.close()
 
 
-if __name__ == '__main__':
-    with sync_playwright() as playwright:
-        run(playwright)
+def read_ids():
+    ids = []
+    # 数据库连接参数
+    db_params = {
+        'dbname': 'freshrss',
+        'user': 'freshrss',
+        'password': 'freshrss',
+        'host': 'erhe.top',
+        'port': '20788'
+    }
+
+    # SQL 查询语句
+    query = 'SELECT id, url FROM freshrss_toor_feed order by id ASC;'
+
+    try:
+        # 建立数据库连接
+        conn = psycopg2.connect(**db_params)
+
+        # 创建 cursor 对象
+        cur = conn.cursor()
+
+        # 执行 SQL 查询
+        cur.execute(query)
+
+        # 获取查询结果
+        records = cur.fetchall()
+
+        for record in records:
+            id, url = record
+            print(id, url)
+            ids.append(id)
+
+    except psycopg2.Error as e:
+        print(f"Database error: {e}")
+        conn.rollback()
+    except Exception as e:
+        print(f"An error occurred: {e}")
+    finally:
+        # 关闭 cursor 和连接
+        if 'cur' in locals():
+            cur.close()
+        if 'conn' in locals():
+            conn.close()
+
+    if ids:
+        return ids
+    else:
+        return None
 
-    check_data_count()
+
+if __name__ == '__main__':
+    ids = read_ids()
+
+    if not ids:
+        print('未获取到订阅源id, 程序退出')
+        exit(0)
+    # with sync_playwright() as playwright:
+    #     run(playwright)
+    #
+    # check_data_count()