def login_interface(username, password):
# 1) 先查看当前用户数据是否存在
# {用户数据字典} or None
user_dic = db_handler.select(username)
# 2) 判断用户是否存在
if user_dic:
# 若有冻结用户，则需要判断是否被锁定
if user_dic.get('locked'):
return False, '当前用户已被锁定'
# 给用户输入的密码做一次加密
password = common.get_pwd_md5(password)
# 3) 校验密码是否一致
if password == user_dic.get('password'):
msg = f'用户: [{username}] 登录成功!'
user_logger.info(msg)
return True, msg
else:
msg = f'用户: [{username}]密码错误!'
user_logger.warn(msg)
return False, msg
msg = f'用户: [{username}]用户不存在，请重新输入！!'
return False, msg
def register_interface(user_name, user_passwd):
"""
判断用户是否存在,存在返回False,不存在则注册新用户
:param user_name: 用户名
:param user_passwd: 用户密码
:return:
"""
# 查看用户是否存在
user_dic = db_handle.select_user(user_name)
# 判断用户是否存在,存在返回false以及提示信息
if user_dic:
msg = f'用户名 [f{user_name}] 重复,请重新输入新的用户名'
user_logger.warn(msg)
return False, msg
# 用户密码MD5数据加密
user_passwd = common.get_pwd_md5(user_passwd)
# 用户不存在则注册
db_handle.save_user(username=user_name, userpasswd=user_passwd)
# 记录日志功能
msg = f'用户 [{user_name}] 注册成功!'
user_logger.info(msg)
return True, msg
def login_interface(user_dic, conn):
user_list = models.User.select_sql(user_name=user_dic.get("user_name"))
if user_list:
user_obj = user_list[0]
user_name = user_obj.get("user_name")
pwd = user_dic.get("pwd")
if user_obj.pwd == common.get_pwd_md5(pwd):
session = common.get_session(user_name)
addr = user_dic.get("addr")
user_data.mutex.acquire()
user_data.online_user[addr] = [session, user_obj.user_id]
user_data.mutex.release()
print(user_data.online_user)
send_dic = {
"flag": True,
"msg": "登陆成功",
"session": session,
"is_vip": user_obj.is_vip
}
common.send_msg(send_dic, conn)
else:
send_dic = {
"flag": False,
"msg": "登陆失败",
}
common.send_msg(send_dic, conn)
else:
send_dic = {
"flag": False,
"msg": "登陆失败",
}
common.send_msg(send_dic, conn)
def login_interface(back_dic, conn):
user_name = back_dic.get('name')
user_pwd = common.get_pwd_md5(back_dic.get('pwd'))
user_type = back_dic.get('user_type')
user_obj_list = models.User.select(user_name=user_name)
if user_obj_list:
for user_obj in user_obj_list:
if user_obj.user_type == user_type:
if user_obj.pwd == user_pwd:
session = common.get_session_md5(user_name)
addr = back_dic.get('addr')
user_data.metux.acquire()
user_data.online_user[addr] = [session, user_obj.user_id]
user_data.metux.release()
send_dic = {
'flag': True,
'msg': '用户登录成功！',
'session': session
}
else:
send_dic = {'flag': False, 'msg': '该用户密码错误！'}
break
else:
send_dic = {'flag': False, 'msg': '该用户不存在！'}
else:
send_dic = {'flag': False, 'msg': '该用户不存在！'}
print(send_dic)
common.send_msg(send_dic, conn)
def register_interface(username, password, balance=15000):
# 调用数据层
user_dic = db_handle.select(username)
# 用户已存在
if user_dic:
return False, '用户已存在'
# 用户不存在
password = common.get_pwd_md5(password)
user_dic = {
'username': username,
'password': password,
'balance': balance,
# 记录用户流水的列表
'flow': [],
# 记录用户的购物车
'shop_car': {},
# 记录用户是否被冻结
'locked': False
}
db_handle.save(user_dic)
msg = f'用户[{username}] 注册成功'
user_logger.info(msg)
return True, msg
def register_interface(back_dic, conn):
user_name = back_dic.get('name')
user_pwd = common.get_pwd_md5(back_dic.get('pwd'))
user_type = back_dic.get('user_type')
user_obj_list = models.User.select(user_name=user_name)
if user_obj_list:
for user_obj in user_obj_list:
if user_obj.user_type == user_type:
send_dic = {'flag': False, 'msg': '该用户已经存在了！'}
break
else:
user_obj = models.User(user_name=user_name,
pwd=user_pwd,
is_vip=0,
is_locked=0,
user_type=user_type,
register_time=datetime.datetime.now())
user_obj.save()
send_dic = {'flag': True, 'msg': '用户注册成功'}
else:
user_obj = models.User(user_name=user_name,
pwd=user_pwd,
is_vip=0,
is_locked=0,
user_type=user_type,
register_time=datetime.datetime.now())
user_obj.save()
send_dic = {'flag': True, 'msg': '用户注册成功'}
print(send_dic)
common.send_msg(send_dic, conn)
def login():
while True:
username = input('请输入用户名:').strip()
passworld = input('请输入密码') .strip()
passworld = common.get_pwd_md5(passworld)
user_data= user_interface.login_interface(username,passworld)
global login_stat
login_stat = user_data
return login_stat
def login(usrname, password):
if usr_data := db_handler.select(usrname):
password = common.get_pwd_md5(password)
if usr_data['password'] == password:
return 'sucess', common.disp_basic_info(usr_data)
else:
return 'pwd_err', '密码输入错误！'
def rigister():
user_name = input('请输入用户名:')
passworld = input('请输入密码:')
re_passworld = input('请确认密码:')
# 判断密码是否一致
if passworld == re_passworld:
#调用接口层的注册接口,将用户名与密码交给接口层进行处理
passworld = common.get_pwd_md5(passworld)
user_interface.regist_interface(user_name, passworld)
else:
print('两次输入的密码不一致,请重新输入')
def login_interface(username, password):
user_dict = db_handler.select(username)
if user_dict:
if user_dict['locked']:
return False, f'用户{username}已被锁定 无法登录'
if common.get_pwd_md5(password) == user_dict.get('password'):
return True, f'[{user_dict.get("username")}] 登录成功'
else:
return False, '密码错误'
return False, '用户不存在'
def login_interface(username, password):
user_dict = db_handler.select(username)
if user_dict.get('locked'):
return False, '当前用户已被锁定'
if user_dict:
password = common.get_pwd_md5(password)
if password == user_dict.get('password'):
return True, f'用户:[{username}]登录成功!'
else:
return False, '密码错误'
return False, '用户不存在,请重新输入!'
def login_interface(user_name,password):
user_dic=db_handler.select(user_name)
if user_dic['locaked']:
shop_logger.info(f'{user_name}已经被锁定')
return False,'当前用户已被锁定'
password=common.get_pwd_md5(password)
if user_dic:
if password == user_dic['password']:
return True,f'{user_name}登录成功'
else:
return False,'用户名或者密码错误'
else:
return False,'用户不存在'
def login_interface(username, password):
password = common.get_pwd_md5(password)
user_dict = db_handler.select(username)
if user_dict:
if user_dict['locked'] == '0':
if user_dict['password'] == password:
return 1, '欢迎来访！'
else:
return 0, '密码错误！'
else:
return 0, '该账户已经冻结'
else:
return 0, '用户名不存在'
def register_interface(username, password, balance=15000):
user_dic = db_handler.select(username)
if user_dic:
return False, '用户名已存在'
user_dic = {
'username': username,
'password': common.get_pwd_md5(password),
'balance': balance,
'flow': [], # 用户流水
'shop_car': {}, # 用户购物车
'locked': False
}
db_handler.save(user_dic)
return True, f'{username}注册成功'
def login_interface(username,password):
# 先查看当前用户数据是否存在
# {用户数据字典} or None
user_dic = db_handler.select(username)
# 用于判断用户是否存在
if user_dic:
# 给用户输入的密码做加密
password = common.get_pwd_md5(password)
# 校验密码是否一致
if password == user_dic.get('password'):
return True,f'用户: [{username}] 登录成功!'
else:
return False, '密码错误'
return False, '用户不存在,请重新输入!'
def register_interface(username,password,balance):
# 1,查看用户是否存在
# 2,调用数据处理层中的select 函数 会返回用户字典或者None
user_dic = db_handler.select(username)
#{user:user,pwd:pwd...} or None
# 2,若用户不存在，则保存用户信息
if user_dic:
# return (False,'用户名已存在!')
return False, '用户名已存在!'
password = common.get_pwd_md5(password)
# 3,组织用户的数据字典信息
user_dic = {
'username': username,
'password': password,
'balance': balance,
# 用于记录用户流水的列表
'flow': [],
# 用于记录用户的购物车
'shop_car': {},
# locked: 用于记录用户是否被冻结
# False: 未冻结 True： 已被冻结
'locaked': False
}
# 记录流水
# 保存数据
db_handler.save(user_dic)
return True,f'{username} 注册成功!'
# 存不是目的，目的是为了更方便的取数据
# 文件名:用户名.json
# 拼接用户json文件的路径
user_path = os.path.join(
settings.USER_DATA_PATH, f'{username}.json'
)
# 查看用户是否存在
# 若不存在，则让用户重新受保护
if os.path.exists(user_path):
print('请重新输入!')
with open(user_path, 'r', encoding='utf-8') as f:
user_dic = json.load(f)
if user_dic:
print('用户已存在,请重新输入!')
# 若用户不存在，则保存用户数据
# 组织用户的数据字典信息
user_path = os.path.join(
settings.USER_DATA_PATH, f'{username}.json'
)
with open(user_path, 'w', encoding='utf-8') as f1:
json.dump(user_dic, f1, ensure_ascii=False)
def register_interface(username,password,balance=15000):
password = common.get_pwd_md5(password)
user_dic = {
'username': username,
'password': password,
'balance': balance,
'flow': [],
'shop_car':{},
'locked':False
}
db_handler.save(user_dic)
msg = f'{username} 注册成功！'
user_logger.info(msg)
return True, msg
def swapper(*args, **kwargs):
print(f"当前用户：{kwargs['account']}")
current_account_data = db_handler.select(kwargs['account'])
err_times = 0
while err_times < 3:
input_password = input("请输入您的密码以确认您的操作：")
input_password = common.get_pwd_md5(input_password)
if current_account_data['password'] == input_password:
print("认证成功！")
res = func(*args, **kwargs)
return res
else:
print("输入错误！")
err_times += 1
return False, "连续三次输入密码错误！"
def resister_interface(user_name,password,balance=15000):
user_dic=db_handler.select(user_name)
if user_dic:
return False,'用户名已存在'
password=common.get_pwd_md5(password)
user_dic = {
'username': user_name,
'password': password,
'balance': balance,
'flow': [],
'shop_car': {},
'locaked': False
}
db_handler.save(user_dic)
shop_logger.info(f'{user_name}注册成功')
return True,f'{user_name}注册成功'
def login_interface(username, password):
user_dict = db_handler.check(username)
if user_dict:
if user_dict['locked']:
return False, f"{username}账户已冻结，无法登录"
password = common.get_pwd_md5(password)
if password == user_dict.get('password'):
msg=f"{username} 登录成功。"
ATM_logger.info(msg)
return True, msg
msg=f"{username} 登录失败，密码不正确，请重新输入"
ATM_logger.error(msg)
return False, msg
msg = f"账户{username}不存在。"
# ATM_logger.error(msg)
return False, msg
def register_interface(username, password, balance=15000):
dic = db_handler.select(username)
if dic:
return 0, '用户名已存在！'
else:
password = common.get_pwd_md5(password)
user_dict = {
'username': username,
'password': password,
'balance': balance,
'flow': [],
'shop_car': {},
'locked': 0
}
db_handler.save(user_dict)
return 1, '注册成功！'
def admin_login_interface(username,password):
user_dic = db_handler.select(username)
try:
if user_dic:
password = common.get_pwd_md5(password)
if user_dic['is_admin']:
if password == user_dic.get('password'):
msg = f'管理员[{username}]登录成功！'
user_logger.info(msg)
return True, msg
else:
msg = '用户名或密码错误！'
user_logger.warn(msg)
return False, msg
except KeyError:
return False, '该用户不是管理员！请重新输入'
return False,'用户不存在，请重新输入！'
def login_api(username, password):
user_dic = db_handler.select(username)
if user_dic:
password = common.get_pwd_md5(password)
if user_dic.get('locked'):
return False, '用户被冻结'
if username == user_dic.get('username') and password == user_dic.get(
'password'):
return True, f'{username} 登录成功！'
else:
return False, f'{username} 密码错误！'
return False, '用户名不存在,请重新输入'
def login_interface(username,password):
# 1. 判断用户是否存在
user_dic = db_handler.select(username)
if user_dic:
if user_dic['locked'] == False:
password = common.get_pwd_md5(password)
if password == user_dic.get('password'):
msg = f'用户[{username}]登录成功！'
user_logger.info(msg)
return True,msg
else:
msg = '用户名或密码错误！'
user_logger.warn(msg)
return False,msg
else:
return False,'该用户已锁定，请联系管理员！'
return False,'用户不存在，请重新输入！'
def login_interface(user_name, user_passwd):
# 获取用户数据
user_data_dic = db_handle.select_user(user_name)
if user_data_dic:
user_passwd = common.get_pwd_md5(user_passwd)
if user_passwd == user_data_dic.get('password'):
msg = f'用户 [{user_name}] 登录成功!'
user_logger.info(msg)
return True, msg
else:
msg = '用户名或密码错误'
user_logger.warn(msg)
return False, msg
else:
msg = '用户名或密码错误'
user_logger.warn(msg)
return False, msg
def login_interface(username, password):
user_dic = db_handle.select(username)
if user_dic:
if user_dic['locked']:
return False, '当前用户已被锁定'
password = common.get_pwd_md5(password)
if password == user_dic.get('password'):
msg = f'用户 [{username}] 登录成功！'
user_logger.info(msg)
return True, msg
msg = f'用户 [{username}] 密码错误,登录失败！'
user_logger.warn(msg)
return False, '密码错误'
return False, f'用户 [{username}] 不存在,登录失败！'
def register_interface(username, password, balance=15000):
user_dict = db_handler.select(username)
if user_dict:
return False, '用户名已存在,请重新输入!'
# 密码加密
password = common.get_pwd_md5(password)
user_dict = {
'username': username,
'password': password,
'balance': balance,
'flow': [], # 记录用户流水的列表
'shop_car': {}, # 购物车
'locked': False
}
db_handler.save(user_dict)
return True, f'{username}注册成功!'
def register_interface(username, password, balance=15000):
user_dict = db_handler.check(username)
if user_dict:
msg=f"{username} 用户已经存在，请重新输入用户名："
ATM_logger.error(msg)
return False, msg
else:
password = common.get_pwd_md5(password)
user_dict = {
"username": username,
"password": password,
'balance': balance,
'flow': [], # 用户流水
'shop_car': {},
'locked': False, # 用户是否被冻结
}
msg=f"{username} 用户创建成功"
db_handler.save(user_dict)
# ATM_logger.info(msg)
return True, msg
def regist(usrname, password, limit=15000):
if db_handler.select(usrname):
return False, "当前用户名已存在，请重新输入！"
else:
password = common.get_pwd_md5(password)
usr_profile = {
'usrname': usrname,
'password': password,
'credit_limit': limit,
'balance': limit,
'shopping_cart': {},
'statement': {},
'isfreeze': False,
}
try:
db_handler.create(usr_profile)
except e:
print(e)
return False, "创建用户过程中出现错误，请稍后再试！"
else:
return True, "用户创建成功！"
def register_api(username, password, balance=15000):
if not username or not password:
return False, '用户名或密码为空'
# 查询用户是否存在
user_dic = db_handler.select(username)
if user_dic:
return False, f'{username} 用户名已存在！'
# 用户不存在，则保存用户数据
# 密码加密
password = common.get_pwd_md5(password)
user_dic = {
'username': username,
'password': password,
'balance': balance,
'flow': [], # 用于记录用户流水的列表
'shop_car': {}, # 用于记录用户购物车
'locked': False, # 用于记录用户是否被冻结
}
db_handler.save(user_dic)
msg = f'{username} 注册成功！'
return True, user_logger.info(msg)
