def register_interface(back_dic, conn):
# 面条版: 业务逻辑
username = back_dic.get('username')
user_obj = Models.User.select(user_name=username) # [obj]
print(user_obj) # []
if not user_obj:
password = back_dic.get('password')
# 创建数据
user_obj = Models.User(user_name=username,
pwd=common.get_md5(password),
user_type=back_dic.get('user_type'),
register_time=datetime.datetime.now(),
is_vip=0,
is_locked=0)
user_obj.save()
send_dic = {'flag': True, 'msg': '注册成功!'}
else:
send_dic = {'flag': False, 'msg': '用户已存在!'}
print(1111)
common.send_msg(send_dic, conn)
def register_interface(back_dic,):
# 面条版: 业务逻辑
username = back_dic.get('username')
user_obj = User(name=username).select(whereStr ="name = {}".format(username))
if not user_obj:
password = back_dic.get('password')
# 创建数据
user_obj = User(name=username,
password=common.get_md5(password),
user_type=back_dic.get('user_type'),
register_time=str(datetime.datetime.now()),
is_vip=0,
is_locked=0
)
user_obj.save()
send_dic = {
'flag': True,
'msg': '注册成功!'
}
else:
send_dic = {'flag': False,
'msg': '用户已存在!'}
return send_dic
def login(client):
if user_dic.get('cookie'):
print("您已登录，无需重复登录！")
return
while True:
name = input("请输入用户名：")
password = input("请输入密码：")
send_dic = {
'type': 'login',
'name': name,
'password': common.get_md5(password),
'user_type': 'user'
}
back_dic = common.send_back(send_dic, client)
if back_dic.get('flag'):
for n in back_dic.get('notice'):
print(n['title'])
print(n['content'])
time.sleep(3)
print(back_dic.get('msg'))
user_dic['cookie'] = back_dic.get('session')
user_dic['is_vip'] = back_dic.get('is_vip')
break
else:
print(back_dic.get('msg'))
def login(message, conn):
name = message['name']
password = message['password']
user = models.UserInfo.select_one(name=name)
if user:
if user.user_type == message['user_type']:
if user.password == password:
send_dic = {
'flag': True,
'msg': '%s:%s 登录成功' % (user.user_type, name)
}
session = common.get_md5(name, password)
send_dic['session'] = session
print(user_info.live_user, '<==')
user_info.mutex.acquire()
user_info.live_user[message['addr']] = [session, user.id]
user_info.mutex.release()
print(user_info.live_user, '<==')
else:
send_dic = {
'flag': False,
'msg': '%s:%s 密码错误!' % (user.user_type, name)
}
else:
send_dic = {
'flag': False,
'msg': '%s不是%s用户!' % (user.name, message['user_type'])
}
else:
send_dic = {
'flag': False,
'msg': '%s:%s 不存在!' % (message['user_type'], name)
}
common.send_message(send_dic, conn)
def register_interface(name, pwd, balance=15000): # 默认额度15000
# 调用数据处理层拿到用户信息
user_dic = db_handler.select(name)
# 判断用户信息是否存在
if user_dic: # user信息， False
logger.info('用户: [%s] 已存在!' % name)
return False, '用户已存在'
# 加密用户输入的密码
md5_pwd = common.get_md5(pwd)
user_dic = {
'name': name,
'pwd': md5_pwd,
'balance': balance,
'flow': [],
'shopping_cart': {}
}
db_handler.save(user_dic)
logger.info('用户: [%s] 注册成功!' % name)
return True, '注册成功!' # (True, '注册成功!')
def register_interface(back_dic, conn):
"""
用户注册接口
:param bacd_dic:
:param conn:
"""
username = back_dic.get('username')
user_pwd = back_dic.get('pwd')
user_type = back_dic.get('user_type')
user_obj = Models.UserInfo.sql_select(user_name=username)
user_pwd = common.get_md5(user_pwd)
# 判断用户是否存在
if not user_obj:
user_reg = Models.UserInfo(
user_name=username,
user_pwd=user_pwd,
is_vip=0,
is_locked=0,
user_type=user_type,
register_time=datetime.datetime.now()
)
user_reg.sql_save()
send_dic = {
'flag': True,
'msg': '注册成功！'
}
else:
send_dic = {
'flag': False,
'msg': '用户已经存在！'
}
common.send_msg(send_dic, conn)
def login_interface(back_dic, conn):
"""
用户登录接口
:param back_dic:
:param conn:
"""
username = back_dic.get('username')
pwd = back_dic.get('pwd')
user_pwd = common.get_md5(pwd)
user_obj = Models.UserInfo.sql_select(user_name=username)
# 在服务器中保存session信息
addr = back_dic.get('addr')
if user_obj:
user_obj = user_obj[0]
# 判断密码是否相同
if user_obj.user_pwd == user_pwd:
send_dic = {
'flag': True,
'msg': '登录成功',
'is_vip': user_obj.is_vip,
'new_notice': None
}
session = common.get_session(username)
send_dic['session'] = session
# 将登录信息保存到online_info中
lock.acquire()
user_data.online_info[addr] = [session, user_obj.user_id] # 保存，session 和用户id用户登录验证
lock.release()
# 获取最新公告
new_notice = common_interface.get_new_notice_interface()
send_dic['new_notice'] = new_notice
else:
send_dic = {
'flag': False,
'msg': '用户名或密码错误'
}
else:
send_dic = {
'flag': False,
'msg': '用户不存在，还不去注册去'
}
# {"('127.0.0.1', 63696)": ['104207c87d76dba69375219c4909eb85', 1]}
# print(user_data.online_info)
# print(user_data.mutex)
common.send_msg(send_dic, conn)
def login_interface(user, pwd):
user_dict = db_handler.select(user)
if not user_dict:
return False, '用户不存在'
else:
if user_dict.get('pwd') == common.get_md5(user, pwd):
return True, '登录成功'
# 密码错误
else:
return False, '密码错误'
def login_interface(name, pwd):
user_dic = db_hander.select(name)
if not user_dic:
return False, '用户不存在'
md5_pwd = common.get_md5(pwd)
if md5_pwd == user_dic['pwd']:
logger.info('用户[%s]登录成功' % name)
return True, '登录成功'
else:
return False, '密码错误'
def upload_movie(client):
# 获取本地电影的列
while True:
movie_list = common.get_movie_list()
if not movie_list:
print('暂无本地电影')
break
for index, movies in enumerate(movie_list, 1):
print(index, movies)
choice = input('输入电影的编号：').strip()
if not choice.isdigit():
continue
choice = int(choice) - 1
if choice not in range(len(movie_list)):
continue
movie_name = movie_list[choice]
file_path = os.path.join(settings.LOCAL_MOVIE_PATH, movie_name) # 文件路径
file_md5 = common.get_md5(file_path)
send_dic = {
'type': 'check_movie',
'session': user_dic['cookie'],
'file_md5': file_md5
}
file_size = os.path.getsize(file_path)
back_dic = common.send_back(send_dic, client)
if back_dic['flag']:
print('选择上传的电影is_free')
is_free = input('输入是否是免费<y/>n:').strip()
# if is_free == 'y':
# is_free =1
# else:
# is_free = 0
is_free = 1 if is_free == 'y' else 0
send_dic = {
'type': 'upload_movie',
'session': user_dic['cookie'],
'file_md5': file_md5,
'file_size': file_size,
'movie_name': movie_name,
'is_free': is_free
}
back_dic = common.send_back(send_dic, client, file_path)
if back_dic['flag']:
print(back_dic['msg'])
break
else:
print(back_dic['msg'])
break
else:
print(back_dic['msg'])
break
def login_interface(name, pwd):
user_dict = db_handles.select(name)
if not user_dict:
return False, '用户名不存在'
# 调用公共功能层的登陆密码加密和文件中的密码进行对比判断是否一致
new_md5 = common.get_md5(pwd) # common 加密返回值 变量接收一下即可
if new_md5 == user_dict['pwd']: # 输入的加密密码是否和系统存的加密密码一致
logger_obj = common.get_logger('用户登录功能')
logger_obj.info('尊敬的%s用户您已成功登录' % name)
return True, '%s登录成功' % user_dict['name']
else:
return False, '输入系统密码与系统密码不一致'
def register_interface(name, pwd, balance=15000):
# 密码加密存储
md5_pwd = common.get_md5(pwd)
user_dic = {
'name': name,
'pwd': md5_pwd,
'balance': balance,
'flow': [],
'shopping_car': {}
}
db_hander.save(user_dic)
logger.info("用户%s注册成功" % name)
return True, "用户[%s]注册成功" % name
def login_interface(name, pwd):
user_dic = db_handler.select(name)
if user_dic['lock']:
user_logger.info(f'用户[{name}]已被冻结,请联系管理员!')
return False, '用户已被冻结，请联系管理员'
pwd = common.get_md5(pwd)
if pwd == user_dic['pwd']:
msg1 = f'用户{name}登陆成功！'
user_logger.info(msg1)
return True, msg1
else:
msg2 = f'用户{name}密码错误，登录失败！'
user_logger.info(msg2)
return False, msg2
def register_interface(username, password, balance=15000):
#加密处理
password = common.get_md5(password)
user_dict = {
'user': username,
'pwd': password,
'balance': balance,
'flow': [],
'shop_cart': {},
'lock': False #冻结账户
}
#保存注册用户的信息
db_handler.save(user_dict)
user_logger.info('%s用户注册成功' % user_dict['user'])
return True
def register_interface(name, pwd, balance=15000):
pwd = common.get_md5(pwd)
user_dic = {
'name': name,
'pwd': pwd,
'balance': balance,
'flow': [],
'shop_cart': {},
'lock': False
}
db_handler.save(user_dic)
msg = f"用户{name}注册成功"
user_logger.info(msg)
return msg
def login_interface(name, pwd):
# 用户信息 / None
user_dic = db_handler.select(name)
if not user_dic:
return False, '用户不存在!'
md5_pwd = common.get_md5(pwd)
if md5_pwd == user_dic['pwd']:
return True, '登录成功!'
else:
return False, '密码错误!'
def login_interface(username, pwd):
#获取用户登录输入的密码加密后的结果
pwd = common.get_md5(pwd)
#获取用户注册的加密密码
user_dic = db_handler.select(username)
if user_dic['lock'] == True:
user_logger.info('%s用户已经被冻结，请联系管理员' % username)
return False, '%s用户已经被冻结，请联系管理员' % username
if pwd == user_dic['pwd']:
user_logger.info('%s用户登录成功' % username)
return True, '登录成功'
else:
user_logger.info('输入信息不正确,请重新输入')
return False, '输入信息不正确,请重新输入'
def login_interface(user, pwd):
user_dic = db_handler.select(user)
pwd = common.get_md5(pwd)
if user_dic['lock']:
user_logger.info(f'用户[{user}]已被冻结,请联系管理员!')
return False, '用户已被冻结,请联系管理员!'
if user_dic['pwd'] == pwd:
user_logger.info(f'{user}登陆成功!')
return True, f'{user}登陆成功!'
user_logger.info(f'{user}输入密码错误!')
return False, f'{user}密码错误!'
def register_interface(user, passwd):
# 调用接口，加密
pwd = common.get_md5(passwd)
# 把用户所有信息做成一个字典，然后调用接口保存
user_dict = {
'user': user,
'pwd': pwd,
'balance': 10000,
'worter': [],
'shop_car': [],
'lock': True
}
db_hander.save(user_dict)
return f'{user_dict["user"]}用户注册成功！'
def user_login(client):
if user_dic['session']:
print("您已经登录!")
return
while True:
name = input("请输入用户名>>:")
password = input("请输入密码>>:")
send_dic = {'type': 'login',
'name': name,
'password': common.get_md5(password),
'user_type': 'user'}
back_dic = common.send_back_message(send_dic, client)
if back_dic['flag']:
print(back_dic['msg'])
user_dic['session'] = back_dic['session']
break
else:
print(back_dic['msg'])
def register_interface(user, pwd, balance=1500):
user_dict = db_handler.select(user)
if user_dict:
return False, "用户已存在"
else:
# 做注册业务逻辑的处理
user_info = {
'user': user,
'pwd': common.get_md5(user, pwd),
'flow_list': [],
'balance': balance,
'shop_car': {},
'locked': False # True
}
# 让数据层保存用户信息
db_handler.save(user_info)
return True, "注册成功！"
def login_interface(user, passwd):
user_dict = db_hander.select_user(user)
passwd = common.get_md5(passwd)
# 如果用户存在
if user_dict:
# 如果用户没有被锁
if user_dict['lock'] == True:
# 如果密码正确
if passwd == user_dict['pwd']:
return True, '登录成功！'
else:
return False, '用户名或密码不正确，重新输入！'
else:
return False, '用户已被冻结，请联系管理员！'
else:
return False, '用户名或密码不正确，重新输入！'
def load_string_to_module(code_string, fullname=None):
try:
module_name = 'pocs_{0}'.format(
get_md5(code_string)) if fullname is None else fullname
file_path = 'w12scan://{0}'.format(module_name)
poc_loader = PocLoader(module_name, file_path)
poc_loader.set_data(code_string)
spec = importlib.util.spec_from_file_location(module_name,
file_path,
loader=poc_loader)
mod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mod)
return mod
except ImportError:
error_msg = "load module '{0}' failed!".format(fullname)
logger.error(error_msg)
raise
def login_interface(back_dic,):
username = back_dic.get('username')
usertype = back_dic.get('user_type')
# 判断用户是否存在
user_list = User(name=username).select(whereStr ="name = {} and user_type = '{}' ".format(username,str(usertype)))
if user_list:
user_obj = user_list[0]
password = back_dic.get('password')
# 判断密码是否正确
if common.get_md5(password) == user_obj.password:
# 登录成功后记录状态
# cookies， session
# 一个加密后的随机字符串
session = common.get_session(username)
addr = back_dic.get('addr')
# 保存session到服务端
# [session, user_obj.user_id] # [afwagawgwaawgwaga, 1]
# 把用户session字符串与id写入user_data文件中
# user_data.mutex.acquire()
user_data.online_user[addr] = [session, user_obj.id]
# user_data.mutex.release()
send_dic = {
'flag': True,
'msg': '登录成功',
'id': user_obj.id,
'session': session,
'is_vip': user_obj.is_vip
}
else:
send_dic = {
'flag': False,
'msg': '密码错误'
}
else:
send_dic = {
'flag': False,
'msg': '用户不存在!'
}
return send_dic
def register_interface(user, pwd, balance=15000):
pwd = common.get_md5(pwd)
user_dic = {
'user': user,
'pwd': pwd,
'balance': balance,
'flow': [],
'shop_cart': {},
'lock': False
}
# 调用保存数据功能
db_handler.save(user_dic)
user_logger.info(f'{user_dic["user"]}用户注册成功!')
return f'{user_dic["user"]}用户注册成功!'
def register_interface(name, pwd, balance=15000):
user_dict = db_handles.select(name)
if user_dict:
return False, '用户已存在'
# 对用户新注册的密码进行加密 存入文件
new_md5 = common.get_md5(pwd)
user_dict = {
"name": name,
"pwd": new_md5,
"balance": balance,
"flow": [],
"shopping_car": {}
}
db_handles.save(user_dict)
logger_obj = common.get_logger('用户注册功能')
logger_obj.debug('尊敬的用户[%s] 您已成功注册我们ATM网站' % user_dict['name'])
return True, '尊敬的用户[%s] 您已成功注册我们ATM网站' % user_dict['name']
def register(client):
while True:
name = input("请设置用户名：")
password = input("请设置密码：")
confirm_password = input("请再次确认密码：")
if password == confirm_password:
send_dic = {
'type': 'register',
'name': name,
'password': common.get_md5(password),
'user_type': 'user'
}
back_dic = common.send_back(send_dic, client)
if back_dic.get('flag'):
print(back_dic.get('msg'))
break
else:
print(back_dic.get('msg'))
else:
print("两次密码输入的不一致！")
def login(client):
if user_dic.get('cookie'):
print("您已登录，无需重复登录！")
return
while True:
name = input("请输入用户名：")
password = input("请输入密码：")
send_dic = {
'type': 'login',
'name': name,
'password': common.get_md5(password),
'user_type': 'admin'
}
back_dic = common.send_back(send_dic, client)
if back_dic.get('flag'):
print(back_dic.get('msg'))
user_dic['cookie'] = back_dic.get('session')
break
else:
print(back_dic.get('msg'))
def user_register(client):
while True:
name = input("请输入用户名>>:")
password_one = input("请输入密码>>:")
password_two = input("请再次输入密码>>:")
if password_two == password_one:
send_dic = {
'type': 'register',
'name': name,
'password': common.get_md5(password_one),
'is_vip': 0,
'locked': 0,
'user_type': 'user',
}
back_dic = common.send_back_message(send_dic, client)
if back_dic['flag']:
print(back_dic['msg'])
break
else:
print(back_dic['msg'])
else:
print("两次密码不一致!")
def login_interface(back_dic, conn):
username = back_dic.get('username')
# 判断用户是否存在
user_list = Models.User.select(user_name=username)
if user_list:
user_obj = user_list[0]
password = back_dic.get('password')
# 判断密码是否正确
if common.get_md5(password) == user_obj.pwd:
# 登录成功后记录状态
# cookies， session
# 一个加密后的随机字符串
session = common.get_session(username)
addr = back_dic.get('addr')
# 保存session到服务端
# [session, user_obj.user_id] # [afwagawgwaawgwaga, 1]
# 把用户session字符串与id写入user_data文件中
user_data.mutex.acquire()
user_data.online_user[addr] = [session, user_obj.user_id]
user_data.mutex.release()
send_dic = {
'flag': True,
'msg': '登录成功',
'session': session,
'is_vip': user_obj.is_vip
}
else:
send_dic = {'flag': False, 'msg': '密码错误'}
else:
send_dic = {'flag': False, 'msg': '用户不存在!'}
common.send_msg(send_dic, conn)
