webapp/app/services/fake_service.py

import random
from app.core.biz_exception import NotFoundError, PermissionDeniedError, BizLogicError
from typing import List
from datetime import date, datetime, timedelta
from app.schemas.fake import FakeUser

# ----------------------------------------
# 邮箱域名
# ----------------------------------------
DOMAINS = [
    "gmail-app.com",
    "outlooksearch.com",
    "hotmails.vip",
    "gmail365.cc",
    "ymails.top",
    "teamymail.cfd"
]

def get_email_prefix(email: str) -> str:
    if not email or "@" not in email:
        return ""
    return email.split("@", 1)[0]

def generate_stable_email(base: str) -> str:
    if not base or not base.strip():
        base = "user"
    index = abs(hash(base)) % len(DOMAINS)
    return f"{base}@{DOMAINS[index]}"

# ----------------------------------------
# 国籍对应名字和护照格式
# 全 ASCII
# ----------------------------------------
NATIONALITY_DATA = {
    "China": {
        "male_names": ["Wei", "Jian", "Liang", "Hao", "Jun"],
        "female_names": ["Mei", "Li", "Fang", "Xia", "Lan"],
        "surnames": ["Zhang", "Wang", "Li", "Liu", "Chen"],
        "passport_pattern": "E?########"
    },
    "India": {
        "male_names": ["Arjun", "Rohan", "Raj", "Vikram", "Siddharth"],
        "female_names": ["Ananya", "Priya", "Aisha", "Neha", "Kavya"],
        "surnames": ["Sharma", "Patel", "Singh", "Kumar", "Gupta"],
        "passport_pattern": "?#######"
    },
    # "United States": {
    #     "male_names": ["James", "John", "Robert", "Michael", "William"],
    #     "female_names": ["Mary", "Patricia", "Linda", "Elizabeth", "Jennifer"],
    #     "surnames": ["Smith", "Johnson", "Williams", "Brown", "Jones"],
    #     "passport_pattern": "#########"
    # },
    # "United Kingdom": {
    #     "male_names": ["Oliver", "George", "Harry", "Jack", "Jacob"],
    #     "female_names": ["Olivia", "Amelia", "Isla", "Ava", "Emily"],
    #     "surnames": ["Smith", "Jones", "Taylor", "Brown", "Williams"],
    #     "passport_pattern": "#########"
    # },
    # "Ireland": {
    #     "male_names": ["Conor", "Sean", "Patrick", "Liam", "Finn"],
    #     "female_names": ["Aoife", "Saoirse", "Emily", "Grace", "Ciara"],
    #     "surnames": ["Murphy", "Kelly", "OBrien", "Walsh", "Ryan"],
    #     "passport_pattern": "P########"
    # },
    # "Germany": {
    #     "male_names": ["Max", "Lukas", "Felix", "Leon", "Paul"],
    #     "female_names": ["Anna", "Emma", "Sophia", "Marie", "Laura"],
    #     "surnames": ["Mueller", "Schmidt", "Schneider", "Fischer", "Weber"],
    #     "passport_pattern": "C########"
    # },
    # "France": {
    #     "male_names": ["Louis", "Gabriel", "Raphael", "Arthur", "Jules"],
    #     "female_names": ["Emma", "Louise", "Alice", "Chloe", "Lina"],
    #     "surnames": ["Martin", "Bernard", "Thomas", "Robert", "Richard"],
    #     "passport_pattern": "########"
    # },
    # "Japan": {
    #     "male_names": ["Hiroshi", "Takumi", "Kenta", "Yuki", "Ryo"],
    #     "female_names": ["Yui", "Sakura", "Mao", "Aya", "Hana"],
    #     "surnames": ["Sato", "Suzuki", "Takahashi", "Tanaka", "Watanabe"],
    #     "passport_pattern": "TR#######"
    # }
}

# ----------------------------------------
# 居住国家地址/电话/邮编（全 ASCII）
# ----------------------------------------
RESIDENCE_DATA = {
    "China": {
        "cities": ["Beijing", "Shanghai", "Shenzhen", "Guangzhou", "Chengdu"],
        "streets": ["Heping Road", "Jiefang Road", "Renmin Street", "Zhongshan Avenue", "Fuxing Street"],
        "postcode_prefix": "1",
        "phone_code": "+86",
        "phone_length": 11
    },
    "India": {
        "cities": ["Delhi", "Mumbai", "Bangalore", "Kolkata", "Chennai"],
        "streets": ["MG Road", "Connaught Place", "Brigade Road", "Park Street", "Anna Salai"],
        "postcode_prefix": "5",
        "phone_code": "+91",
        "phone_length": 10
    },
    "United Kingdom": {
        "cities": ["London", "Manchester", "Birmingham", "Leeds", "Glasgow"],
        "streets": ["High Street", "Station Road", "Main Street", "Church Road", "London Road"],
        "postcode_prefix": "SW",
        "phone_code": "+44",
        "phone_length": 10
    },
    "Ireland": {
        "cities": ["Dublin", "Cork", "Galway", "Limerick", "Waterford"],
        "streets": ["OConnell Street", "Patrick Street", "Main Street", "High Street", "Bridge Street"],
        "postcode_prefix": "D",
        "phone_code": "+353",
        "phone_length": 9
    }
}

