def delete_movie(client):
while True:
print('---删除电影---')
# 1. 获取服务端可以删除的电影
send_dict = {
'type': 'get_movie_list',
'cookies': user_info.get('cookies'),
'movie_type': 'all'
}
recv_dict = common.send_and_recv(send_dict, client)
if not recv_dict.get('flag'):
print(recv_dict.get('msg'))
break
# 2. 打印可删除的电影
# movie_list --> [[电影名，ID，收费免费],[],[]]
movie_list = recv_dict.get('movie_list')
for index, movie_name in enumerate(movie_list):
print(index, movie_name)
choice = input('请输入要删除的电影编号(q.退出):').strip()
if choice == 'q':
break
if not choice.isdigit():
print('请输入数字！')
continue
choice = int(choice)
if choice not in range(len(movie_list)):
print('输入不在范围')
continue
# 获取电影的id，因为名字可能一样，但是数据库中id是唯一的
movie_name_id = movie_list[choice][1]
send_dict = {
'type': 'delete_movie',
'cookies': user_info.get('cookies'),
'movie_id': movie_name_id
}
recv_dict2 = common.send_and_recv(send_dict, client)
if recv_dict2.get('flag'):
print(recv_dict2.get('msg'))
break
def check_download_record(client):
send_dict = {
'type': 'check_record',
'cookies': user_info.get('cookies')
}
recv_dict = common.send_and_recv(send_dict, client)
# 判断有无下载记录
if recv_dict.get('flag'):
record_list = recv_dict.get('record_list')
for line in record_list:
print(line)
else:
print(recv_dict.get('msg'))
def check_movies(client):
print('---查看电影---')
send_dict = {
'type': 'get_movie_list',
'cookies': user_info.get('cookies'),
'movie_type': 'all'
}
recv_dict = common.send_and_recv(send_dict, client)
if recv_dict.get('flag'):
movie_list = recv_dict.get('movie_list')
for index, move in enumerate(movie_list):
print(index, move)
else:
print(recv_dict.get('msg'))
def check_notice(client):
send_dict = {
'type': 'check_notice',
'cookies': user_info.get('cookies')
}
recv_dict = common.send_and_recv(send_dict, client)
# 判断有无公告
if recv_dict.get('flag'):
notice_list = recv_dict.get('notice_list') # -->[[],[],[]]
for line in notice_list:
print(line)
else:
print(recv_dict.get('msg'))
def login(client):
while True:
user_name = input('输入用户名：').strip()
password = input('输入密码：').strip()
send_dict = {
'type': 'login',
'username': user_name,
'password': password,
'user_type': 'admin'
}
recv_dict = common.send_and_recv(send_dict, client)
if recv_dict.get('flag'):
print(recv_dict.get('msg'))
# 登录成功后设置session值
user_info['cookies'] = recv_dict.get('session')
break
else:
print(recv_dict.get('msg'))
def charge_vip(client):
while True:
print('---充值会员---')
is_vip = input('确认充值VIP(y/n)? :').strip()
if is_vip == 'n':
break
elif is_vip == 'y':
send_dict = {
'type': 'charge_vip',
'cookies': user_info.get('cookies')
}
recv_dict = common.send_and_recv(send_dict, client)
else:
print('输入不规范！')
continue
if recv_dict.get('flag'):
print(recv_dict.get('msg'))
break
def register(client):
while True:
user_name = input('请输入用户名：').strip()
password = input('请输入密码：').strip()
re_password = input('确认密码：').strip()
if not password == re_password:
print('两次密码不一致，重新输入！')
continue
send_dict = {
'type': 'register',
'username': user_name,
'password': password,
'user_type': 'admin'
}
recv_dict = common.send_and_recv(send_dict, client)
if recv_dict.get('flag'):
print(recv_dict.get('msg'))
user_info['cookies'] = recv_dict.get('session')
break
else:
print(recv_dict.get('msg'))
def send_notice(client):
while True:
title = input('请输入公告标题(15字以内):').strip()
if len(title) > 15:
continue
content = input('输入公告内容(100字以内):')
if len(content) > 100:
continue
send_dict = {
'type': 'send_notice',
'cookies': user_info.get('cookies'),
'title': title,
'content': content
}
recv_dict = common.send_and_recv(send_dict, client)
if recv_dict.get('flag'):
print(recv_dict.get('msg'))
break
else:
print(recv_dict.get('msg'))
break
def upload_movie(client): # 此处需注意，原文函数名应该不对
while True:
# 获取上传电影目录中的所有电影，并选择
movie_list = common.get_movie_list()
# 若没有可上传的电影
if not movie_list:
print('没有可以上传的电影，请先后台添加后上传')
break
for index, movie_name in enumerate(movie_list):
print(index, movie_name)
choice = input('请选择要上传的电影编号(q.退出)：').strip()
if choice == 'q':
break
if not choice.isdigit():
print('请输入数字！')
continue
choice = int(choice)
if choice not in range(len(movie_list)):
print('输入的不在范围，重新输入！')
continue
# 电影名字
movie_name = movie_list[choice]
# 电影路径
movie_path = os.path.join(setting.UPLOAD_MOVIE_PATH, movie_name)
# 获取电影大小
movie_size = os.path.getsize(movie_path)
# 获取电影md5值
movie_md5 = common.get_movie_md5(movie_path)
# 校验电影是否存在
send_dict = {
'type': 'check_movie',
'cookies': user_info.get('cookies'),
'movie_md5': movie_md5
}
back_dict = common.send_and_recv(send_dict, client)
if not back_dict.get('flag'):
print(back_dict.get('msg'))
continue
# 确认电影是否免费
is_free = input('y/n 免费/收费').strip()
number = 1
if is_free == 'y':
number = 0
# 电影不存在， 发送上传电影请求
send_dict = {
'type': 'upload_movie',
'cookies': user_info.get('cookies'),
'movie_md5': movie_md5,
'movie_name': movie_name,
'movie_size': movie_size,
'is_free': number
}
recv_dict = common.send_and_recv(send_dict, client, file=movie_path)
if recv_dict.get('flag'):
print(recv_dict.get('msg'))
break