"""
股票名称映射数据库
包含常见A股股票的名称到代码的映射
"""
from typing import Optional

# 常见A股股票名称映射（按行业分类）
STOCK_NAME_MAP = {
    # 白酒
    "贵州茅台": "600519",
    "茅台": "600519",
    "五粮液": "000858",
    "泸州老窖": "000568",
    "山西汾酒": "600809",
    "洋河股份": "002304",

    # 银行
    "工商银行": "601398",
    "工行": "601398",
    "建设银行": "601939",
    "建行": "601939",
    "农业银行": "601288",
    "农行": "601288",
    "中国银行": "601988",
    "中行": "601988",
    "交通银行": "601328",
    "交行": "601328",
    "招商银行": "600036",
    "招行": "600036",
    "兴业银行": "601166",
    "浦发银行": "600000",
    "民生银行": "600016",
    "光大银行": "601818",
    "平安银行": "000001",
    "宁波银行": "002142",

    # 保险
    "中国平安": "601318",
    "平安": "601318",
    "中国人寿": "601628",
    "中国太保": "601601",
    "新华保险": "601336",

    # 证券
    "中信证券": "600030",
    "中信": "600030",
    "海通证券": "600837",
    "国泰君安": "601211",
    "华泰证券": "601688",
    "广发证券": "000776",
    "招商证券": "600999",
    "东方证券": "600958",

    # 科技
    "中兴通讯": "000063",
    "中兴": "000063",
    "立讯精密": "002475",
    "京东方A": "000725",
    "京东方": "000725",
    "TCL科技": "000100",
    "海康威视": "002415",
    "大华股份": "002236",
    "科大讯飞": "002230",
    "讯飞": "002230",
    "紫光国微": "002049",
    "中芯国际": "688981",
    "韦尔股份": "603501",

    # 新能源汽车
    "比亚迪": "002594",
    "宁德时代": "300750",
    "宁德": "300750",
    "长城汽车": "601633",
    "长城": "601633",
    "上汽集团": "600104",
    "上汽": "600104",
    "广汽集团": "601238",
    "广汽": "601238",
    "吉利汽车": "00175",  # 港股
    "理想汽车": "02015",  # 港股
    "小鹏汽车": "09868",  # 港股
    "蔚来": "09866",  # 港股

    # 医药
    "恒瑞医药": "600276",
    "恒瑞": "600276",
    "药明康德": "603259",
    "迈瑞医疗": "300760",
    "迈瑞": "300760",
    "片仔癀": "600436",
    "云南白药": "000538",
    "白药": "000538",
    "爱尔眼科": "300015",
    "智飞生物": "300122",

    # 消费
    "伊利股份": "600887",
    "伊利": "600887",
    "海天味业": "603288",
    "海天": "603288",
    "格力电器": "000651",
    "格力": "000651",
    "美的集团": "000333",
    "美的": "000333",
    "海尔智家": "600690",
    "海尔": "600690",
    "老板电器": "002508",

    # 地产
    "万科A": "000002",
    "万科": "000002",
    "保利发展": "600048",
    "保利": "600048",
    "招商蛇口": "001979",
    "金地集团": "600383",
    "金地": "600383",

    # 能源
    "中国石油": "601857",
    "中石油": "601857",
    "中国石化": "600028",
    "中石化": "600028",
    "中国神华": "601088",
    "神华": "601088",
    "陕西煤业": "601225",
    "长江电力": "600900",
    "三峡能源": "600905",

    # 通信
    "中国移动": "600941",
    "移动": "600941",
    "中国电信": "601728",
    "电信": "601728",
    "中国联通": "600050",
    "联通": "600050",
    "中国卫通": "601698",
    "卫通": "601698",

    # 航空航天
    "中国国航": "601111",
    "国航": "601111",
    "南方航空": "600029",
    "南航": "600029",
    "东方航空": "600115",
    "东航": "600115",
    "中国卫星": "600118",
    "航天科技": "000901",

    # 钢铁
    "宝钢股份": "600019",
    "宝钢": "600019",
    "河钢股份": "000709",
    "河钢": "000709",
    "鞍钢股份": "000898",
    "鞍钢": "000898",

    # 有色金属
    "紫金矿业": "601899",
    "紫金": "601899",
    "中国铝业": "601600",
    "中铝": "601600",
    "江西铜业": "600362",
    "江铜": "600362",
    "洛阳钼业": "603993",

    # 化工
    "万华化学": "600309",
    "万华": "600309",
    "华鲁恒升": "600426",
    "恒力石化": "600346",
    "荣盛石化": "002493",

    # 电力设备
    "隆基绿能": "601012",
    "隆基": "601012",
    "阳光电源": "300274",
    "通威股份": "600438",
    "通威": "600438",
    "特变电工": "600089",

    # 军工
    "中航沈飞": "600760",
    "沈飞": "600760",
    "中航西飞": "000768",
    "西飞": "000768",
    "中国船舶": "600150",
    "中船": "600150",
    "航发动力": "600893",
    "航天发展": "000547",

    # 互联网
    "腾讯控股": "00700",  # 港股
    "腾讯": "00700",
    "阿里巴巴": "09988",  # 港股
    "阿里": "09988",
    "美团": "03690",  # 港股
    "京东": "09618",  # 港股
    "拼多多": "PDD",  # 美股
    "百度": "09888",  # 港股
    "网易": "09999",  # 港股
    "小米集团": "01810",  # 港股
    "小米": "01810",

    # 指数
    "上证指数": "000001",
    "上证": "000001",
    "沪指": "000001",
    "深证成指": "399001",
    "深成指": "399001",
    "创业板指": "399006",
    "创业板": "399006",
    "科创50": "000688",
    "沪深300": "000300",
    "中证500": "000905",
    "中证1000": "000852",
}


def search_stock_by_name(name: str) -> Optional[str]:
    """
    根据股票名称搜索代码

    Args:
        name: 股票名称或简称

    Returns:
        股票代码或None
    """
    # 精确匹配
    if name in STOCK_NAME_MAP:
        return STOCK_NAME_MAP[name]

    # 模糊匹配（包含关系）
    for stock_name, code in STOCK_NAME_MAP.items():
        if name in stock_name or stock_name in name:
            return code

    return None


def get_stock_name(code: str) -> Optional[str]:
    """
    根据代码获取股票名称

    Args:
        code: 股票代码

    Returns:
        股票名称或None
    """
    for name, stock_code in STOCK_NAME_MAP.items():
        if stock_code == code:
            return name
    return None