def get_user_list_control(list_info):
user_list_return_info = {
"timestamp": CampServer.time_now_str(),
"request_id": list_info["request_id"],
"code": 200,
"sub_code": 0,
"data": None
}
try:
if "usernames" in list_info:
user_list_return_info["data"] = DataBaseControl.get_user_many(
u_key='username',
user_list=list_info["usernames"]
)
elif "emails" in list_info:
user_list_return_info["data"] = DataBaseControl.get_user_many(
u_key='email',
user_list=list_info["emails"]
)
elif "phones" in list_info:
user_list_return_info["data"] = DataBaseControl.get_user_many(
u_key='phone',
user_list=list_info["phones"]
)
user_list_return_info["code"] = 200
user_list_return_info["sub_code"] = 0
return user_list_return_info
except Exception as ex:
print(ex)
user_list_return_info["code"] = 500
user_list_return_info["sub_code"] = -1
return user_list_return_info
def verification_code(username, email, info_type):
if CampServer.is_frequently(email) is False:
code = random.randint(100000, 999999)
info_dict = {
"username": username,
"email": email,
"code": code,
"timestamp": CampServer.time_now_str(),
"type": info_type
}
DataBaseControl.insert_verification_code(**info_dict)
return code
else:
return None
def sign_in_control(sign_in_info):
login_return_info = {
"timestamp": CampServer.time_now_str(),
"request_id": sign_in_info["request_id"],
"code": 200,
"sub_code": 0
}
try:
info = {
"username": sign_in_info["username"],
"pwd": CampServer.sha256_key(sign_in_info["username"], sign_in_info["pwd"]),
"appid": sign_in_info["appid"],
"clientId": sign_in_info["clientId"],
"timestamp": sign_in_info["timestamp"],
"login_channel": sign_in_info["login_channel"],
}
check_account = False
if "email" in sign_in_info:
info["email"] = sign_in_info["email"]
check_account = True
if check_account is False:
raise Exception('lack email')
if "phone" in sign_in_info:
info["phone"] = sign_in_info["phone"]
try:
DataBaseControl.sign_in_information(**info)
login_return_info["code"] = 200
login_return_info["sub_code"] = 0
return login_return_info
except Exception as ex:
print(ex)
login_return_info["code"] = 200
login_return_info["sub_code"] = 1
return login_return_info
except Exception as ex:
print(ex)
login_return_info["code"] = 500
login_return_info["sub_code"] = -1
return login_return_info
def update_control(update_info):
update_return_info = {
"timestamp": CampServer.time_now_str(),
"request_id": update_info["request_id"],
"code": 200,
"sub_code": 0
}
try:
old_info = DataBaseControl.get_user_information_by_username(update_info["username"])
need_update_info = {}
for x in update_info.keys():
if x in ["token", "timestamp", "request_id"]:
continue
elif x == "pwd":
new_pwd = CampServer.sha256_key(update_info["username"], update_info["pwd"])
if new_pwd == old_info["pwd"]:
continue
else:
need_update_info[x] = new_pwd
elif x == "extra_payload":
new_data = CampServer.payload_update(old_info["extra_payload"], update_info["extra_payload"])
if new_data is not None:
need_update_info["extra_payload"] = new_data
else:
continue
elif x == "user_payload":
new_data = CampServer.payload_update(old_info["user_payload"], update_info["user_payload"])
if new_data is not None:
need_update_info["user_payload"] = new_data
else:
continue
else:
need_update_info[x] = update_info[x]
DataBaseControl.update_information_by_username(update_info["username"], need_update_info)
update_return_info["code"] = 200
update_return_info["sub_code"] = 0
return update_return_info
except Exception as ex:
print(ex)
update_return_info["code"] = 500
update_return_info["sub_code"] = -1
return update_return_info
def delete_user_control(delete_info):
delete_return_info = {
"timestamp": CampServer.time_now_str(),
"request_id": delete_info["request_id"],
"code": 200,
"sub_code": 0
}
try:
DataBaseControl.delete_information(delete_info["username"])
delete_return_info["code"] = 200
delete_return_info["sub_code"] = 0
return delete_return_info
except Exception as ex:
print(ex)
delete_return_info["code"] = 500
delete_return_info["sub_code"] = -1
def is_frequently(email):
time_out = 120
now_time = CampServer.time_now_str()
all_verification_code = DataBaseControl.get_verification_code(email)
if len(all_verification_code) < 1:
return False
else:
for x in all_verification_code:
if (now_time - x["timestamp"]) < time_out:
return True
else:
continue
return False
def forgot_password_verification_code(username, email, request_id, info_type):
raise_code = 0
generate_verification_code_info = {
"timestamp": CampServer.time_now_str(),
"request_id": request_id,
"code": 200,
"sub_code": 0
}
try:
get_info = DataBaseControl.check_username_and_email(username, email)
if len(get_info) > 1:
raise_code = OpenIDRaiseCode.hase_existed
raise Exception
elif len(get_info) == 1:
info_username = get_info[0].get("username", "")
info_email = get_info[0].get("email", "")
if info_username == username and info_email == email:
code = CampServer.verification_code(info_username, info_email, info_type)
if code is not None:
EmailServer().new_password_mail(email, code)
return generate_verification_code_info
else:
raise_code = OpenIDRaiseCode.frequent
raise Exception
elif info_email != email or info_username != username:
raise_code = OpenIDRaiseCode.has_existed
raise Exception
elif len(get_info) < 1:
raise_code = OpenIDRaiseCode.not_existed
raise Exception
except Exception as ex:
print(ex)
if raise_code == 0:
raise_code = OpenIDRaiseCode.lack_param
generate_verification_code_info["code"] = 500
generate_verification_code_info["sub_code"] = raise_code
return generate_verification_code_info
def consume_verification_code_control(info):
raise_code = 0
generate_verification_code_info = {
"timestamp": CampServer.time_now_str(),
"request_id": "",
"code": 200,
"sub_code": 0
}
try:
username = info.get("username", "")
email = info.get("email", "")
request_id = info.get("request_id", "")
info_type = info.get("type", "")
code = info.get("code", "")
generate_verification_code_info["request_id"] = request_id
database_verification_code = DataBaseControl.get_new_one_verification_code(email)
if len(username) < 4 or len(email) < 4:
raise_code = OpenIDRaiseCode.lack_param
raise Exception
elif len(database_verification_code) == 0:
raise_code = OpenIDRaiseCode.not_existed
raise Exception
database_code = database_verification_code["code"]
database_timestamp = database_verification_code["timestamp"]
database_username = database_verification_code["username"]
database_type = database_verification_code["type"]
if database_code == code \
and CampServer.is_600s(database_timestamp) is True \
and database_username == username \
and info_type == database_type:
return generate_verification_code_info
elif database_code != code:
raise_code = OpenIDRaiseCode.auth_fail
raise Exception
elif CampServer.is_600s(database_timestamp) is False:
raise_code = OpenIDRaiseCode.timeout
raise Exception
elif database_username != username:
raise_code = OpenIDRaiseCode.not_existed
raise Exception
elif database_type != info_type:
raise_code = OpenIDRaiseCode.auth_fail
raise Exception
else:
raise_code = OpenIDRaiseCode.database_error
raise Exception
except Exception as ex:
print(ex)
if raise_code == 0:
raise_code = OpenIDRaiseCode.lack_param
generate_verification_code_info["code"] = 500
generate_verification_code_info["sub_code"] = raise_code
return generate_verification_code_info
def login_control(login_info):
login_return_info = {
"timestamp": CampServer.time_now_str(),
"request_id": login_info["request_id"],
"data": {},
"code": 200,
"sub_code": 0
}
try:
get_info = None
if "email" in login_info:
get_info = DataBaseControl.get_user_information_by_email(login_info["email"])
elif "username" in login_info:
get_info = DataBaseControl.get_user_information_by_username(login_info["username"])
elif "phone" in login_info:
get_info = DataBaseControl.get_user_information_by_phone(login_info["phone"])
if get_info is not None and get_info["state"] == 0:
login_success_info = {
"uid": get_info["uid"],
"username": get_info["username"],
"phone": get_info["phone"],
"email": get_info["email"],
"login_success": 0,
"request_id": login_info["request_id"],
"appid": login_info["appid"],
"timestamp": login_info["timestamp"],
"login_channel": login_info["login_channel"],
"user_ip": login_info["user_ip"],
"remote_ip": login_info["remote_ip"]
}
if "token" in login_info:
login_success_info["token"] = login_info["token"]
if "clientId" in login_info:
login_success_info["clientId"] = login_info["clientId"]
if CampServer.sha256_key(login_info["username"], login_info["pwd"]) == get_info["pwd"]:
login_success_info["login_success"] = 0
DataBaseControl.login_operating_information_update(**login_success_info)
del get_info["pwd"]
login_return_info["data"] = get_info
login_return_info["code"] = 200
login_return_info["sub_code"] = 0
return login_return_info
else:
login_return_info["code"] = 200
login_return_info["sub_code"] = 1
DataBaseControl.login_operating_information_update(**login_success_info)
return login_return_info
else:
login_return_info["code"] = 200
login_return_info["sub_code"] = 1
return login_return_info
except Exception as ex:
print(ex)
login_return_info["code"] = 500
login_return_info["sub_code"] = -1
return login_return_info