Python Socketserver实现FTP文件上传下载代码实例

import socket,hashlib,os,json,sys,time
class Ftpclient(object):
def __init__(self):
self.client = socket.socket()
def connect(self,ip,port):
self.client.connect((ip, port))
def help(self):
msg='''
ls 
pwd 
cd ..
get filename
put filename
'''
print(msg)
def interactive(self):
"""
客户端入口
:return:
"""
while True:
verify = self.authenticate() #服务器端认证
if verify:
while True:
cmd = input('输入命令   ').strip()
if len(cmd) == 0:continue
cmd_str = cmd.split()[0]
if hasattr(self,'cmd_%s' %cmd_str):
func = getattr(self,'cmd_%s' %cmd_str)
func(cmd)
else:
self.help()
def cmd_put(self,*args):
"""
上传文件
:param args:
:return:
"""
cmd_solit = args[0].split()
start_time = self.alltime() # 开始时间
if len(cmd_solit)   1:
filename = cmd_solit[1]
if os.path.isfile(filename):
filesize = os.stat(filename).st_size
msg_dic = {
'filename':filename,
'size':filesize,
'overridden':True,
'action':cmd_solit[0]
}
self.client.send( json.dumps(msg_dic).encode('utf-8'))
server_respinse=self.client.recv(1024) #防止粘包，等服务器确认返回
print('文件开始上传',server_respinse)
client_size = 0
f = open(filename,'rb')
for line in f:
client_size += self.client.send(line)
self.processBar(client_size,filesize) #进度条
else:
print('文件传输完毕,大小为 %s'%client_size)
end_time = self.alltime()       # 结束时间
print('本次上传花费了%s 秒'%self.alltime(end_time,start_time))
f.close()
else:
print(filename,'文件不存在')
else:
print('输入有误！')
def cmd_get(self,*args):
"""
下载文件
:param args:
:return:
"""
cmd_solit = args[0].split()
start_time = self.alltime() # 开始时间
filename = cmd_solit[1]
if len(cmd_solit)   1:
msg_dic = {
'filename': filename,
'size': '',
'overridden': True,
'action': cmd_solit[0],
'file_exist':''
}
self.client.send(json.dumps(msg_dic).encode('utf-8'))
self.data = self.client.recv(1024).strip()
cmd_dic = json.loads(self.data.decode('utf-8'))
print(cmd_dic)
if cmd_dic['file_exist']:
if os.path.isfile(filename):
f = open(filename + '.new', 'wb')
else:
f = open(filename, 'wb')
self.client.send(b'200 ok')   #防止粘包，等服务器确认返回
client_size = 0
filesize = cmd_dic['size']
m = hashlib.md5()
while client_size < filesize:
data=self.client.recv(1024)
f.write(data)
client_size +=len(data)
m.update(data)
self.processBar(client_size, filesize)
else:
print('下载完毕')
end_time = self.alltime()  # 结束时间
print('本次下载花费了%s 秒' % self.alltime(end_time, start_time))
f.close()
new_file_md5 = m.hexdigest()
server_file_md5 = self.client.recv(1024)
print('MD5', server_file_md5,new_file_md5)
else:
print('下载的 %s文件不存在'%filename)
else:
print('输入有误！')
def cmd_dir(self,*arge):
cmd_solit = arge[0].split()
if len(cmd_solit)   0:
msg_dic = {
'action': cmd_solit[0]
}
self.client.send(json.dumps(msg_dic).encode())
cmd_dir = self.client.recv(1024)
print(cmd_dir.decode())
else:
print('输入错误！')
def alltime(self,*args):
"""
计算上传、下载时间
:param args:
:return:
"""
if args:
return round(args[0] - args[1])
else:
return time.time()
def processBar(self,num, total):
"""
进度条
:param num:文件总大小
:param total: 已存入文件大小
:return:
"""
rate = num / total
rate_num = int(rate * 100)
if rate_num == 100:
r = '\r%s %d%%\n' % ('=' * int(rate_num /3), rate_num,)
else:
r = '\r%s %d%%' % ('=' * int(rate_num /3), rate_num,)
sys.stdout.write(r)
sys.stdout.flush
def authenticate(self):
"""
用户加密认证
:return:
"""
username = input('输入用户名：  ')
password = input('输入密码：  ')
m = hashlib.md5()
if len(username)   0 and len(password)  0:
username = ''.join(username.split())
password = ''.join(password.split())
m.update(username.encode('utf-8'))
m.update(password.encode('utf-8'))
m = {
'username':username,
'password':password,
'md5':m.hexdigest()
}
self.client.send(json.dumps(m).encode('utf-8'))
server_user_md5 = self.client.recv(1024).strip()
print(server_user_md5.decode())
if server_user_md5.decode() == 'success':
print('登录成功！')
return 'ok'
else:
print('用户名密码错误！')
else:
print('请输入用户名密码')
f = Ftpclient()
f.connect('localhost',9999)
f.interactive()

