406 lines
14 KiB
Python
406 lines
14 KiB
Python
import random
|
||
import time
|
||
from datetime import date, timedelta
|
||
from typing import Optional, Dict
|
||
|
||
import requests
|
||
from bit_browser import retry
|
||
|
||
|
||
CA_PROVINCE_ABBR = {
|
||
"Alberta": "AB",
|
||
"British Columbia": "BC",
|
||
"Manitoba": "MB",
|
||
"New Brunswick": "NB",
|
||
"Newfoundland and Labrador": "NL",
|
||
"Nova Scotia": "NS",
|
||
"Ontario": "ON",
|
||
"Prince Edward Island": "PE",
|
||
"Quebec": "QC",
|
||
"Saskatchewan": "SK",
|
||
"Northwest Territories": "NT",
|
||
"Nunavut": "NU",
|
||
"Yukon": "YT",
|
||
}
|
||
|
||
|
||
CA_COORDS = {
|
||
"AB": [(51.044733, -114.071883, "Calgary"), (53.546124, -113.493823, "Edmonton")],
|
||
"BC": [(49.282729, -123.120738, "Vancouver"), (48.428421, -123.365644, "Victoria")],
|
||
"MB": [(49.895137, -97.138374, "Winnipeg"), (50.445211, -96.823611, "East St Paul")],
|
||
"NB": [(45.963589, -66.643115, "Fredericton"), (46.510712, -67.255044, "Woodstock")],
|
||
"NL": [(53.135509, -57.660435, "Labrador City"), (47.561510, -52.712585, "St. John's")],
|
||
"NS": [(44.648862, -63.575320, "Halifax"), (45.010474, -63.416817, "Truro")],
|
||
"ON": [(43.653225, -79.383186, "Toronto"), (45.421532, -75.697189, "Ottawa")],
|
||
"PE": [(46.238240, -63.131074, "Charlottetown"), (46.392410, -63.787629, "Summerside")],
|
||
"QC": [(45.501689, -73.567256, "Montreal"), (46.813878, -71.207980, "Quebec City")],
|
||
"SK": [(52.133214, -106.670046, "Saskatoon"), (50.445211, -104.618896, "Regina")],
|
||
"NT": [(62.4540, -114.3725, "Yellowknife"), (61.251955, -114.352482, "Yellowknife")],
|
||
"NU": [(63.7467, -68.5167, "Iqaluit"), (64.282327, -76.614813, "Nunavut")],
|
||
"YT": [(60.7212, -135.0568, "Whitehorse"), (64.000000, -138.000000, "Yukon")],
|
||
}
|
||
|
||
|
||
CA_AREA_CODES = {
|
||
"AB": ["403", "587", "825"],
|
||
"BC": ["236", "250", "604", "672", "778"],
|
||
"MB": ["204", "431"],
|
||
"NB": ["506"],
|
||
"NL": ["709"],
|
||
"NS": ["782", "902"],
|
||
"ON": ["226", "249", "289", "343", "365", "416", "437", "519", "548", "613", "639", "647", "705", "807", "905"],
|
||
"PE": ["902"],
|
||
"QC": ["418", "438", "450", "514", "579", "581", "819", "873"],
|
||
"SK": ["306", "639"],
|
||
"NT": ["867"],
|
||
"NU": ["867"],
|
||
"YT": ["867"],
|
||
}
|
||
|
||
|
||
# 主要城市的区号（更精确的城市级约束）
|
||
CITY_AREA_CODES = {
|
||
"Calgary": ["403", "587", "825"],
|
||
"Edmonton": ["780", "587", "825"],
|
||
"Vancouver": ["604", "778", "236", "672"],
|
||
"Halifax": ["902", "782"],
|
||
"Toronto": ["416", "647", "437"],
|
||
}
|
||
|
||
|
||
# 邮编首字母合法性映射（按省份缩写）
|
||
POSTAL_PREFIXES = {
|
||
"AB": {"T"},
|
||
"BC": {"V"},
|
||
"MB": {"R"},
|
||
"NB": {"E"},
|
||
"NL": {"A"},
|
||
"NS": {"B"},
|
||
"ON": {"K", "L", "M"},
|
||
"PE": {"C"},
|
||
"QC": {"G", "H", "J"},
|
||
"SK": {"S"},
|
||
"NT": {"X"},
|
||
"NU": {"X"},
|
||
"YT": {"Y"},
|
||
}
|
||
|
||
|
||
REMOTE_PROVINCES = {"NL", "NT", "NU", "YT"}
|
||
|
||
|
||
def _normalize_province(province: str) -> str:
|
||
"""
|
||
省份入参规范化，支持全称或缩写，返回缩写
|
||
|
||
参数:
|
||
province (str): 省份，可为全称或缩写（如 "Alberta" 或 "AB"）
|
||
|
||
返回值:
|
||
str: 省份缩写（如 "AB"）
|
||
"""
|
||
if not province:
|
||
raise ValueError("province 不能为空")
|
||
p = province.strip()
|
||
if len(p) == 2:
|
||
return p.upper()
|
||
return CA_PROVINCE_ABBR.get(p, p)
|
||
|
||
|
||
def _pick_coords(province_abbr: str, city: Optional[str]) -> tuple[float, float, str]:
|
||
"""
|
||
按省份与可选城市选择一个坐标点
|
||
|
||
参数:
|
||
province_abbr (str): 省份缩写
|
||
city (Optional[str]): 城市名（如 "Calgary"），可为空
|
||
|
||
返回值:
|
||
(lat, lon, city_name): 选中的基础坐标及城市名
|
||
"""
|
||
coords = CA_COORDS.get(province_abbr)
|
||
if not coords:
|
||
# 默认回退至 Calgary
|
||
return 51.044733, -114.071883, "Calgary"
|
||
if city:
|
||
c = city.strip().lower()
|
||
for lat, lon, cname in coords:
|
||
if cname.lower() == c:
|
||
return lat, lon, cname
|
||
return random.choice(coords)
|
||
|
||
|
||
def _random_near(lat: float, lon: float) -> tuple[float, float]:
|
||
"""
|
||
在给定坐标附近生成一个随机偏移坐标
|
||
|
||
参数:
|
||
lat (float): 基准纬度
|
||
lon (float): 基准经度
|
||
|
||
返回值:
|
||
(new_lat, new_lon): 随机偏移后的坐标
|
||
"""
|
||
return lat + (random.random() - 0.5) * 0.1, lon + (random.random() - 0.5) * 0.1
|
||
|
||
|
||
@retry(max_retries=3, delay=1.0, backoff=1.0)
|
||
def _reverse_geocode(lat: float, lon: float) -> Dict:
|
||
"""
|
||
使用 Nominatim 反向地理编码，返回地址字典
|
||
|
||
参数:
|
||
lat (float): 纬度
|
||
lon (float): 经度
|
||
|
||
返回值:
|
||
dict: 包含 address 字段的响应数据
|
||
"""
|
||
url = f"https://nominatim.openstreetmap.org/reverse?format=json&lat={lat}&lon={lon}&zoom=18&addressdetails=1"
|
||
headers = {"User-Agent": "ca_auto_table/1.0"}
|
||
r = requests.get(url, headers=headers, timeout=15)
|
||
r.raise_for_status()
|
||
return r.json()
|
||
|
||
|
||
def _format_address(address: Dict, province_abbr: str) -> str:
|
||
"""
|
||
将 Nominatim 的 address 格式化为完整地址字符串
|
||
|
||
参数:
|
||
address (dict): Nominatim 返回的 address 字段
|
||
province_abbr (str): 省份缩写（如 "AB"）
|
||
|
||
返回值:
|
||
str: 格式化后的地址字符串
|
||
"""
|
||
house = address.get("house_number")
|
||
road = address.get("road") or address.get("residential") or address.get("footway")
|
||
city = address.get("city") or address.get("town") or address.get("village")
|
||
postcode = address.get("postcode") or ""
|
||
if house and road and city:
|
||
return f"{house} {road}, {city}, {province_abbr} {postcode}, Canada"
|
||
# 远端省份允许部分地址
|
||
return f"{city or ''}, {province_abbr} {postcode}, Canada".strip(", ")
|
||
|
||
|
||
def _random_name() -> tuple[str, str]:
|
||
"""
|
||
生成随机英文名（Firstname, Lastname），组合空间可达数百万以上
|
||
|
||
实现策略:
|
||
- 60% 概率使用常见英文名与姓氏列表（更自然）
|
||
- 40% 概率使用音节组合算法动态生成（数量级远超百万）
|
||
|
||
返回值:
|
||
(firstname, lastname)
|
||
"""
|
||
common_first = [
|
||
"James", "Mary", "Robert", "Patricia", "John", "Jennifer", "Michael", "Linda", "William", "Elizabeth",
|
||
"David", "Barbara", "Richard", "Susan", "Joseph", "Jessica", "Thomas", "Sarah", "Charles", "Karen",
|
||
"Christopher", "Nancy", "Daniel", "Lisa", "Matthew", "Betty", "Anthony", "Margaret", "Mark", "Sandra",
|
||
"Donald", "Ashley", "Steven", "Kimberly", "Paul", "Emily", "Andrew", "Donna", "Joshua", "Michelle",
|
||
"Kenneth", "Dorothy", "Kevin", "Carol", "Brian", "Amanda", "George", "Melissa", "Edward", "Deborah",
|
||
"Ronald", "Stephanie", "Timothy", "Rebecca", "Jason", "Laura", "Jeffrey", "Sharon", "Ryan", "Cynthia",
|
||
"Jacob", "Kathleen", "Gary", "Amy", "Nicholas", "Shirley", "Eric", "Angela", "Stephen", "Helen",
|
||
"Jonathan", "Anna", "Larry", "Brenda", "Justin", "Pamela", "Scott", "Nicole", "Brandon", "Samantha",
|
||
"Frank", "Katherine", "Benjamin", "Christine", "Gregory", "Emma", "Raymond", "Ruth", "Samuel", "Julie",
|
||
"Patrick", "Olivia", "Alexander", "Victoria"
|
||
]
|
||
common_last = [
|
||
"Smith", "Johnson", "Williams", "Brown", "Jones", "Garcia", "Miller", "Davis", "Rodriguez", "Martinez",
|
||
"Hernandez", "Lopez", "Gonzalez", "Wilson", "Anderson", "Thomas", "Taylor", "Moore", "Jackson", "Martin",
|
||
"Lee", "Perez", "Thompson", "White", "Harris", "Sanchez", "Clark", "Ramirez", "Lewis", "Robinson",
|
||
"Walker", "Young", "Allen", "King", "Wright", "Scott", "Torres", "Nguyen", "Hill", "Flores",
|
||
"Green", "Adams", "Nelson", "Baker", "Hall", "Rivera", "Campbell", "Mitchell", "Carter", "Roberts",
|
||
"Turner", "Phillips", "Parker", "Evans", "Edwards", "Collins", "Stewart", "Sanchez", "Morris", "Rogers",
|
||
"Reed", "Cook", "Morgan", "Bell", "Murphy", "Bailey", "Cooper", "Richardson", "Cox", "Howard",
|
||
"Ward", "Torres", "Peterson", "Gray", "Ramirez", "James", "Watson", "Brooks", "Kelly", "Sanders",
|
||
"Price", "Bennett", "Wood", "Barnes", "Ross", "Henderson", "Coleman", "Jenkins", "Perry", "Powell",
|
||
"Long", "Patterson", "Hughes", "Flores"
|
||
]
|
||
|
||
if random.random() < 0.6:
|
||
return random.choice(common_first), random.choice(common_last)
|
||
|
||
# 动态音节组合生成，支持数百万组合
|
||
f_beg = [
|
||
"al", "ben", "car", "dan", "el", "fran", "ge", "har", "isa", "jo", "ka", "li", "mar", "no",
|
||
"ol", "pa", "qui", "ra", "sa", "ta", "ul", "vi", "wil", "xa", "ya", "zo"
|
||
]
|
||
f_mid = [
|
||
"a", "e", "i", "o", "u", "ae", "ai", "ia", "ie", "oa", "ou"
|
||
]
|
||
f_end = [
|
||
"n", "ne", "na", "son", "ton", "la", "ra", "rie", "ry", "ley", "ly", "ah"
|
||
]
|
||
|
||
l_beg = [
|
||
"sm", "john", "dav", "wil", "and", "tho", "tay", "mo", "jack", "mar", "lee", "tho", "whi", "har",
|
||
"san", "cla", "ram", "lew", "rob", "walk", "young", "all", "king", "wri", "scott", "tor", "nguy",
|
||
"hil", "flo", "gre", "ada", "nel", "bak", "hal", "riv", "camp", "mit", "car", "rob"
|
||
]
|
||
l_mid = [
|
||
"a", "e", "i", "o", "u", "ar", "er", "or", "an", "en", "in", "on", "un"
|
||
]
|
||
l_suf = [
|
||
"son", "ton", "man", "ley", "ford", "wood", "well", "er", "ers", "ing", "s", "son", "es"
|
||
]
|
||
|
||
def build_name(beg, mid, end, syllables=(2, 3)) -> str:
|
||
parts = [random.choice(beg)]
|
||
for _ in range(random.choice(syllables) - 1):
|
||
parts.append(random.choice(mid))
|
||
parts.append(random.choice(end))
|
||
name = "".join(parts)
|
||
return name.capitalize()
|
||
|
||
first = build_name(f_beg, f_mid, f_end)
|
||
last = build_name(l_beg, l_mid, l_suf)
|
||
return first, last
|
||
|
||
|
||
def _random_birthday() -> str:
|
||
"""
|
||
生成随机生日，格式为 yyyy-mm-dd
|
||
|
||
返回值:
|
||
str: 生日字符串
|
||
"""
|
||
start = date(1950, 1, 1)
|
||
end = date(2000, 12, 31)
|
||
delta_days = (end - start).days
|
||
d = start + timedelta(days=random.randint(0, delta_days))
|
||
return f"{d.year}-{d.month:02d}-{d.day:02d}"
|
||
|
||
|
||
def _random_phone(province_abbr: str) -> str:
|
||
"""
|
||
生成随机加拿大电话号码，带区号
|
||
|
||
参数:
|
||
province_abbr (str): 省份缩写
|
||
|
||
返回值:
|
||
str: 电话，例如 "(403) 555-1234"
|
||
"""
|
||
codes = CA_AREA_CODES.get(province_abbr, ["000"])
|
||
area = random.choice(codes)
|
||
exchange = str(random.randint(200, 899)).zfill(3)
|
||
line = str(random.randint(1000, 9999)).zfill(4)
|
||
return f"({area}) {exchange}-{line}"
|
||
|
||
|
||
def _random_phone_city(province_abbr: str, city: Optional[str]) -> str:
|
||
"""
|
||
按城市优先选择区号，若城市未配置则回退到省份区号
|
||
|
||
参数:
|
||
province_abbr (str): 省份缩写
|
||
city (Optional[str]): 城市名
|
||
|
||
返回值:
|
||
str: 电话，例如 "(403) 555-1234"
|
||
"""
|
||
codes = None
|
||
if city:
|
||
codes = CITY_AREA_CODES.get(city)
|
||
codes = codes or CA_AREA_CODES.get(province_abbr, ["000"])
|
||
area = random.choice(codes)
|
||
exchange = str(random.randint(200, 899)).zfill(3)
|
||
line = str(random.randint(1000, 9999)).zfill(4)
|
||
return f"(#{area}) {exchange}-{line}".replace("#", "")
|
||
|
||
|
||
def _postal_valid_for_province(province_abbr: str, postcode: str) -> bool:
|
||
"""
|
||
校验邮编首字母是否符合省份规范
|
||
|
||
参数:
|
||
province_abbr (str): 省份缩写
|
||
postcode (str): 邮编字符串
|
||
|
||
返回值:
|
||
bool: 合法返回 True，否则 False
|
||
"""
|
||
if not postcode:
|
||
return False
|
||
prefixes = POSTAL_PREFIXES.get(province_abbr)
|
||
if not prefixes:
|
||
return True
|
||
return postcode[0].upper() in prefixes
|
||
|
||
|
||
def generate_canada_info(province: str, city: Optional[str] = None, max_attempts: int = 15, sleep_sec: float = 0.6) -> Dict[str, str]:
|
||
"""
|
||
随机生成加拿大个人与地址信息，可指定省份（全称或缩写）与可选城市
|
||
|
||
参数:
|
||
province (str): 省份（如 "Alberta" 或 "AB"）
|
||
city (Optional[str]): 城市（如 "Calgary"），不传则在省内随机
|
||
max_attempts (int): 反向地理编码最大尝试次数
|
||
sleep_sec (float): 每次失败后的等待秒数，用于尊重 Nominatim 频率限制
|
||
|
||
返回值:
|
||
dict: 包含 Firstname、Lastname、全名、生日、街道地址、城市、电话、邮编、州全称
|
||
"""
|
||
prov_abbr = _normalize_province(province)
|
||
base_lat, base_lon, chosen_city = _pick_coords(prov_abbr, city)
|
||
|
||
address_str = ""
|
||
city_name = ""
|
||
postcode = ""
|
||
for _ in range(max_attempts):
|
||
lat, lon = _random_near(base_lat, base_lon)
|
||
data = _reverse_geocode(lat, lon)
|
||
if not data:
|
||
time.sleep(sleep_sec)
|
||
continue
|
||
addr = data.get("address", {})
|
||
city_name = addr.get("city") or addr.get("town") or addr.get("village") or chosen_city
|
||
postcode = addr.get("postcode") or ""
|
||
address_str = _format_address(addr, prov_abbr)
|
||
if prov_abbr in REMOTE_PROVINCES:
|
||
break
|
||
if addr.get("house_number") and (addr.get("road") or addr.get("residential") or addr.get("footway")) and city_name and _postal_valid_for_province(prov_abbr, postcode):
|
||
break
|
||
time.sleep(sleep_sec)
|
||
|
||
firstname, lastname = _random_name()
|
||
full_name = f"{firstname} {lastname}"
|
||
birthday = _random_birthday()
|
||
phone = _random_phone_city(prov_abbr, city or chosen_city)
|
||
|
||
return {
|
||
"firstname": firstname,
|
||
"lastname": lastname,
|
||
"full_name": full_name,
|
||
"birthday": birthday,
|
||
"address_str": address_str.split(",")[0],
|
||
"city_name": city_name,
|
||
"phone": phone,
|
||
"postcode": postcode,
|
||
"province": next((k for k, v in CA_PROVINCE_ABBR.items() if v == prov_abbr), prov_abbr),
|
||
}
|
||
|
||
|
||
def get_random_canada_info(province, city) -> Dict[str, str]:
|
||
"""
|
||
随机生成加拿大个人与地址信息，省份随机选择，城市随机选择
|
||
|
||
返回值:
|
||
dict: 包含 Firstname、Lastname、全名、生日、街道地址、城市、电话、邮编、州全称
|
||
"""
|
||
return generate_canada_info(province, city)
|
||
|
||
|
||
def main() -> None:
|
||
"""
|
||
演示：生成 Alberta 省 Calgary 的随机信息；可修改为其他省/城市
|
||
"""
|
||
info = generate_canada_info("Alberta", "Calgary")
|
||
print(info)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main() |