def generate_phone(cls, reg_name, pwd='12345678', reg_type=1):
hr = HandleRequest()
hm = HandleMysql()
phone = hm.get_no_exist_phone()
header = hy.read_yaml('request', 'header')
hr.common_head(header)
data = {
"mobile_phone": phone,
"pwd": pwd,
'type': reg_type,
'reg_name': reg_name
}
value_result = hr.send(url=hy.read_yaml('register', 'url'), data=data)
if value_result.json()['code'] == 0:
result_phone = value_result.json()['data']['mobile_phone']
user_id = value_result.json()['data']['id']
person_result = {
reg_name: {
"mobile_phone": result_phone,
'pwd': pwd,
"type": reg_type,
"id": user_id,
"reg_name": reg_name
}
}
hr.close()
hm.close()
return person_result
def get_phone(cls, user, password="******", type_num=1):
hr = HandleRequest()
hm = HandleMysql()
hr.common_head({'X-Lemonban-Media-Type': 'lemonban.v2'})
base_url = hy.read_yaml('api', 'load')
register_url = hy.read_yaml('api', 'register')
all_url = ''.join((base_url, register_url))
while True:
# 获取在数据库中不存在的电话号码
phone = hm.get_right_phone()
data = {
"mobile_phone": phone,
"pwd": password,
"type": type_num,
"reg_name": user
}
result = hr.send(url=all_url, data=data)
if result.json()['code'] == 0 and result.json()["msg"] == "OK":
break
my_result = hm.get_mysql_result(hy.read_yaml('mysql', 'user_id'),
args=phone)
user_id = my_result[0]['id']
result_data = {
user: {
"user_phone": phone,
"pwd": password,
"user_id": user_id,
"reg_name": user
}
}
hm.close()
hr.close()
return result_data
def set_up():
hr = HandleRequest()
hm = HandleMysql()
hr.common_head(hy.read_yaml('api', 'header'))
yield hr, hm, hy, hl
hr.close()
hm.close()
def get_phone(cls, username, password='******', type=1):
hr = HandleRequest()
hm = HandleMysql()
base_url = hy.read_yaml('api', 'load')
register_url = hy.read_yaml('api', 'register')
all_url = ''.join((base_url, register_url))
while True:
phone = hm.get_non_existent()
data = {
"mobile_phone": phone,
"pwd": password,
"type": type,
"reg_name": username
}
hr.common_head(hy.read_yaml('api', 'header'))
result = hr.send(url=all_url, data=data)
if result.json()['code'] == 0 and result.json()['msg'] == 'OK':
break
user_id = hm.get_mysql_result(sql=hy.read_yaml('mysql', 'user_id'),
args=phone)
result_data = {
username: {
"mobilephone": phone,
"pwd": password,
"user_id": user_id[0]['id'],
"reg_name": username
}
}
hr.close()
hm.close()
return result_data
def set_up():
hr = HandleRequest()
hm = HandleMysql()
hl = HandleYaml(PERSONPHONE_PATH)
hr.common_head(hy.read_yaml('request', 'header'))
yield hr, hm, hl
hr.close()
hm.close()
