121 lines
3.9 KiB
Python
121 lines
3.9 KiB
Python
import requests
|
|
from loguru import logger
|
|
import csv
|
|
import os
|
|
import random
|
|
class Api:
|
|
def __init__(self) -> None:
|
|
# self.base_url = 'http://127.0.0.1:6060'
|
|
self.base_url = 'http://192.168.11.67:6060'
|
|
|
|
# 创建店铺
|
|
def create_shop(self, city: str, street: str, shop_name: str) -> dict:
|
|
url = f'{self.base_url}/country/shop'
|
|
item = {
|
|
'city': city,
|
|
'street': street,
|
|
'shop_name': shop_name,
|
|
}
|
|
response = requests.post(url, json=item).json()
|
|
logger.info(response)
|
|
return response
|
|
|
|
# 查询店铺
|
|
def get_shop(self, city: str) -> dict:
|
|
url = f'{self.base_url}/country/shop'
|
|
response = requests.get(url).json()
|
|
# logger.info(response)
|
|
return response
|
|
|
|
# 创建信息
|
|
def create_info(self, child_full_name: str, parent_full_name: str, child_birthday: str, address_str: str, city_name: str, parent_phone: str, postcode: str, province: str, email: str, text: str, status: bool = False, email_content: str | None = None) -> dict:
|
|
"""
|
|
创建信息记录（孩子与家长字段）
|
|
|
|
参数:
|
|
child_full_name (str): 孩子全名
|
|
parent_full_name (str): 家长全名
|
|
child_birthday (str): 孩子生日（字符串）
|
|
address_str (str): 街道地址
|
|
city_name (str): 城市
|
|
parent_phone (str): 家长电话
|
|
postcode (str): 邮编
|
|
province (str): 省/州全称
|
|
email (str): 邮箱
|
|
text (str): 文本内容（如反馈地址）
|
|
status (bool): 状态
|
|
email_content (str | None): 邮件内容
|
|
|
|
返回值:
|
|
dict: 接口返回的数据
|
|
"""
|
|
url = f'{self.base_url}/country/info'
|
|
item = {
|
|
"child_full_name": child_full_name,
|
|
"parent_full_name": parent_full_name,
|
|
"child_birthday": child_birthday,
|
|
"address_str": address_str,
|
|
"city_name": city_name,
|
|
"parent_phone": parent_phone,
|
|
"postcode": postcode,
|
|
"province": province,
|
|
"status": status,
|
|
"email": email,
|
|
"email_content": email_content,
|
|
"text": text
|
|
}
|
|
response = requests.post(url, json=item).json()
|
|
logger.info(response)
|
|
return response
|
|
|
|
# 根据城市 随机获取一个店铺
|
|
def get_random_shop(self) -> dict:
|
|
url = f'{self.base_url}/country/shop/random'
|
|
response = requests.get(url).json()
|
|
# logger.info(response)
|
|
if not response.get('street'):
|
|
logger.error(f'没有店铺')
|
|
return None
|
|
return response
|
|
|
|
def main():
|
|
"""
|
|
从同目录的 `bakeries.csv` 读取面包店数据，按列映射输出或创建店铺
|
|
|
|
列顺序：`Name,Address,City`
|
|
"""
|
|
api = Api()
|
|
csv_path = os.path.join(os.path.dirname(__file__), 'data.csv')
|
|
if not os.path.exists(csv_path):
|
|
logger.error(f'CSV 文件不存在: {csv_path}')
|
|
return
|
|
|
|
with open(csv_path, 'r', encoding='utf-8') as file:
|
|
reader = csv.reader(file)
|
|
header = next(reader, None)
|
|
for row in reader:
|
|
if len(row) < 3:
|
|
logger.warning(f'行列数不足，跳过: {row}')
|
|
continue
|
|
shop_name, street, city = row[1], row[2], row[0]
|
|
if ' (city)' in city:
|
|
city = city.replace(' (city)', '')
|
|
if 'Quebec' in city:
|
|
continue
|
|
if ',' in city:
|
|
city = city.split(',')[0]
|
|
logger.info(f'city: {city}, street: {street}, shop_name: {shop_name}')
|
|
api.create_shop(city, street, shop_name)
|
|
|
|
# def main2():
|
|
# api = Api()
|
|
# city = 'Toronto'
|
|
# shop = api.get_random_shop()
|
|
# if shop:
|
|
# logger.info(shop)
|
|
|
|
# if __name__ == '__main__':
|
|
# main()
|
|
|
|
api = Api()
|