客户端
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)
admin.py
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)
user.py
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()
common.py
服务端
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)
admin_interface.py
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()
common_interface.py
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)
user_interface.py
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
common.py
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)
socket_server.py
原文地址：https://www.cnblogs.com/Mr-shen/p/12080017.html
时间： 2024-08-01 22:06:08