modify scripts

src/db_utils/sqlite_db.py

@@ -631,4 +631,64 @@ class JavbusDBHandler(DatabaseHandler):
         except sqlite3.Error as e:
             logging.error(f"query error: {e}")
 
-        return result
+        return result
+
+    # 处理影片的 无码 字段
+    def reset_movies_uncensored(self):
+        try:
+            logging.info("创建临时表以便于保存待更新记录")
+            self.cursor.execute("""
+                CREATE TEMPORARY TABLE IF NOT EXISTS temp_movies_to_update (
+                    movie_id INTEGER PRIMARY KEY
+                )
+            """)
+            # 清空临时表（以防之前有残留数据）
+            self.cursor.execute("DELETE FROM temp_movies_to_update")
+
+            logging.info(f"开始收集需要更新的影片ID...")
+                # 使用单个SQL语句完成所有条件的查询和插入
+            self.cursor.execute("""
+                INSERT OR IGNORE INTO temp_movies_to_update (movie_id)
+                SELECT DISTINCT m.id
+                FROM javbus_movies m
+                -- 连接演员表
+                LEFT JOIN javbus_actors_movies am ON m.id = am.movie_id
+                LEFT JOIN javbus_actors a ON am.actor_id = a.id
+                -- 连接标签/系列/工作室表
+                LEFT JOIN javbus_labels l ON m.label_id = l.id
+                LEFT JOIN javbus_series s ON m.series_id = s.id
+                LEFT JOIN javbus_studios st ON m.studio_id = st.id
+                -- 筛选条件：任一表的href包含'uncensored'
+                WHERE a.href LIKE '%uncensored%'
+                OR l.href LIKE '%uncensored%'
+                OR s.href LIKE '%uncensored%'
+                OR st.href LIKE '%uncensored%'
+            """)
+            
+            total_count = self.cursor.execute("SELECT COUNT(*) FROM temp_movies_to_update").fetchone()[0]
+            total_movies = self.cursor.execute("SELECT COUNT(*) FROM javbus_movies").fetchone()[0]
+            logging.info(f"共收集到 {total_count} 部需要更新的影片, 共有 {total_movies} 部影片")
+
+            # 1. 将所有记录的uncensored默认设为0
+            logging.info("开始将所有影片的uncensored设为默认值0...")
+            self.cursor.execute("UPDATE javbus_movies SET uncensored = 0")
+            logging.info(f"已将 {self.cursor.rowcount} 条记录的uncensored设为0")
+
+            # 2. 将临时表中匹配的记录设为1
+            logging.info("开始将匹配的影片的uncensored设为1...")
+            self.cursor.execute("""
+                UPDATE javbus_movies 
+                SET uncensored = 1 
+                WHERE id IN (SELECT movie_id FROM temp_movies_to_update)
+            """)
+            logging.info(f"已将 {self.cursor.rowcount} 条记录的uncensored设为1")
+
+            # 3. 清理临时表，也可以不清理，以便于抽检
+
+            self.conn.commit()
+            logging.info("所有更新已提交")
+
+        except sqlite3.Error as e:
+            self.conn.rollback()
+            logging.error("Error inserting movie: %s", e)
+            logging.error(f"query error: {e}")