客户端
core文件夹
from lib import common from conf import setting import os userinfo = {'cookies': None} def register(client): while True: username = input('请输入用户名：').strip() password = input('请输入密码：').strip() re_password = input('请确认密码：').strip() if password == re_password: send_dic = { 'username': username, 'password': password, 'user_type': 'admin', 'func_type': 'register' } back_dic = common.send_and_back(send_dic, client) if back_dic.get('flag'): print(back_dic.get('msg')) break else: print(back_dic.get('msg')) else: print('两次密码不一致！！！') continue def login(client): while True: username = input('请输入用户名：').strip() password = input('请输入密码：').strip() send_dic = { 'username': username, 'password': password, 'func_type': 'login' } back_dic = common.send_and_back(send_dic, client) if back_dic.get('flag'): print(back_dic.get('msg')) userinfo['cookies'] = back_dic.get('session') break else: print(back_dic.get('msg')) def upload_movie(client): while True: upload_movie_dir = setting.UPLOAD_MOVIE_PATH movie_list = os.listdir(upload_movie_dir) 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(upload_movie_dir, movie_name) movie_size = os.path.getsize(movie_path) movie_md5 = common.get_movie_md5(movie_path) send_dic = { 'func_type': 'check_movie', 'movie_md5': movie_md5, 'cookies': userinfo.get('cookies') } back_dic = common.send_and_back(send_dic, client) if back_dic.get('flag'): choice2 = input('请输入是否是收费的（y or n):').strip() is_free = 0 if choice2 == 'y': is_free = 1 send_dic2 = { 'func_type': 'upload_movie', 'movie_md5': movie_md5, 'cookies': userinfo.get('cookies'), 'movie_name': movie_name, 'movie_size': movie_size, 'is_free': is_free } back_dic2 = common.send_and_back(send_dic2, client, file=movie_path) if back_dic2.get('flag'): print(back_dic2.get('msg')) break else: print(back_dic.get('msg')) def delete_movie(client): while True: send_dic = { 'func_type': 'get_movie_list', 'cookies': userinfo.get('cookies'), 'movie_type': 'all' } back_dic = common.send_and_back(send_dic, client) if not back_dic.get('flag'): print(back_dic.get('msg')) break movie_list = back_dic.get('movie_list') for index, movie_name_id in enumerate(movie_list): print(index, movie_name_id) if not movie_list: print('没有可删除的电影了') break choice = input('请输入需要删除的电影编号或q退出：').strip() if choice == 'q': break if not choice.isdigit(): continue choice = int(choice) if choice not in range(len(movie_list)): continue movie_name_id = movie_list[choice] send_dic = { 'func_type': 'delete_movie', 'cookies': userinfo.get('cookies'), 'movie_id': movie_name_id[1] } back_dic2 = common.send_and_back(send_dic, client) if back_dic2.get('flag'): print(back_dic2.get('msg')) break def send_notice(client): title = input('请输入公告标题：').strip() content = input('请输入公告内容：').strip() send_dic = { 'title': title, 'content': content, 'cookies': userinfo.get('cookies'), 'func_type': 'send_notice' } back_dic = common.send_and_back(send_dic, client) print(back_dic.get('msg')) func_dic = { '1': register, '2': login, '3': upload_movie, '4': delete_movie, '5': send_notice, } def admin_view(client): while True: print(''' 1.注册 2.登录 3.上传视频 4.删除视频 5.发布公告 q.退出 ''') choice = input('请输入功能编号：').strip() if choice == 'q': break if choice not in func_dic: print('请输入功能内编号！！！') continue func_dic[choice](client)
from lib import common from conf import setting import time import os user_info = {'cookies': None, 'is_vip': 0} def register(client): while True: username = input('请输入用户名：').strip() password = input('请输入密码：').strip() re_password = input('请确认密码：').strip() if password == re_password: send_dic = { 'username': username, 'password': password, 'user_type': 'user', 'func_type': 'register' } back_dic = common.send_and_back(send_dic, client) if back_dic.get('flag'): print(back_dic.get('msg')) break else: print(back_dic.get('msg')) def login(client): while True: username = input('请输入用户名：').strip() password = input('请输入密码：').strip() send_dic = { 'func_type': 'login', 'username': username, 'password': password } back_dic = common.send_and_back(send_dic, client) if back_dic.get('flag'): print(back_dic.get('msg')) user_info['cookies'] = back_dic.get('session') is_vip = back_dic.get('is_vip') if is_vip: user_info['is_vip'] = is_vip break else: print(back_dic.get('msg')) def buy_vip(client): if not user_info.get('is_vip'): choice = input('请输入y确认冲会员（会员费10000元）,输入其他键退出：').strip() if choice == 'y': send_dic = { 'cookies': user_info.get('cookies'), 'func_type': 'buy_vip' } back_dic = common.send_and_back(send_dic, client) print(back_dic.get('msg')) login(client) else: print('穷b！！！') else: print('您已经是会员了') def check_movies(client): send_dic = { 'func_type': 'get_movie_list', 'movie_type': 'all', 'cookies': user_info.get('cookies') } back_dic = common.send_and_back(send_dic, client) if back_dic.get('flag'): print(back_dic.get('movie_list')) else: print(back_dic.get('msg')) def download_free_movie(client): while True: send_dic = { 'func_type': 'get_movie_list', 'movie_type': 'free', 'cookies': user_info.get('cookies') } back_dic = common.send_and_back(send_dic, client) if not back_dic.get('flag'): print(back_dic.get('msg')) break movie_list = back_dic.get('movie_list') if not movie_list: print('没有可下载的免费的电影') break for index, movie_id_name in enumerate(movie_list): print(index, movie_id_name) choice = input('请输入要下载的电影编号').strip() if not choice.isdigit(): print('请输入数字') continue choice = int(choice) if choice not in range(len(movie_list)): print('请输入编范围内的数字') continue movie_name, movie_id, movie_type = movie_list[choice] send_dic2 = { 'cookies': user_info.get('cookies'), 'func_type': 'download_movie', 'movie_id': movie_id, 'movie_name': movie_name } back_dic2 = common.send_and_back(send_dic2, client) wait_time = back_dic2.get('wait_time') if wait_time: print('广告时间10s') time.sleep(wait_time) movie_size = back_dic2.get('movie_size') movie_path = os.path.join(setting.DOWNLOAD_MOVIE_PATH, movie_name) recv_data = 0 with open(movie_path, 'wb') as f: while recv_data < movie_size: data = client.recv(1024) f.write(data) recv_data += len(data) break def download_pay_movie(client): while True: send_dic = { 'cookies': user_info.get('cookies'), 'func_type': 'get_movie_list', 'movie_type': 'pay' } back_dic = common.send_and_back(send_dic, client) if not back_dic.get('flag'): print(back_dic.get('msg')) break movie_list = back_dic.get('movie_list') if not movie_list: print('没有可下载的电影') break for index, movie_name_id in enumerate(movie_list): print(index, movie_name_id) choice = input('请输入y确认购买电影（会员：200元，普通用户800元），输入其他退出：').strip() if choice == 'y': if user_info.get('is_vip'): print('会员购买电影成功，开始选择电影...') else: print('普通用户购买电影成功，开始选择电影...') else: print('穷b， 没钱还想看电影，滚犊子吧') break choice2 = input('请输入需要下载电影的编号：').strip() if not choice2.isdigit(): continue choice2 = int(choice2) if choice2 not in range(len(movie_list)): continue movie_name, movie_id, movie_type = movie_list[choice2] send_dic2 = { 'cookies': user_info.get('cookies'), 'func_type': 'download_movie', 'movie_name': movie_name, 'movie_id': movie_id } back_dic2 = common.send_and_back(send_dic2, client) movie_size = back_dic2.get('movie_size') movie_path = os.path.join(setting.DOWNLOAD_MOVIE_PATH, movie_name) recv_data = 0 with open(movie_path, 'wb') as f: while recv_data < movie_size: data = client.recv(1024) f.write(data) recv_data += len(data) break def check_download_record(client): send_dic = { 'func_type': 'check_download_record', 'cookies': user_info.get('cookies') } back_dic = common.send_and_back(send_dic, client) if back_dic.get('flag'): print(back_dic.get('download_movie_list')) else: print(back_dic.get('msg')) def check_notice(client): send_dic = { 'func_type': 'check_notice', 'cookies': user_info.get('cookies') } back_dic = common.send_and_back(send_dic, client) if back_dic.get('flag'): print(back_dic.get('notice_list')) else: print(back_dic.get('msg')) func_dic = { '1': register, '2': login, '3': buy_vip, '4': check_movies, '5': download_free_movie, '6': download_pay_movie, '7': check_download_record, '8': check_notice } def user_view(client): while True: print(''' 1 注册 2 登录 3 冲会员 4 查看视频 5 下载免费视频 6 下载收费视频 7 查看下载记录 8 查看公告 q 退出 ''') choice = input('请输入功能编号：').strip() if choice == 'q': break if choice not in func_dic: continue func_dic[choice](client)
lib文件夹
import struct import hashlib import json import os def send_and_back(send_dic, client, file=None): # 发送报头和真实数据 json_bytes = json.dumps(send_dic).encode('utf-8') header = struct.pack('i', len(json_bytes)) client.send(header) client.send(json_bytes) if file: with open(file, 'rb') as f: for i in f: client.send(i) # 接收服务端返回的数据 headers = client.recv(4) bytes_len = struct.unpack('i', headers)[0] json_bytes_data = client.recv(bytes_len).decode('utf8') back_dic = json.loads(json_bytes_data) return back_dic def get_movie_md5(movie_path): movie_size = os.path.getsize(movie_path) index_list = [0, movie_size//3, (movie_size//3)*2, movie_size-10] md5_obj = hashlib.md5() with open(movie_path, 'rb') as f: for line in index_list: f.seek(line) data = f.read(10) md5_obj.update(data) return md5_obj.hexdigest()
服务端
interface文件夹
from db.models import Movie from db.models import Notice import datetime from lib import common from threading import Lock from conf import setting import os mutex = Lock() @common.login_auth def check_movie_interface(back_dic, conn): movie_md5 = back_dic.get('movie_md5') movie_obj_list = Movie.select_data(movie_md5=movie_md5) if movie_obj_list: send_dic = { 'flag': False, 'msg': '电影已经存在！！' } else: send_dic = {'flag': True, 'msg': '可以继续上传'} common.send_data(send_dic, conn) @common.login_auth def upload_movie_interface(back_dic, conn): # 保证电影的名字唯一性 movie_name = common.get_session() + back_dic.get('movie_name') movie_path = os.path.join(setting.MOVIE_DIR, movie_name) movie_size = back_dic.get('movie_size') recv_data = 0 with open(movie_path, 'wb') as f: while recv_data < movie_size: data = conn.recv(1024) f.write(data) recv_data += len(data) movie_obj = Movie( movie_name=movie_name, movie_size=movie_size, movie_md5=back_dic.get('movie_md5'), is_free=back_dic.get('is_free'), is_delete=0, upload_time=str(datetime.datetime.now()), user_id=back_dic.get('user_id')) movie_obj.insert_data() send_dic = { 'flag': True, 'msg': '电影上传成功' } common.send_data(send_dic, conn) @common.login_auth def delete_movie_interface(back_dic, conn): movie_id = back_dic.get('movie_id') movie_obj = Movie.select_data(m_id=movie_id)[0] movie_obj.is_delete = 1 movie_obj.update_data() send_dic = { 'flag': True, 'msg': '电影删除成功' } common.send_data(send_dic, conn) @common.login_auth def send_notice_interface(back_dic, conn): notice_obj = Notice( title=back_dic.get('title'), content=back_dic.get('content'), create_time=str(datetime.datetime.now()), user_id=back_dic.get('user_id')) notice_obj.insert_data() send_dic = { 'msg': '公告发布成功' } common.send_data(send_dic, conn)
from lib import common from db.models import Movie, DownLoadRecord from db.models import User from db import user_data import datetime from threading import Lock from conf import setting import os mutex = Lock() # 未删除电影接口 @common.login_auth def get_movie_list_interface(back_dic, conn): movie_obj_list = Movie.select_data() if movie_obj_list: movie_list = [] for movie_obj in movie_obj_list: if not movie_obj.is_delete: if back_dic.get('movie_type') == 'all': movie_list.append([movie_obj.movie_name, movie_obj.m_id, '免费电影' if movie_obj.is_free else '收费电影']) elif back_dic.get('movie_type') == 'free': if movie_obj.is_free: movie_list.append([movie_obj.movie_name, movie_obj.m_id, '免费电影']) else: if not movie_obj.is_free: movie_list.append([movie_obj.movie_name, movie_obj.m_id, '收费电影']) send_dic = { 'flag': True, 'movie_list': movie_list } else: send_dic = { 'flag': False, 'msg': '服务端没有电影' } common.send_data(send_dic, conn) def register_interface(back_dic, conn): username = back_dic.get('username') user_obj_list = User.select_data(user_name=username) if user_obj_list: send_dic = { 'flag': False, 'msg': '用户已存在！' } else: pwd = common.get_md5(back_dic.get('password')) user_obj = User( user_name=username, user_pwd=pwd, user_type=back_dic.get('user_type'), register_time=str(datetime.datetime.now()) ) user_obj.insert_data() send_dic = { 'flag': True, 'msg': f'用户{username}注册成功' } common.send_data(send_dic, conn) def login_interface(back_dic, conn): user_obj = User.select_data(user_name=back_dic.get('username'))[0] if not user_obj: send_dic = {'flag': False, 'msg': '用户不存在'} else: if user_obj.user_pwd == common.get_md5(back_dic.get('password')): addr = back_dic.get('addr') session = common.get_session() mutex.acquire() user_data.user_online[addr] = [session, user_obj.user_id] mutex.release() send_dic = { 'flag': True, 'msg': '登录成功！', 'session': session } if user_obj.is_vip: send_dic['is_vip'] = user_obj.is_vip else: send_dic = {'flag': False, 'msg': '密码错误！！！'} common.send_data(send_dic, conn) @common.login_auth def download_movie_interface(back_dic, conn): user_id = back_dic.get('user_id') user_obj = User.select_data(user_id=user_id)[0] movie_name = back_dic.get('movie_name') movie_path = os.path.join(setting.MOVIE_DIR, movie_name) send_dic = { 'movie_size': os.path.getsize(movie_path), 'wait_time': 0 } if not user_obj.is_vip: send_dic['wait_time'] = 10 common.send_data(send_dic, conn, file=movie_path) download_obj = DownLoadRecord( movie_id=back_dic.get('movie_id'), user_id=user_id, download_time=str(datetime.datetime.now()) ) download_obj.insert_data()
from db.models import User, DownLoadRecord, Movie, Notice from lib import common @common.login_auth def buy_vip_interface(back_dic, conn): user_id = back_dic.get('user_id') user_obj = User.select_data(user_id=user_id)[0] user_obj.is_vip = 1 user_obj.update_data() send_dic = {'msg': '充值会员成功，请重新登录'} common.send_data(send_dic, conn) @common.login_auth def check_download_record_interface(back_dic, conn): user_id = back_dic.get('user_id') down_load_obj_list = DownLoadRecord.select_data(user_id=user_id) if not down_load_obj_list: send_dic = { 'flag': False, 'msg': '没有下载记录' } else: download_list = [] for download_obj in down_load_obj_list: movie_id = download_obj.movie_id movie_obj = Movie.select_data(m_id=movie_id)[0] download_list.append([movie_obj.movie_name, download_obj.download_time]) send_dic = { 'flag': True, 'download_movie_list': download_list } common.send_data(send_dic, conn) @common.login_auth def check_notice_interface(back_dic, conn): notice_obj_list = Notice.select_data() if not notice_obj_list: send_dic = { 'flag': False, 'msg': '暂无公告' } common.send_data(send_dic, conn) else: notice_list = [] for notice_obj in notice_obj_list: notice_list.append([notice_obj.title, notice_obj.content, notice_obj.create_time]) send_dic = { 'flag': True, 'notice_list': notice_list } common.send_data(send_dic, conn)
lib文件夹
import struct import json import hashlib import uuid from threading import Lock from db.user_data import user_online mutex = Lock() def send_data(back_dic, conn, file=None): bytes_data = json.dumps(back_dic).encode('utf-8') header = struct.pack('i', len(bytes_data)) conn.send(header) conn.send(bytes_data) if file: with open(file, 'rb') as f: for i in f: conn.send(i) def get_md5(pwd): md5_obj = hashlib.md5() md5_obj.update(pwd.encode('utf-8')) salt = '天王盖地虎' md5_obj.update(salt.encode('utf-8')) return md5_obj.hexdigest() def get_session(): uuid_str = str(uuid.uuid4()) session_val = get_md5(uuid_str) return session_val def login_auth(func): def inner(*args, **kwargs): # args = (back_dic, conn) client_session = args[0].get('cookies') addr = args[0].get('addr') mutex.acquire() # 'addr': [session, user_obj.user_id] session_id_list = user_online.get(addr) mutex.release() if session_id_list: if client_session == session_id_list[0]: args[0]['user_id'] = session_id_list[1] res = func(*args, **kwargs) return res else: send_dic = {'flag': False, 'msg': '携带session值错误'} send_data(send_dic, args[1]) else: send_dic = {'flag': False, 'msg': '用户未登录，没有执行权限'} send_data(send_dic, args[1]) return inner
tcp_server文件夹
from interface import admin_interface from interface import common_interface from interface import user_interface import socket import struct import json from concurrent.futures import ThreadPoolExecutor from threading import Lock from db import user_data mutex = Lock() pool = ThreadPoolExecutor(50) func_dic = {'register': common_interface.register_interface, 'login': common_interface.login_interface, 'check_movie': admin_interface.check_movie_interface, 'upload_movie': admin_interface.upload_movie_interface, 'get_movie_list': common_interface.get_movie_list_interface, 'delete_movie': admin_interface.delete_movie_interface, 'send_notice': admin_interface.send_notice_interface, 'buy_vip': user_interface.buy_vip_interface, 'download_movie': common_interface.download_movie_interface, 'check_download_record': user_interface.check_download_record_interface, 'check_notice': user_interface.check_notice_interface} def working(conn, addr): while True: try: headers = conn.recv(4) bytes_len = struct.unpack('i', headers)[0] json_data = conn.recv(bytes_len).decode('utf-8') back_dic = json.loads(json_data) # 保证每个客户端过来都是唯一的，将addr添加到客户端发过来的字典中 back_dic['addr'] = str(addr) func_type = back_dic.get('func_type') if func_type in func_dic: func_dic[func_type](back_dic, conn) except Exception as e: print(e) # 清除当前客户端存放在服务端的session值 mutex.acquire() user_data.user_online.pop(str(addr)) mutex.release() break conn.close() def run(): server = socket.socket() server.bind(('127.0.0.1', 9876)) server.listen(50) while True: conn, addr = server.accept() print(addr) # 异步提交任务 pool.submit(working, conn, addr)