628 lines
23 KiB
Python
628 lines
23 KiB
Python
import sqlite3
|
||
import json
|
||
import config
|
||
import logging
|
||
from datetime import datetime
|
||
|
||
# 连接 SQLite 数据库
|
||
DB_PATH = f"{config.global_share_data_dir}/sqlite/shared.db" # 替换为你的数据库文件
|
||
conn = sqlite3.connect(DB_PATH, check_same_thread=False)
|
||
cursor = conn.cursor()
|
||
|
||
# """从指定表中通过 href 查找 id"""
|
||
def get_id_by_href(table: str, href: str) -> int:
|
||
if href is None:
|
||
return None
|
||
cursor.execute(f"SELECT id FROM {table} WHERE href = ?", (href,))
|
||
row = cursor.fetchone()
|
||
return row[0] if row else None
|
||
|
||
|
||
def insert_actor_index(name, href, from_actor_list=None, from_movie_list=None):
|
||
try:
|
||
# **查询是否已存在该演员**
|
||
cursor.execute("SELECT id, name, from_actor_list, from_movie_list FROM javdb_actors WHERE href = ?", (href,))
|
||
existing_actor = cursor.fetchone()
|
||
|
||
if existing_actor: # **如果演员已存在**
|
||
actor_id, existing_name, existing_actor_list, existing_movie_list = existing_actor
|
||
|
||
# **如果没有传入值，则保持原有值**
|
||
from_actor_list = from_actor_list if from_actor_list is not None else existing_actor_list
|
||
from_movie_list = from_movie_list if from_movie_list is not None else existing_movie_list
|
||
|
||
cursor.execute("""
|
||
UPDATE javdb_actors
|
||
SET name = ?,
|
||
from_actor_list = ?,
|
||
from_movie_list = ?,
|
||
updated_at = datetime('now', 'localtime')
|
||
WHERE href = ?
|
||
""", (name, from_actor_list, from_movie_list, href))
|
||
else: # **如果演员不存在，插入**
|
||
cursor.execute("""
|
||
INSERT INTO javdb_actors (href, name, from_actor_list, from_movie_list)
|
||
VALUES (?, ?, COALESCE(?, 0), COALESCE(?, 0))
|
||
""", (href, name, from_actor_list, from_movie_list))
|
||
|
||
conn.commit()
|
||
|
||
performer_id = get_id_by_href('javdb_actors', href)
|
||
if performer_id:
|
||
logging.debug(f'Inserted/Updated actor index, id: {performer_id}, name: {name}, href: {href}')
|
||
|
||
return performer_id
|
||
|
||
except sqlite3.Error as e:
|
||
conn.rollback()
|
||
logging.error(f"数据库错误: {e}")
|
||
return None
|
||
except Exception as e:
|
||
conn.rollback()
|
||
logging.error(f"未知错误: {e}")
|
||
return None
|
||
|
||
def insert_movie_index(title, href, from_actor_list=None, from_movie_makers=None, from_movie_series=None):
|
||
try:
|
||
# **先检查数据库中是否已有该电影**
|
||
cursor.execute("SELECT id, from_actor_list, from_movie_makers, from_movie_series FROM javdb_movies WHERE href = ?", (href,))
|
||
existing_movie = cursor.fetchone()
|
||
|
||
if existing_movie: # **如果电影已存在**
|
||
movie_id, existing_actor, existing_maker, existing_series = existing_movie
|
||
|
||
# **如果没有传入值，就用原来的值**
|
||
from_actor_list = from_actor_list if from_actor_list is not None else existing_actor
|
||
from_movie_makers = from_movie_makers if from_movie_makers is not None else existing_maker
|
||
from_movie_series = from_movie_series if from_movie_series is not None else existing_series
|
||
|
||
cursor.execute("""
|
||
UPDATE javdb_movies
|
||
SET title = ?,
|
||
from_actor_list = ?,
|
||
from_movie_makers = ?,
|
||
from_movie_series = ?,
|
||
updated_at = datetime('now', 'localtime')
|
||
WHERE href = ?
|
||
""", (title, from_actor_list, from_movie_makers, from_movie_series, href))
|
||
else: # **如果电影不存在，插入**
|
||
cursor.execute("""
|
||
INSERT INTO javdb_movies (title, href, from_actor_list, from_movie_makers, from_movie_series)
|
||
VALUES (?, ?, COALESCE(?, 0), COALESCE(?, 0), COALESCE(?, 0))
|
||
""", (title, href, from_actor_list, from_movie_makers, from_movie_series))
|
||
|
||
conn.commit()
|
||
|
||
movie_id = get_id_by_href('javdb_movies', href)
|
||
if movie_id:
|
||
logging.debug(f'Inserted/Updated movie index, id: {movie_id}, title: {title}, href: {href}')
|
||
|
||
return movie_id
|
||
|
||
except Exception as e:
|
||
conn.rollback()
|
||
logging.error(f"Error inserting/updating movie: {e}")
|
||
return None
|
||
|
||
|
||
# 插入演员和电影的关联数据
|
||
def insert_actor_movie(performer_id, movie_id, tags=''):
|
||
try:
|
||
cursor.execute("""
|
||
INSERT INTO javdb_actors_movies (actor_id, movie_id, tags)
|
||
VALUES (?, ?, ?)
|
||
ON CONFLICT(actor_id, movie_id) DO UPDATE SET tags=excluded.tags
|
||
""",
|
||
(performer_id, movie_id, tags)
|
||
)
|
||
conn.commit()
|
||
|
||
#logging.debug(f'insert one performer_movie, performer_id: {performer_id}, movie_id: {movie_id}')
|
||
|
||
return performer_id
|
||
|
||
except Exception as e:
|
||
conn.rollback()
|
||
logging.error("Error inserting movie: %s", e)
|
||
return None
|
||
|
||
# 插入演员数据
|
||
def insert_or_update_actor(actor):
|
||
try:
|
||
cursor.execute('''
|
||
INSERT INTO javdb_actors (name, href, pic, is_full_data, updated_at)
|
||
VALUES (?, ?, ?, 1, datetime('now', 'localtime'))
|
||
ON CONFLICT(href) DO UPDATE SET name=excluded.name, pic=excluded.pic, is_full_data=1, updated_at=datetime('now', 'localtime')
|
||
''', (actor['name'], actor['href'], actor['pic']))
|
||
|
||
cursor.execute('SELECT id FROM javdb_actors WHERE href = ?', (actor['href'],))
|
||
conn.commit()
|
||
|
||
actor_id = get_id_by_href('javdb_actors', actor['href'])
|
||
if actor_id is None:
|
||
logging.warning(f'insert data error. name: {actor['name']}, href: {actor['href']}')
|
||
return None
|
||
|
||
logging.debug(f'insert one actor, id: {actor_id}, name: {actor['name']}, href: {actor['href']}')
|
||
|
||
# 插入别名
|
||
for alias in actor.get("alias") or []:
|
||
cursor.execute('''
|
||
INSERT OR IGNORE INTO javdb_actors_alias (actor_id, alias, updated_at)
|
||
VALUES (?, ?, datetime('now', 'localtime'))
|
||
''', (actor_id, alias))
|
||
|
||
conn.commit()
|
||
|
||
# 插入影片列表
|
||
for movie in actor.get("credits") or []:
|
||
movie_id = get_id_by_href('javdb_movies', movie['href'])
|
||
# 影片不存在，先插入
|
||
if movie_id is None:
|
||
movie_id = insert_movie_index(movie['title'], movie['href'], from_actor_list=1)
|
||
if movie_id:
|
||
tmp_id = insert_actor_movie(actor_id, movie_id)
|
||
if tmp_id :
|
||
logging.debug(f'insert one performer_movie, performer_id: {actor_id}, movie_id: {movie_id}')
|
||
else:
|
||
logging.warning(f'insert performer_movie failed. performer_id: {actor_id}, moive href: {movie['href']}')
|
||
|
||
return actor_id
|
||
except Exception as e:
|
||
logging.error(f"插入/更新演员 {actor['name']} 失败: {e}")
|
||
conn.rollback()
|
||
|
||
# 删除演员
|
||
def delete_actor_by_href(href):
|
||
try:
|
||
cursor.execute('DELETE FROM javdb_actors WHERE href = ?', (href,))
|
||
conn.commit()
|
||
logging.info(f"成功删除演员: {href}")
|
||
except Exception as e:
|
||
logging.error(f"删除演员 {href} 失败: {e}")
|
||
conn.rollback()
|
||
|
||
# 查询
|
||
def query_actors(**filters):
|
||
try:
|
||
sql = "SELECT href, name FROM javdb_actors WHERE 1=1"
|
||
params = []
|
||
|
||
if "id" in filters:
|
||
sql += " AND id = ?"
|
||
params.append(filters["id"])
|
||
if "href" in filters:
|
||
sql += " AND href = ?"
|
||
params.append(filters["href"])
|
||
if "name" in filters:
|
||
sql += " AND name LIKE ?"
|
||
params.append(f"%{filters['name']}%")
|
||
if "is_full_data" in filters:
|
||
sql += " AND is_full_data = ?"
|
||
params.append(filters["is_full_data"])
|
||
if 'limit' in filters:
|
||
sql += " limit ?"
|
||
params.append(filters["limit"])
|
||
|
||
cursor.execute(sql, params)
|
||
#return [row[0].lower() for row in cursor.fetchall()] # 返回小写
|
||
return [{'href': row[0], 'name': row[1]} for row in cursor.fetchall()]
|
||
|
||
except sqlite3.Error as e:
|
||
logging.error(f"查询 href 失败: {e}")
|
||
return None
|
||
|
||
|
||
# 插入或更新发行商 """
|
||
def insert_or_update_makers(data):
|
||
try:
|
||
cursor.execute("""
|
||
INSERT INTO javdb_makers (name, href, updated_at)
|
||
VALUES (?, ? , datetime('now', 'localtime'))
|
||
ON CONFLICT(href) DO UPDATE SET
|
||
name = excluded.name,
|
||
updated_at = datetime('now', 'localtime')
|
||
""", (data["name"], data["href"]))
|
||
conn.commit()
|
||
|
||
# 获取 performer_id
|
||
cursor.execute("SELECT id FROM javdb_makers WHERE href = ?", (data["href"],))
|
||
dist_id = cursor.fetchone()[0]
|
||
if dist_id:
|
||
logging.debug(f"成功插入/更新发行商: {data['name']}")
|
||
return dist_id
|
||
else:
|
||
return None
|
||
except sqlite3.Error as e:
|
||
conn.rollback()
|
||
logging.error(f"数据库错误: {e}")
|
||
return None
|
||
|
||
# 删除发行商（按 id 或 name） """
|
||
def delete_maker(identifier):
|
||
try:
|
||
if isinstance(identifier, int):
|
||
cursor.execute("DELETE FROM javdb_makers WHERE id = ?", (identifier,))
|
||
elif isinstance(identifier, str):
|
||
cursor.execute("DELETE FROM javdb_makers WHERE name = ?", (identifier,))
|
||
conn.commit()
|
||
logging.info(f"成功删除发行商: {identifier}")
|
||
except sqlite3.Error as e:
|
||
conn.rollback()
|
||
logging.error(f"删除失败: {e}")
|
||
|
||
# 查询发行商（按 id 或 name） """
|
||
def query_maker(identifier):
|
||
try:
|
||
if isinstance(identifier, int):
|
||
cursor.execute("SELECT * FROM javdb_makers WHERE id = ?", (identifier,))
|
||
else:
|
||
cursor.execute("SELECT * FROM javdb_makers WHERE name LIKE ?", (f"%{identifier}%",))
|
||
|
||
distributor = cursor.fetchone()
|
||
if distributor:
|
||
return dict(zip([desc[0] for desc in cursor.description], distributor))
|
||
else:
|
||
logging.warning(f"未找到发行商: {identifier}")
|
||
return None
|
||
except sqlite3.Error as e:
|
||
logging.error(f"查询失败: {e}")
|
||
return None
|
||
|
||
# 按条件查询 href 列表
|
||
def query_maker_hrefs(**filters):
|
||
try:
|
||
sql = "SELECT href FROM javdb_makers WHERE 1=1"
|
||
params = []
|
||
|
||
if "id" in filters:
|
||
sql += " AND id = ?"
|
||
params.append(filters["id"])
|
||
if "url" in filters:
|
||
sql += " AND href = ?"
|
||
params.append(filters["href"])
|
||
if "name" in filters:
|
||
sql += " AND name LIKE ?"
|
||
params.append(f"%{filters['name']}%")
|
||
|
||
cursor.execute(sql, params)
|
||
return [row[0] for row in cursor.fetchall()] # 链接使用小写
|
||
|
||
except sqlite3.Error as e:
|
||
logging.error(f"查询 href 失败: {e}")
|
||
return None
|
||
|
||
# """ 插入或更新制作公司 """
|
||
def insert_or_update_series(data):
|
||
try:
|
||
cursor.execute("""
|
||
INSERT INTO javdb_series (name, href, updated_at)
|
||
VALUES (?, ?, datetime('now', 'localtime'))
|
||
ON CONFLICT(href) DO UPDATE SET
|
||
name = excluded.name,
|
||
updated_at = datetime('now', 'localtime')
|
||
""", (data["name"], data["href"]))
|
||
conn.commit()
|
||
|
||
# 获取 performer_id
|
||
cursor.execute("SELECT id FROM javdb_series WHERE href = ?", (data["href"],))
|
||
stu_id = cursor.fetchone()[0]
|
||
if stu_id:
|
||
logging.debug(f"成功插入/更新发行商: {data['name']}")
|
||
return stu_id
|
||
else:
|
||
return None
|
||
except sqlite3.Error as e:
|
||
conn.rollback()
|
||
logging.error(f"数据库错误: {e}")
|
||
return None
|
||
|
||
# """ 删除制作公司（按 id 或 name） """
|
||
def delete_series(identifier):
|
||
try:
|
||
if isinstance(identifier, int):
|
||
cursor.execute("DELETE FROM javdb_series WHERE id = ?", (identifier,))
|
||
elif isinstance(identifier, str):
|
||
cursor.execute("DELETE FROM javdb_series WHERE name = ?", (identifier,))
|
||
conn.commit()
|
||
logging.info(f"成功删除制作公司: {identifier}")
|
||
except sqlite3.Error as e:
|
||
conn.rollback()
|
||
logging.error(f"删除失败: {e}")
|
||
|
||
# """ 查询制作公司（按 id 或 name） """
|
||
def query_series(identifier):
|
||
try:
|
||
if isinstance(identifier, int):
|
||
cursor.execute("SELECT * FROM javdb_series WHERE id = ?", (identifier,))
|
||
else:
|
||
cursor.execute("SELECT * FROM javdb_series WHERE name LIKE ?", (f"%{identifier}%",))
|
||
|
||
studio = cursor.fetchone()
|
||
if studio:
|
||
return dict(zip([desc[0] for desc in cursor.description], studio))
|
||
else:
|
||
logging.warning(f"未找到制作公司: {identifier}")
|
||
return None
|
||
except sqlite3.Error as e:
|
||
logging.error(f"查询失败: {e}")
|
||
return None
|
||
|
||
# 按条件查询 href 列表
|
||
def query_series_hrefs(**filters):
|
||
try:
|
||
sql = "SELECT href FROM javdb_series WHERE 1=1"
|
||
params = []
|
||
|
||
if "id" in filters:
|
||
sql += " AND id = ?"
|
||
params.append(filters["id"])
|
||
if "href" in filters:
|
||
sql += " AND href = ?"
|
||
params.append(filters["href"])
|
||
if "name" in filters:
|
||
sql += " AND name LIKE ?"
|
||
params.append(f"%{filters['name']}%")
|
||
|
||
cursor.execute(sql, params)
|
||
return [row[0] for row in cursor.fetchall()] # 链接使用小写
|
||
|
||
except sqlite3.Error as e:
|
||
logging.error(f"查询 href 失败: {e}")
|
||
return None
|
||
|
||
|
||
# """插入或更新电影数据"""
|
||
def insert_or_update_movie(movie):
|
||
try:
|
||
# 获取相关 ID
|
||
makers_id = get_id_by_href('javdb_makers', movie['maker_link'])
|
||
series_id = get_id_by_href('javdb_series', movie['series_link'])
|
||
|
||
|
||
cursor.execute("""
|
||
INSERT INTO javdb_movies (href, title, cover_url, serial_number, release_date, duration,
|
||
maker_id, series_id, is_full_data, updated_at)
|
||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, 1, datetime('now', 'localtime'))
|
||
ON CONFLICT(href) DO UPDATE SET
|
||
title=excluded.title,
|
||
cover_url=excluded.cover_url,
|
||
serial_number=excluded.serial_number,
|
||
release_date=excluded.release_date,
|
||
duration=excluded.duration,
|
||
maker_id=excluded.maker_id,
|
||
series_id=excluded.series_id,
|
||
is_full_data=1,
|
||
updated_at=datetime('now', 'localtime')
|
||
""", (movie['href'], movie['title'], movie['cover_url'], movie['serial_number'],
|
||
movie['release_date'], movie['duration'], makers_id, series_id))
|
||
|
||
conn.commit()
|
||
|
||
# 获取插入的 movie_id
|
||
movie_id = get_id_by_href('javdb_movies', movie['href'])
|
||
if movie_id is None:
|
||
return None
|
||
|
||
logging.debug(f'insert one move, id: {movie_id}, title: {movie['title']}, href: {movie['href']}')
|
||
|
||
# 插入 performers_movies 关系表
|
||
for performer in movie.get('actors', []):
|
||
performer_id = get_id_by_href('javdb_actors', performer['href'])
|
||
# 如果演员不存在，先插入
|
||
if performer_id is None:
|
||
performer_id = insert_actor_index(performer['name'], performer['href'], from_movie_list=1)
|
||
if performer_id:
|
||
tmp_id = insert_actor_movie(performer_id, movie_id)
|
||
if tmp_id:
|
||
logging.debug(f"insert one perfomer_movie. perfomer_id: {performer_id}, movie_id:{movie_id}")
|
||
else:
|
||
logging.debug(f'insert perfomer_movie failed. perfomer_id: {performer_id}, movie_id:{movie_id}')
|
||
else:
|
||
logging.warning(f'insert perfomer failed. name: {performer['name']}, href: {performer['href']}')
|
||
|
||
return movie_id
|
||
|
||
except Exception as e:
|
||
conn.rollback()
|
||
logging.error("Error inserting movie: %s", e)
|
||
return None
|
||
|
||
# """插入或更新电影数据(异常url的处理，比如404链接)"""
|
||
def insert_or_update_movie_404(title, href, is_full_data=1):
|
||
try:
|
||
# 插入或更新电影信息
|
||
cursor.execute(
|
||
"""
|
||
INSERT INTO javdb_movies (title, href, is_full_data, updated_at)
|
||
VALUES (?, ?, ?, datetime('now', 'localtime'))
|
||
ON CONFLICT(href) DO UPDATE SET
|
||
title=excluded.title, is_full_data=excluded.is_full_data, updated_at = datetime('now', 'localtime')
|
||
""",
|
||
(title, href, is_full_data)
|
||
)
|
||
conn.commit()
|
||
|
||
# 获取插入的 movie_id
|
||
movie_id = get_id_by_href('javdb_movies', href)
|
||
if movie_id is None:
|
||
return None
|
||
|
||
return movie_id
|
||
|
||
except Exception as e:
|
||
conn.rollback()
|
||
logging.error("Error inserting movie: %s", e)
|
||
return None
|
||
|
||
|
||
# 删除电影数据"""
|
||
def delete_movie(identifier):
|
||
try:
|
||
if isinstance(identifier, int):
|
||
cursor.execute("DELETE FROM javdb_movies WHERE id = ?", (identifier,))
|
||
elif isinstance(identifier, str):
|
||
cursor.execute("DELETE FROM javdb_movies WHERE href = ?", (identifier,))
|
||
else:
|
||
logging.warning("无效的删除参数")
|
||
return
|
||
conn.commit()
|
||
logging.info(f"Deleted movie with {identifier}")
|
||
|
||
except sqlite3.Error as e:
|
||
conn.rollback()
|
||
logging.error("Error deleting movie: %s", e)
|
||
|
||
# 查找电影数据"""
|
||
def query_movies(identifier):
|
||
try:
|
||
if isinstance(identifier, int):
|
||
cursor.execute("SELECT * FROM javdb_movies WHERE id = ?", (identifier,))
|
||
elif "http" in identifier:
|
||
cursor.execute("SELECT * FROM javdb_movies WHERE href = ?", (identifier,))
|
||
else:
|
||
cursor.execute("SELECT * FROM javdb_movies WHERE title LIKE ?", (f"%{identifier}%",))
|
||
|
||
movie = cursor.fetchone()
|
||
if movie:
|
||
cursor.execute("SELECT * FROM javdb_actors_movies WHERE performer_id = ?", (movie[0],))
|
||
performers = [row[0] for row in cursor.fetchall()]
|
||
result = dict(zip([desc[0] for desc in cursor.description], performers))
|
||
result["performers"] = performers
|
||
return result
|
||
else:
|
||
logging.warning(f"find no data: {identifier}")
|
||
return None
|
||
|
||
except sqlite3.Error as e:
|
||
logging.error(f"查询失败: {e}")
|
||
return None
|
||
|
||
# 按条件查询 href 列表
|
||
def query_movie_hrefs(**filters):
|
||
try:
|
||
sql = "SELECT href, title FROM javdb_movies WHERE 1=1"
|
||
params = []
|
||
|
||
if "id" in filters:
|
||
sql += " AND id = ?"
|
||
params.append(filters["id"])
|
||
if "href" in filters:
|
||
sql += " AND href = ?"
|
||
params.append(filters["href"])
|
||
if "title" in filters:
|
||
sql += " AND title LIKE ?"
|
||
params.append(f"%{filters['title']}%")
|
||
if "is_full_data" in filters:
|
||
sql += " AND is_full_data = ?"
|
||
params.append(filters["is_full_data"])
|
||
if 'limit' in filters:
|
||
sql += " limit ?"
|
||
params.append(filters["limit"])
|
||
|
||
cursor.execute(sql, params)
|
||
#return [row[0].lower() for row in cursor.fetchall()] # 链接使用小写
|
||
return [{'href': row[0], 'title': row[1]} for row in cursor.fetchall()]
|
||
|
||
except sqlite3.Error as e:
|
||
logging.error(f"查询 href 失败: {e}")
|
||
return []
|
||
|
||
# 插入一条任务日志
|
||
def insert_task_log():
|
||
try:
|
||
cursor.execute("""
|
||
INSERT INTO javdb_task_log (task_status) VALUES ('Start')
|
||
""")
|
||
conn.commit()
|
||
|
||
task_id = cursor.lastrowid
|
||
if task_id is None:
|
||
return None
|
||
update_task_log(task_id=task_id, task_status='Start')
|
||
|
||
return task_id # 获取插入的 task_id
|
||
except sqlite3.Error as e:
|
||
logging.error(f"插入任务失败: {e}")
|
||
return None
|
||
|
||
# 更新任务日志的字段
|
||
def update_task_log_inner(task_id, **kwargs):
|
||
try:
|
||
fields = ", ".join(f"{key} = ?" for key in kwargs.keys())
|
||
params = list(kwargs.values()) + [task_id]
|
||
|
||
sql = f"UPDATE javdb_task_log SET {fields}, updated_at = datetime('now', 'localtime') WHERE task_id = ?"
|
||
cursor.execute(sql, params)
|
||
conn.commit()
|
||
except sqlite3.Error as e:
|
||
logging.error(f"更新任务 {task_id} 失败: {e}")
|
||
|
||
# 更新任务日志的字段
|
||
def update_task_log(task_id, task_status):
|
||
try:
|
||
# 获取 performers、studios 等表的最终行数
|
||
cursor.execute("SELECT COUNT(*) FROM javdb_actors where is_full_data=1")
|
||
full_data_actors = cursor.fetchone()[0]
|
||
cursor.execute("SELECT COUNT(*) FROM javdb_actors")
|
||
total_actors = cursor.fetchone()[0]
|
||
|
||
cursor.execute("SELECT COUNT(*) FROM javdb_movies where is_full_data=1")
|
||
full_data_movies = cursor.fetchone()[0]
|
||
cursor.execute("SELECT COUNT(*) FROM javdb_movies")
|
||
total_movies = cursor.fetchone()[0]
|
||
|
||
cursor.execute("SELECT COUNT(*) FROM javdb_makers")
|
||
total_makers = cursor.fetchone()[0]
|
||
|
||
cursor.execute("SELECT COUNT(*) FROM javdb_series")
|
||
total_series = cursor.fetchone()[0]
|
||
|
||
# 更新 task_log
|
||
update_task_log_inner(task_id,
|
||
full_data_actors=full_data_actors,
|
||
total_actors=total_actors,
|
||
full_data_movies=full_data_movies,
|
||
total_movies=total_movies,
|
||
total_makers=total_makers,
|
||
total_series=total_series,
|
||
task_status=task_status)
|
||
|
||
except sqlite3.Error as e:
|
||
logging.error(f"更新任务 {task_id} 失败: {e}")
|
||
|
||
|
||
# 任务结束，更新字段
|
||
def finalize_task_log(task_id):
|
||
try:
|
||
# 更新 task_log
|
||
update_task_log(task_id, task_status="Success")
|
||
except sqlite3.Error as e:
|
||
logging.error(f"任务 {task_id} 结束失败: {e}")
|
||
|
||
|
||
# 测试代码
|
||
if __name__ == "__main__":
|
||
|
||
sample_data = [
|
||
{
|
||
'name': '上原亜衣',
|
||
'href': 'https://www.javdb.com/actors/MkAX',
|
||
'pic': 'https://c0.jdbstatic.com/avatars/mk/MkAX.jpg',
|
||
'alias': ['上原亜衣', '下原舞', '早瀬クリスタル', '阿蘇山百式屏風奉行']
|
||
},
|
||
{
|
||
'name': '大橋未久',
|
||
'href': 'https://www.javdb.com/actors/21Jp',
|
||
'pic': 'https://c0.jdbstatic.com/avatars/21/21Jp.jpg',
|
||
'alias': ['大橋未久']
|
||
},
|
||
]
|
||
|
||
for actor in sample_data:
|
||
insert_or_update_actor(actor)
|
||
|
||
print(query_actors("name LIKE '%未久%'"))
|
||
#delete_actor_by_href('https://www.javdb.com/actors/MkAX')
|
||
print(query_actors())
|