import socketserver,json,os,hashlib,sys,paramiko
import settings
class Mysocketserver(socketserver.BaseRequestHandler):
def put(self,*args):
'''
接受客户端上传文件
:return:
'''
cmd_dic = args[0]
filename = cmd_dic['filename'] #获取文件名
filesize= cmd_dic['size']    #获取文件大小（字节）
if os.path.isfile(filename):  #判断文件是否存在
f = open(filename + '.new','wb')
else:
f = open(filename, 'wb')
self.request.send(b'200 ok')  #防止粘包
print('%s 文件开始上传' % self.client_address[0])
received_size = 0
while received_size < filesize:
data = self.request.recv(1024)
f.write(data)
received_size += len(data)
else:
print('文件传输完毕',filename)
def get(self, *args):
'''
客户端下载文件
:return:
'''
msg_dic = {
'filename': '',
'size': '',
'overridden': True,
'action': '',
'file_exist': ''
}
cmd_solit = args[0]
filename = cmd_solit['filename']
file_exist = os.path.isfile(filename)
msg_dic['file_exist'] = file_exist
print(file_exist)
if file_exist:
filesize = os.stat(filename).st_size
msg_dic['filename'] = filename
msg_dic['size'] = filesize
msg_dic['action'] = cmd_solit['action']
self.request.send(json.dumps(msg_dic).encode('utf-8'))
server_respang = self.request.recv(1024) #防止粘包
print('开始传输',server_respang)
f = open(filename,'rb')
m = hashlib.md5()
for lien in f:
m.update(lien)
self.request.send(lien)
else:
print('传输完成')
f.close()
self.request.send(m.hexdigest().encode())
else:
print('文件不存在')
self.request.send(json.dumps(msg_dic).encode('utf-8'))
def client_authentication(self):
"""
客户端认证
:return:
"""
self.client_user= self.request.recv(1024).strip()
client_xinxi = json.loads(self.client_user.decode('utf-8'))
try:
with open(settings.school_db_file + client_xinxi['username'],'rb') as f:
data = json.load(f)
if data['md5'] == client_xinxi['md5']: #判断用户输入是否和服务器端MD5是否一致
print('验证成功！')
self.request.send('success'.encode())
return 'success'
else:
self.request.send('error'.encode())
except Exception as e:
print('没有此用户',e)
self.request.send('error'.encode())
def dir(self,*args):
"""
查看目录
:param args:
:return:
"""
cmd_split = args[0]
dd=cmd_split['action']
result_os = os.popen(dd).read()
self.request.send(result_os.encode())
def handle(self):
"""
服务器端入口
:return:
"""
while True:
try:
success = self.client_authentication()
if success:
self.data=self.request.recv(1024).strip()
cmd_dic = json.loads(self.data.decode('utf-8'))
action = cmd_dic['action']
if hasattr(self,action):
func = getattr(self,action)
func(cmd_dic)
except ConnectionResetError as e:
print('连接断开',self.client_address[0])
break
if __name__ == '__main__':
HOST,PORT='localhost',9999
server=socketserver.ThreadingTCPServer((HOST,PORT),Mysocketserver)
server.serve_forever()

import os
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
DB_FILE = os.path.join(BASE_DIR, "data\\")
school_db_file = os.path.join(DB_FILE)
print(school_db_file)