def add_group_interface(name):
user = select_user(name)
if user:
group_list = get_group_chat_list()
print("群列表")
print(group_list)
group_name = input("输入群名:")
group = select_group_chat(group_name)
if group is not None:
group.add_user(name)
save_group_chat(group)
user.add_group_chat(group_name)
save_user(user)
user_logger.info('%s 加入了%s' % (name, group_name))
else:
flg = input("是否建群? 1 yes, 2 no")
if flg == '1':
from lib.GroupChat import GroupChat
group = GroupChat(group_name)
group.add_user(name)
save_group_chat(group)
user.add_group_chat(group_name)
save_user(user)
elif flg == '2':
pass
else:
pass
def repay_lib(name, account):
user_dic = db_handler.select_user(name)
user_dic['balance'] += account
user_dic['bankflow'].append('%s还款%s元' % (name, account))
db_handler.save_user(user_dic)
bank_logger.info('%s还款%s元' % (name, account))
return True, '还款成功'
def add_friend_interface(name):
user = select_user(name)
if user:
users_list = get_user_list()
print("用户列表")
print(users_list)
friend_name = input('输入朋友姓名')
if friend_name in users_list:
user.add_friend(friend_name)
save_user(user)
friend = select_user(friend_name)
friend.add_friend(name)
save_user(friend)
return True, '添加成功'
else:
return False, '没有该用户'
def login(name, password):
user_dic = db_handler.select_user(name)
if user_dic:
if user_dic['password'] == password:
return True, '登入成功'
else:
return False, '密码错误'
else:
return False, '账户名错误'
def withdraw_lib(name, account):
user_dic = db_handler.select_user(name)
if user_dic['balance'] > account:
user_dic['balance'] -= account
user_dic['bankflow'].append('%s取款:%s' % (name, account))
db_handler.save_user(user_dic)
bank_logger().info('取款:%s' % account)
return True, '取款功能'
else:
return False, '您的余额不足'
def check_account():
print('查看账户')
name = input('请输入你要查看的账户名(q退出)>>>:').strip()
if name == 'q': return
user_dic = db_handler.select_user(name)
if user_dic:
print(user_dic)
return
else:
print('账户名不存在')
def login_interface(name, password):
user = select_user(name)
if user:
if user.check_login_info(password):
user_logger.info('%s登录了' % name)
return True, '登陆成功'
else:
return False, '用户密码错误或已经锁定'
else:
return False, '用户不存在'
def pay_lib(name, cost):
user_dic = db_handler.select_user(name)
if user_dic['balance'] > cost:
user_dic['balance'] -= cost
user_dic['bankflow'].append('%s购物支付了%s' % (name, cost))
db_handler.save_user(user_dic)
bank_logger.info('%s购物支付了%s' % (name, cost))
return True, '支付成功'
else:
return False, '余额不足'
def login_interface(name, password):
now_user = db_handler.select_user(name)
if now_user:
if now_user.password == password:
logging.info('%s 登陆' % name)
return True, '欢迎您，' + name, now_user
else:
return False, '用户密码错误', None
else:
return False, '用户不存在', None
def register_interface(name, password):
user = select_user(name)
if user:
return False, '用户已经存在'
else:
from lib.User import User
new_user = User(name, password)
save_user(new_user)
user_logger.info('%s 注册了' % name)
return True, '注册成功'
def transfer_lib(from_name, to_name, account):
if from_name == to_name:
return False, '不能给自己转账'
to_dic = db_handler.select_user(to_name)
if to_dic:
from_dic = db_handler.select_user(from_name)
if account > from_dic['balance']:
return False, '你的余额不足'
else:
from_dic['balance'] -= account
from_dic['bankflow'].append('您向%s转账%s元' %
(to_dic['name'], account))
to_dic['balance'] += account
to_dic['bankflow'].append('您收到%s转账%s元' %
(from_dic['name'], account))
db_handler.save_user(from_dic)
db_handler.save_user(to_dic)
bank_logger.info('向%s转账%s元' % (to_dic['name'], account))
return True, '转账成功'
else:
return False, '对方账户不存在'
def login(conn, message):
status, name, password = message.split('|')
status = int(status)
now_user = db_handler.select_user(name, status)
is_success = 0
is_member = 0
if status == 0:
is_member = now_user.is_member
if now_user:
if now_user.password == password:
is_success = 1
msg = str(is_success) + '|' + str(is_member)
msg = msg.encode('utf-8')
conn.send(msg)
def lock():
print('锁定账户')
name = input('请输入你要锁定的账户名(q退出)>>>:').strip()
if name == 'q': return
user_dic = db_handler.select_user(name)
if user_dic:
if user_dic['locked'] == True:
print('该账户已经锁定')
return
else:
user_dic['locked'] = True
db_handler.save_user(user_dic)
print('该账户锁定成功')
return
else:
print('账户名不存在')
def unlock():
print('解锁账户')
name = input('请输入你要解锁的账户名(q退出)>>>:').strip()
if name == 'q': return
user_dic = db_handler.select_user(name)
if user_dic:
if user_dic['locked'] == False:
print('该账户已经解锁')
return
else:
user_dic['locked'] = False
db_handler.save_user(user_dic)
print('该账户解锁成功')
return
else:
print('账户名不存在')
def register_lib(name, password):
user_dic = db_handler.select_user(name)
if user_dic:
return False, '账户已存在'
else:
user_dic = {
'name': name,
'password': password,
'locked': False,
'balance': 2000,
'credit': 2000,
'bankflow': []
}
db_handler.save_user(user_dic)
user_shop = {'name': name, 'shoppingcart': {}}
db_handler.save_user_shop(user_shop)
user_logger.info('%s注册成功' % name)
return True, '注册成功'
def un_lock_user_interface(name):
user = select_user(name)
if user:
user.unlock()
save_user(user)
user_logger.info('%s解锁了' % name)
def change_password_lib(name, pwd):
user_dic = db_handler.select_user(name)
user_dic['password'] = pwd
db_handler.save_user(user_dic)
return True, '密码修改成功'
def check_record_lib(name):
user_dic = db_handler.select_user(name)
return user_dic['bankflow']
def get_friends_list(name):
user = select_user(name)
if user:
return user.friends_list
def get_groups_list(name):
user = select_user(name)
if user:
return user.groups_list
def get_balance(name):
# 获取登入账户信息
user_dic = db_handler.select_user(name)
return user_dic['balance'], user_dic['credit']
def lock_user(name):
user_dic = db_handler.select_user(name)
user_dic['locked'] = True
db_handler.save_user(user_dic)
def lock_user_interface(name):
user = select_user(name)
if user:
user.lock()
save_user(user)
user_logger.info('%s被锁定了' % name)
