281 lines
12 KiB
Python
281 lines
12 KiB
Python
import re
|
||
from typing import Any, Optional
|
||
|
||
|
||
COUNTRY_ENTRIES = [
|
||
("阿富汗", ["Afghanistan", "AF", "AFG"]),
|
||
("阿尔巴尼亚", ["Albania", "AL", "ALB"]),
|
||
("阿尔及利亚", ["Algeria", "DZ", "DZA"]),
|
||
("安道尔", ["Andorra", "AD", "AND"]),
|
||
("安哥拉", ["Angola", "AO", "AGO"]),
|
||
("安提瓜和巴布达", ["Antigua and Barbuda", "AG", "ATG"]),
|
||
("阿根廷", ["Argentina", "AR", "ARG"]),
|
||
("亚美尼亚", ["Armenia", "AM", "ARM"]),
|
||
("澳大利亚", ["Australia", "AU", "AUS"]),
|
||
("奥地利", ["Austria", "AT", "AUT"]),
|
||
("阿塞拜疆", ["Azerbaijan", "AZ", "AZE"]),
|
||
("巴哈马", ["Bahamas", "BS", "BHS"]),
|
||
("巴林", ["Bahrain", "BH", "BHR"]),
|
||
("孟加拉国", ["Bangladesh", "BD", "BGD"]),
|
||
("巴巴多斯", ["Barbados", "BB", "BRB"]),
|
||
("白俄罗斯", ["Belarus", "BY", "BLR"]),
|
||
("比利时", ["Belgium", "BE", "BEL"]),
|
||
("伯利兹", ["Belize", "BZ", "BLZ"]),
|
||
("贝宁", ["Benin", "BJ", "BEN"]),
|
||
("不丹", ["Bhutan", "BT", "BTN"]),
|
||
("玻利维亚", ["Bolivia", "BO", "BOL", "Bolivia (Plurinational State of)"]),
|
||
("波斯尼亚和黑塞哥维那", ["Bosnia and Herzegovina", "BA", "BIH"]),
|
||
("博茨瓦纳", ["Botswana", "BW", "BWA"]),
|
||
("巴西", ["Brazil", "BR", "BRA"]),
|
||
("文莱", ["Brunei", "BN", "BRN", "Brunei Darussalam"]),
|
||
("保加利亚", ["Bulgaria", "BG", "BGR"]),
|
||
("布基纳法索", ["Burkina Faso", "BF", "BFA"]),
|
||
("布隆迪", ["Burundi", "BI", "BDI"]),
|
||
("柬埔寨", ["Cambodia", "KH", "KHM"]),
|
||
("喀麦隆", ["Cameroon", "CM", "CMR"]),
|
||
("加拿大", ["Canada", "CA", "CAN"]),
|
||
("佛得角", ["Cape Verde", "CV", "CPV", "Cabo Verde"]),
|
||
("中非", ["Central African Republic", "CF", "CAF"]),
|
||
("乍得", ["Chad", "TD", "TCD"]),
|
||
("智利", ["Chile", "CL", "CHL"]),
|
||
("中国", ["China", "CN", "CHN", "Mainland China", "PRC", "People's Republic of China"]),
|
||
("中国（香港）", ["Hong Kong", "HK", "HKG", "Hong Kong SAR", "China Hong Kong", "Hong Kong, China"]),
|
||
("中国（澳门）", ["Macao", "Macau", "MO", "MAC", "Macao SAR", "China Macao", "Macau, China"]),
|
||
("中国（台湾）", ["Taiwan", "TW", "TWN", "Chinese Taipei", "Taiwan, China"]),
|
||
("哥伦比亚", ["Colombia", "CO", "COL"]),
|
||
("科摩罗", ["Comoros", "KM", "COM"]),
|
||
("刚果（布）", ["Republic of the Congo", "Congo", "Congo-Brazzaville", "CG", "COG"]),
|
||
("刚果（金）", ["Democratic Republic of the Congo", "DR Congo", "Congo-Kinshasa", "CD", "COD"]),
|
||
("哥斯达黎加", ["Costa Rica", "CR", "CRI"]),
|
||
("科特迪瓦", ["Cote d'Ivoire", "Côte d'Ivoire", "Ivory Coast", "CI", "CIV"]),
|
||
("克罗地亚", ["Croatia", "HR", "HRV"]),
|
||
("古巴", ["Cuba", "CU", "CUB"]),
|
||
("塞浦路斯", ["Cyprus", "CY", "CYP"]),
|
||
("捷克", ["Czech Republic", "Czechia", "CZ", "CZE"]),
|
||
("丹麦", ["Denmark", "DK", "DNK"]),
|
||
("吉布提", ["Djibouti", "DJ", "DJI"]),
|
||
("多米尼克", ["Dominica", "DM", "DMA"]),
|
||
("多米尼加", ["Dominican Republic", "DO", "DOM"]),
|
||
("厄瓜多尔", ["Ecuador", "EC", "ECU"]),
|
||
("埃及", ["Egypt", "EG", "EGY"]),
|
||
("萨尔瓦多", ["El Salvador", "SV", "SLV"]),
|
||
("赤道几内亚", ["Equatorial Guinea", "GQ", "GNQ"]),
|
||
("厄立特里亚", ["Eritrea", "ER", "ERI"]),
|
||
("爱沙尼亚", ["Estonia", "EE", "EST"]),
|
||
("埃斯瓦蒂尼", ["Eswatini", "SZ", "SWZ", "Swaziland"]),
|
||
("埃塞俄比亚", ["Ethiopia", "ET", "ETH"]),
|
||
("斐济", ["Fiji", "FJ", "FJI"]),
|
||
("芬兰", ["Finland", "FI", "FIN"]),
|
||
("法国", ["France", "FR", "FRA"]),
|
||
("加蓬", ["Gabon", "GA", "GAB"]),
|
||
("冈比亚", ["Gambia", "GM", "GMB"]),
|
||
("格鲁吉亚", ["Georgia", "GE", "GEO"]),
|
||
("德国", ["Germany", "DE", "DEU"]),
|
||
("加纳", ["Ghana", "GH", "GHA"]),
|
||
("希腊", ["Greece", "GR", "GRC"]),
|
||
("格林纳达", ["Grenada", "GD", "GRD"]),
|
||
("危地马拉", ["Guatemala", "GT", "GTM"]),
|
||
("几内亚", ["Guinea", "GN", "GIN"]),
|
||
("几内亚比绍", ["Guinea-Bissau", "GW", "GNB"]),
|
||
("圭亚那", ["Guyana", "GY", "GUY"]),
|
||
("海地", ["Haiti", "HT", "HTI"]),
|
||
("洪都拉斯", ["Honduras", "HN", "HND"]),
|
||
("匈牙利", ["Hungary", "HU", "HUN"]),
|
||
("冰岛", ["Iceland", "IS", "ISL"]),
|
||
("印度", ["India", "IN", "IND"]),
|
||
("印度尼西亚", ["Indonesia", "ID", "IDN"]),
|
||
("伊朗", ["Iran", "IR", "IRN", "Iran (Islamic Republic of)"]),
|
||
("伊拉克", ["Iraq", "IQ", "IRQ"]),
|
||
("爱尔兰", ["Ireland", "IE", "IRL"]),
|
||
("以色列", ["Israel", "IL", "ISR"]),
|
||
("意大利", ["Italy", "IT", "ITA"]),
|
||
("牙买加", ["Jamaica", "JM", "JAM"]),
|
||
("日本", ["Japan", "JP", "JPN"]),
|
||
("约旦", ["Jordan", "JO", "JOR"]),
|
||
("哈萨克斯坦", ["Kazakhstan", "KZ", "KAZ"]),
|
||
("肯尼亚", ["Kenya", "KE", "KEN"]),
|
||
("基里巴斯", ["Kiribati", "KI", "KIR"]),
|
||
("朝鲜", ["North Korea", "Korea, DPRK", "Democratic People's Republic of Korea", "KP", "PRK"]),
|
||
("韩国", ["South Korea", "Republic of Korea", "Korea", "KR", "KOR"]),
|
||
("科威特", ["Kuwait", "KW", "KWT"]),
|
||
("吉尔吉斯斯坦", ["Kyrgyzstan", "KG", "KGZ"]),
|
||
("老挝", ["Laos", "Lao PDR", "Lao People's Democratic Republic", "LA", "LAO"]),
|
||
("拉脱维亚", ["Latvia", "LV", "LVA"]),
|
||
("黎巴嫩", ["Lebanon", "LB", "LBN"]),
|
||
("莱索托", ["Lesotho", "LS", "LSO"]),
|
||
("利比里亚", ["Liberia", "LR", "LBR"]),
|
||
("利比亚", ["Libya", "LY", "LBY"]),
|
||
("列支敦士登", ["Liechtenstein", "LI", "LIE"]),
|
||
("立陶宛", ["Lithuania", "LT", "LTU"]),
|
||
("卢森堡", ["Luxembourg", "LU", "LUX"]),
|
||
("马达加斯加", ["Madagascar", "MG", "MDG"]),
|
||
("马拉维", ["Malawi", "MW", "MWI"]),
|
||
("马来西亚", ["Malaysia", "MY", "MYS"]),
|
||
("马尔代夫", ["Maldives", "MV", "MDV"]),
|
||
("马里", ["Mali", "ML", "MLI"]),
|
||
("马耳他", ["Malta", "MT", "MLT"]),
|
||
("马绍尔群岛", ["Marshall Islands", "MH", "MHL"]),
|
||
("毛里塔尼亚", ["Mauritania", "MR", "MRT"]),
|
||
("毛里求斯", ["Mauritius", "MU", "MUS"]),
|
||
("墨西哥", ["Mexico", "MX", "MEX"]),
|
||
("密克罗尼西亚", ["Micronesia", "FM", "FSM", "Federated States of Micronesia"]),
|
||
("摩尔多瓦", ["Moldova", "MD", "MDA", "Republic of Moldova"]),
|
||
("摩纳哥", ["Monaco", "MC", "MCO"]),
|
||
("蒙古", ["Mongolia", "MN", "MNG"]),
|
||
("黑山", ["Montenegro", "ME", "MNE"]),
|
||
("摩洛哥", ["Morocco", "MA", "MAR"]),
|
||
("莫桑比克", ["Mozambique", "MZ", "MOZ"]),
|
||
("缅甸", ["Myanmar", "MM", "MMR", "Burma"]),
|
||
("纳米比亚", ["Namibia", "NA", "NAM"]),
|
||
("瑙鲁", ["Nauru", "NR", "NRU"]),
|
||
("尼泊尔", ["Nepal", "NP", "NPL"]),
|
||
("荷兰", ["Netherlands", "NL", "NLD"]),
|
||
("新西兰", ["New Zealand", "NZ", "NZL"]),
|
||
("尼加拉瓜", ["Nicaragua", "NI", "NIC"]),
|
||
("尼日尔", ["Niger", "NE", "NER"]),
|
||
("尼日利亚", ["Nigeria", "NG", "NGA"]),
|
||
("北马其顿", ["North Macedonia", "MK", "MKD", "Macedonia"]),
|
||
("挪威", ["Norway", "NO", "NOR"]),
|
||
("阿曼", ["Oman", "OM", "OMN"]),
|
||
("巴基斯坦", ["Pakistan", "PK", "PAK"]),
|
||
("帕劳", ["Palau", "PW", "PLW"]),
|
||
("巴勒斯坦", ["Palestine", "PS", "PSE", "State of Palestine"]),
|
||
("巴拿马", ["Panama", "PA", "PAN"]),
|
||
("巴布亚新几内亚", ["Papua New Guinea", "PG", "PNG"]),
|
||
("巴拉圭", ["Paraguay", "PY", "PRY"]),
|
||
("秘鲁", ["Peru", "PE", "PER"]),
|
||
("菲律宾", ["Philippines", "PH", "PHL"]),
|
||
("波兰", ["Poland", "PL", "POL"]),
|
||
("葡萄牙", ["Portugal", "PT", "PRT"]),
|
||
("卡塔尔", ["Qatar", "QA", "QAT"]),
|
||
("罗马尼亚", ["Romania", "RO", "ROU"]),
|
||
("俄罗斯", ["Russia", "Russian Federation", "RU", "RUS"]),
|
||
("卢旺达", ["Rwanda", "RW", "RWA"]),
|
||
("圣基茨和尼维斯", ["Saint Kitts and Nevis", "KN", "KNA"]),
|
||
("圣卢西亚", ["Saint Lucia", "LC", "LCA"]),
|
||
("圣文森特和格林纳丁斯", ["Saint Vincent and the Grenadines", "VC", "VCT"]),
|
||
("萨摩亚", ["Samoa", "WS", "WSM"]),
|
||
("圣马力诺", ["San Marino", "SM", "SMR"]),
|
||
("圣多美和普林西比", ["Sao Tome and Principe", "ST", "STP", "São Tomé and Príncipe"]),
|
||
("沙特阿拉伯", ["Saudi Arabia", "SA", "SAU"]),
|
||
("塞内加尔", ["Senegal", "SN", "SEN"]),
|
||
("塞尔维亚", ["Serbia", "RS", "SRB", "Kosovo", "XK", "XKS", "Republic of Kosovo"]),
|
||
("塞舌尔", ["Seychelles", "SC", "SYC"]),
|
||
("塞拉利昂", ["Sierra Leone", "SL", "SLE"]),
|
||
("新加坡", ["Singapore", "SG", "SGP"]),
|
||
("斯洛伐克", ["Slovakia", "SK", "SVK"]),
|
||
("斯洛文尼亚", ["Slovenia", "SI", "SVN"]),
|
||
("所罗门群岛", ["Solomon Islands", "SB", "SLB"]),
|
||
("索马里", ["Somalia", "SO", "SOM"]),
|
||
("南非", ["South Africa", "ZA", "ZAF"]),
|
||
("南苏丹", ["South Sudan", "SS", "SSD"]),
|
||
("西班牙", ["Spain", "ES", "ESP"]),
|
||
("斯里兰卡", ["Sri Lanka", "LK", "LKA"]),
|
||
("苏丹", ["Sudan", "SD", "SDN"]),
|
||
("苏里南", ["Suriname", "SR", "SUR"]),
|
||
("瑞典", ["Sweden", "SE", "SWE"]),
|
||
("瑞士", ["Switzerland", "CH", "CHE"]),
|
||
("叙利亚", ["Syria", "SY", "SYR", "Syrian Arab Republic"]),
|
||
("塔吉克斯坦", ["Tajikistan", "TJ", "TJK"]),
|
||
("坦桑尼亚", ["Tanzania", "TZ", "TZA", "United Republic of Tanzania"]),
|
||
("泰国", ["Thailand", "TH", "THA"]),
|
||
("东帝汶", ["Timor-Leste", "East Timor", "TL", "TLS"]),
|
||
("多哥", ["Togo", "TG", "TGO"]),
|
||
("汤加", ["Tonga", "TO", "TON"]),
|
||
("特立尼达和多巴哥", ["Trinidad and Tobago", "TT", "TTO"]),
|
||
("突尼斯", ["Tunisia", "TN", "TUN"]),
|
||
("土耳其", ["Turkey", "TR", "TUR", "Türkiye"]),
|
||
("土库曼斯坦", ["Turkmenistan", "TM", "TKM"]),
|
||
("图瓦卢", ["Tuvalu", "TV", "TUV"]),
|
||
("乌干达", ["Uganda", "UG", "UGA"]),
|
||
("乌克兰", ["Ukraine", "UA", "UKR"]),
|
||
("阿联酋", ["United Arab Emirates", "AE", "ARE", "UAE"]),
|
||
("英国", ["United Kingdom", "UK", "GB", "GBR", "Great Britain", "Britain", "England"]),
|
||
("美国", ["United States", "United States of America", "US", "USA", "U.S.", "U.S.A."]),
|
||
("乌拉圭", ["Uruguay", "UY", "URY"]),
|
||
("乌兹别克斯坦", ["Uzbekistan", "UZ", "UZB"]),
|
||
("瓦努阿图", ["Vanuatu", "VU", "VUT"]),
|
||
("梵蒂冈", ["Vatican City", "Holy See", "VA", "VAT"]),
|
||
("委内瑞拉", ["Venezuela", "VE", "VEN", "Venezuela (Bolivarian Republic of)"]),
|
||
("越南", ["Vietnam", "Viet Nam", "VN", "VNM"]),
|
||
("也门", ["Yemen", "YE", "YEM"]),
|
||
("赞比亚", ["Zambia", "ZM", "ZMB"]),
|
||
("津巴布韦", ["Zimbabwe", "ZW", "ZWE"]),
|
||
]
|
||
|
||
|
||
COUNTRY_OPTIONS = [entry[0] for entry in COUNTRY_ENTRIES]
|
||
CANONICAL_COUNTRY_SET = set(COUNTRY_OPTIONS)
|
||
INVALID_COUNTRY_VALUES = {
|
||
"",
|
||
"-",
|
||
"--",
|
||
"unknown",
|
||
"n/a",
|
||
"na",
|
||
"none",
|
||
"null",
|
||
"global",
|
||
"world",
|
||
"worldwide",
|
||
"xx",
|
||
}
|
||
NUMERIC_LIKE_PATTERN = re.compile(r"^[\d\s,._%+\-]+$")
|
||
|
||
COUNTRY_ALIAS_MAP = {}
|
||
COUNTRY_VARIANTS_MAP = {}
|
||
for canonical, aliases in COUNTRY_ENTRIES:
|
||
COUNTRY_ALIAS_MAP[canonical.casefold()] = canonical
|
||
variants = [canonical, *aliases]
|
||
COUNTRY_VARIANTS_MAP[canonical] = variants
|
||
for alias in aliases:
|
||
COUNTRY_ALIAS_MAP[alias.casefold()] = canonical
|
||
|
||
|
||
def normalize_country(value: Any) -> Optional[str]:
|
||
if value is None:
|
||
return None
|
||
|
||
if not isinstance(value, str):
|
||
return None
|
||
|
||
normalized = re.sub(r"\s+", " ", value.strip())
|
||
normalized = normalized.replace("(", "（").replace(")", "）")
|
||
|
||
if not normalized:
|
||
return None
|
||
|
||
lowered = normalized.casefold()
|
||
if lowered in INVALID_COUNTRY_VALUES:
|
||
return None
|
||
|
||
if NUMERIC_LIKE_PATTERN.fullmatch(normalized):
|
||
return None
|
||
|
||
if normalized in CANONICAL_COUNTRY_SET:
|
||
return normalized
|
||
|
||
return COUNTRY_ALIAS_MAP.get(lowered)
|
||
|
||
|
||
def get_country_search_variants(value: Any) -> list[str]:
|
||
canonical = normalize_country(value)
|
||
if canonical is None:
|
||
return []
|
||
|
||
variants = []
|
||
seen = set()
|
||
for item in COUNTRY_VARIANTS_MAP.get(canonical, [canonical]):
|
||
if not isinstance(item, str):
|
||
continue
|
||
normalized = re.sub(r"\s+", " ", item.strip())
|
||
if not normalized:
|
||
continue
|
||
key = normalized.casefold()
|
||
if key in seen:
|
||
continue
|
||
seen.add(key)
|
||
variants.append(normalized)
|
||
|
||
return variants
|