#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
获取"韩寒创业简史 中"的adid和author
"""

import sqlite3


def get_hanhan_middle_adid():
    """
    获取"韩寒创业简史 中"的adid
    """
    db_path = r"\\ll\D\xtrssvjj\db\dpd.db"
    try:
        conn = sqlite3.connect(db_path)
        cur = conn.cursor()
        
        # 查询包含"韩寒创业简史"和"中"的记录
        cur.execute("SELECT adid, adname FROM srts WHERE adname LIKE '%韩寒创业简史%' AND adname LIKE '%中%' LIMIT 1")
        result = cur.fetchone()
        conn.close()
        
        if result:
            adid, adname = result
            print(f"找到adid: {adid}")
            print(f"adname: {adname}")
            return adid, adname
        else:
            print("未找到韩寒创业简史 中")
            return None, None
    except Exception as e:
        print(f"数据库查询错误: {e}")
        return None, None


def get_hanhan_middle_author(adname):
    """
    从author数据库查询author
    """
    db_path = r"\\ll\d\wxvddata\dpd.db"
    try:
        conn = sqlite3.connect(db_path)
        cur = conn.cursor()
        
        # 使用adname的前60个字符进行模糊匹配
        query_name = adname[:60]
        cur.execute("SELECT author FROM ads WHERE name LIKE ? LIMIT 1", (f"%{query_name}%",))
        result = cur.fetchone()
        conn.close()
        
        if result:
            author = result[0]
            print(f"找到author: {author}")
            return author
        else:
            print("未找到author")
            return None
    except Exception as e:
        print(f"数据库查询错误: {e}")
        return None


if __name__ == "__main__":
    adid, adname = get_hanhan_middle_adid()
    if adid and adname:
        author = get_hanhan_middle_author(adname)
        if author:
            print(f"\n完整信息:")
            print(f"adid: {adid}")
            print(f"adname: {adname}")
            print(f"author: {author}")
        else:
            print("\n警告: author为空，将使用'未知'")
            print(f"adid: {adid}")
            print(f"adname: {adname}")