# ----------------------------------------
# 辅助函数
# ----------------------------------------
def generate_passport(pattern: str) -> str:
    s = ""
    for c in pattern:
        if c == "#":
            s += str(random.randint(0, 9))
        elif c == "?":
            s += random.choice("ABCDEFGHIJKLMNOPQRSTUVWXYZ")
        else:
            s += c
    return s

def generate_name(nationality: str, gender: str):
    data = NATIONALITY_DATA[nationality]
    if gender == "Male":
        first = random.choice(data["male_names"])
    else:
        first = random.choice(data["female_names"])
    last = random.choice(data["surnames"])
    return first, last

def generate_address(residence_country: str):
    data = RESIDENCE_DATA.get(residence_country)
    street = random.choice(data["streets"])
    city = random.choice(data["cities"])
    postcode = data["postcode_prefix"] + "".join([str(random.randint(0, 9)) for _ in range(5)])
    return f"{random.randint(1, 200)} {street}", city, postcode

# def generate_phone(residence_country: str):
#     data = RESIDENCE_DATA.get(residence_country)
#     num = "".join([str(random.randint(0,9)) for _ in range(data["phone_length"])])
#     return data["phone_code"], num

def generate_phone(country: str):
    if country == "China":
        prefix = "86"
        first_digit = "1"
        length = 11
    elif country == "India":
        prefix = "91"
        first_digit = random.choice(["7", "8", "9"])
        length = 10
    elif country == "Ireland":
        prefix = "353"
        first_digit = "8"
        length = 9
    elif country == "United Kingdom":
        prefix = "44"
        first_digit = "7"
        length = 10
    elif country == "United States":
        prefix = "1"
        first_digit = str(random.randint(2,9))
        length = 10
    elif country == "Germany":
        prefix = "49"
        first_digit = str(random.randint(1,9))
        length = 10
    else:
        prefix = "+00"
        first_digit = "1"
        length = 10
    
    # 生成国内号码（ASCII数字）
    number = first_digit + "".join(str(random.randint(0,9)) for _ in range(length-1))
    return prefix, '0' + number

# ----------------------------------------
# 主函数
# ----------------------------------------
def generate_fake_users(num: int, living_country: str = "Ireland") -> List[FakeUser]:
    users = []
    if living_country not in RESIDENCE_DATA.keys():
        raise BizLogicError(f'Living country only have {RESIDENCE_DATA.keys()}')
    for _ in range(num):
        gender = random.choice(["Male","Female"])
        nationality = random.choice(list(NATIONALITY_DATA.keys()))
        first_name, last_name = generate_name(nationality, gender)
        passport_no = generate_passport(NATIONALITY_DATA[nationality]["passport_pattern"])

        address_line1, city, postcode = generate_address(living_country)
        phone_country_code, phone_no = generate_phone(living_country)

        fake_email = f"{first_name.lower()}.{last_name.lower()}@example.com"
        alias_email = generate_stable_email(f"{first_name.lower()}.{last_name.lower()}")

        travel_date = date.today() + timedelta(days=random.randint(30,90))
        passport_expiry_date = date.today() + timedelta(days=random.randint(365,3650))
        birthday = date.today() - timedelta(days=random.randint(18*365,50*365))

        user = FakeUser(
            provider="fake",
            visa_center="fake",
            order_no=f"{random.randint(10000000,99999999)}",
            social_account=f"fake-{first_name.lower()}{random.randint(1,99)}",
            account=fake_email,
            password="Password123!",

            last_name=last_name,
            first_name=first_name,
            gender=gender,
            birthday=birthday,
            nationality=nationality,
            passport_no=passport_no,
            passport_expiry_date=passport_expiry_date,
            email=fake_email,
            alias_email=alias_email,

            address_line1=address_line1,
            address_line2="",
            city=city,
            state="",
            postcode=postcode,
            phone_country_code=phone_country_code,
            phone_no=phone_no,

            travel_date=travel_date,
            cover_letter="This is a test cover letter.",
            passport_image_url="https://placekitten.com/400/400",
            selfie_image_url="https://placekitten.com/401/401",
            application_form_url="https://example.com/form",

            priority=random.randint(1,5),
            expected_submit_start='2000-01-01',
            expected_submit_end='2100-01-01',
            rules="",
            status=0,
            placeholder=1,
            appointment_datetime=datetime.now() + timedelta(days=random.randint(20,200)),
            appointment_letter_url="https://example.com/appointment",
            pnr_number="",
            payment_link="",
            payment_help=1,
            note="This is a test note."
        )
        users.append(user)

    return users