diff --git a/src/db_utils/sqlite_db.py b/src/db_utils/sqlite_db.py index 7ea8dd7..65ebfef 100644 --- a/src/db_utils/sqlite_db.py +++ b/src/db_utils/sqlite_db.py @@ -132,6 +132,9 @@ class DatabaseHandler: def finalize_task_log(self, task_id): return 1 + def get_statics(self): + return {} + def close(self): self.cursor.close() self.conn.close() @@ -517,4 +520,36 @@ class JavbusDBHandler(DatabaseHandler): except sqlite3.Error as e: logging.error(f"查询 href 失败: {e}") return None - \ No newline at end of file + + + # 查询状态 + def get_statics(self): + result = {} + try: + # 获取 performers、studios 等表的最终行数 + self.cursor.execute(f"SELECT COUNT(*) FROM {self.tbl_name_actors} ") + result['actors'] = self.cursor.fetchone()[0] + + self.cursor.execute(f"SELECT COUNT(*) FROM {self.tbl_name_actors} where uncensored=1") + result['actors_uncensored'] = self.cursor.fetchone()[0] + + self.cursor.execute(f"SELECT COUNT(*) FROM {self.tbl_name_actors} where is_full_data=1") + result['actors_full_data'] = self.cursor.fetchone()[0] + + self.cursor.execute(f"SELECT COUNT(*) FROM {self.tbl_name_actors} where is_full_data=1 and is_full_data=1") + result['actors_uncensored_full_data'] = self.cursor.fetchone()[0] + + # 获取 performers、studios 等表的最终行数 + self.cursor.execute(f"SELECT COUNT(*) FROM {self.tbl_name_movies} ") + result['movies'] = self.cursor.fetchone()[0] + + self.cursor.execute(f"SELECT COUNT(*) FROM {self.tbl_name_movies} where uncensored=1") + result['movies_uncensored'] = self.cursor.fetchone()[0] + + self.cursor.execute(f"SELECT COUNT(*) FROM {self.tbl_name_movies} where is_full_data=1 and is_full_data=1") + result['movies_uncensored_full_data'] = self.cursor.fetchone()[0] + + except sqlite3.Error as e: + logging.error(f"query error: {e}") + + return result \ No newline at end of file diff --git a/src/monitor/check_javbus.py b/src/monitor/check_javbus.py new file mode 100644 index 0000000..ba01c68 --- /dev/null +++ b/src/monitor/check_javbus.py @@ -0,0 +1,13 @@ +import json +import time +import src.db_utils.sqlite_db as sqlite_db +db_tools = sqlite_db.JavbusDBHandler() + + +if __name__ == "__main__": + # 命令行参数处理 + result = db_tools.get_statics() + print(result) + + +