#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import sqlite3

db_path = r'\\ll\D\xtrssvjj\db\dpd.db'
adid = 8440


def get_author(adname):
    """
    获取作者信息
    """
    # 正确的author数据库路径
    author_db_path = r"\\ll\d\wxvddata\dpd.db"
    try:
        conn = sqlite3.connect(author_db_path)
        cursor = conn.cursor()
        
        # 使用adname的前60个字符进行模糊匹配
        search_name = adname[:60]
        cursor.execute("SELECT author FROM ads WHERE name LIKE ? LIMIT 1", (f"%{search_name}%",))
        result = cursor.fetchone()
        conn.close()
        
        return result[0] if result else "未知"
        
    except Exception as e:
        print(f"获取作者信息失败: {e}")
        return "未知"


def generate_subtitle(author):
    """
    生成副标题
    """
    return f"译自@{author}"


def main():
    print("=== 获取虚伪的 长恨歌 副标题 ===")
    
    # 获取adname
    try:
        conn = sqlite3.connect(db_path)
        cursor = conn.cursor()
        
        cursor.execute("SELECT DISTINCT adname FROM srts WHERE adid = ?", (adid,))
        adname = cursor.fetchone()[0]
        print(f"adname: {adname}")
        
    except Exception as e:
        print(f"获取adname失败: {e}")
        adname = "虚伪的_长恨歌"
    finally:
        conn.close()
    
    # 获取作者
    author = get_author(adname)
    print(f"author: {author}")
    
    # 生成副标题
    subtitle = generate_subtitle(author)
    print(f"副标题: {subtitle}")


if __name__ == "__main__":
    main()