Python-Socketserver实现FTP,文件上传下载

Posted 谁吃了我的唐僧肉

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python-Socketserver实现FTP,文件上传下载相关的知识,希望对你有一定的参考价值。

一、Socketserver实现FTP,文件上传、下载

 

  目录结构

    

 

 

 

1、socketserver实现ftp文件上传下载,可以同时多用户登录、上传、下载

  效果图:

  

 

二、上面只演示了下载,上传也是一样的,来不及演示了,上代码

  

 

 1、客户端

  1 import socket,hashlib,os,json,sys,time
  2 
  3 
  4 
  5 class Ftpclient(object):
  6 
  7 
  8     def __init__(self):
  9         self.client = socket.socket()
 10 
 11     def connect(self,ip,port):
 12         self.client.connect((ip, port))
 13 
 14 
 15     def help(self):
 16         msg=\'\'\'
 17         ls 
 18 
 19         pwd 
 20         
 21         cd ..
 22         
 23         get filename
 24         
 25         put filename
 26         
 27         \'\'\'
 28         print(msg)
 29 
 30 
 31     def interactive(self):
 32         """
 33         客户端入口
 34         :return:
 35         """
 36         while True:
 37             verify = self.authenticate() #服务器端认证
 38             if verify:
 39                 while True:
 40                     cmd = input(\'输入命令  >>\').strip()
 41                     if len(cmd) == 0:continue
 42                     cmd_str = cmd.split()[0]
 43                     if hasattr(self,\'cmd_%s\' %cmd_str):
 44                         func = getattr(self,\'cmd_%s\' %cmd_str)
 45                         func(cmd)
 46                     else:
 47                         self.help()
 48 
 49 
 50     def cmd_put(self,*args):
 51         """
 52         上传文件
 53         :param args:
 54         :return:
 55         """
 56         cmd_solit = args[0].split()
 57         start_time = self.alltime()  # 开始时间
 58         if len(cmd_solit) > 1:
 59             filename = cmd_solit[1]
 60             if os.path.isfile(filename):
 61                 filesize = os.stat(filename).st_size
 62                 msg_dic = {
 63                     \'filename\':filename,
 64                     \'size\':filesize,
 65                     \'overridden\':True,
 66                     \'action\':cmd_solit[0]
 67                 }
 68 
 69                 self.client.send( json.dumps(msg_dic).encode(\'utf-8\'))
 70                 server_respinse=self.client.recv(1024)  #防止粘包,等服务器确认返回
 71                 print(\'文件开始上传\',server_respinse)
 72                 client_size = 0
 73                 f = open(filename,\'rb\')
 74                 for line in f:
 75                     client_size += self.client.send(line)
 76                     self.processBar(client_size,filesize)  #进度条
 77                 else:
 78                     print(\'文件传输完毕,大小为 %s\'%client_size)
 79                     end_time = self.alltime()              # 结束时间
 80                     print(\'本次上传花费了%s 秒\'%self.alltime(end_time,start_time))
 81                     f.close()
 82             else:
 83                 print(filename,\'文件不存在\')
 84         else:
 85             print(\'输入有误!\')
 86 
 87 
 88     def cmd_get(self,*args):
 89         """
 90         下载文件
 91         :param args:
 92         :return:
 93         """
 94         cmd_solit = args[0].split()
 95         start_time = self.alltime()  # 开始时间
 96         filename = cmd_solit[1]
 97         if len(cmd_solit) > 1:
 98             msg_dic = {
 99                 \'filename\': filename,
100                 \'size\': \'\',
101                 \'overridden\': True,
102                 \'action\': cmd_solit[0],
103                 \'file_exist\':\'\'
104             }
105             self.client.send(json.dumps(msg_dic).encode(\'utf-8\'))
106             self.data = self.client.recv(1024).strip()
107 
108             cmd_dic = json.loads(self.data.decode(\'utf-8\'))
109             print(cmd_dic)
110             if cmd_dic[\'file_exist\']:
111                 if os.path.isfile(filename):
112                     f = open(filename + \'.new\', \'wb\')
113                 else:
114                     f = open(filename, \'wb\')
115 
116                 self.client.send(b\'200 ok\')     #防止粘包,等服务器确认返回
117                 client_size = 0
118                 filesize = cmd_dic[\'size\']
119                 m = hashlib.md5()
120                 while client_size <  filesize:
121                     data=self.client.recv(1024)
122                     f.write(data)
123                     client_size +=len(data)
124                     m.update(data)
125                     self.processBar(client_size, filesize)
126                 else:
127                     print(\'下载完毕\')
128                     end_time = self.alltime()   # 结束时间
129                     print(\'本次下载花费了%s 秒\' % self.alltime(end_time, start_time))
130                     f.close()
131                     new_file_md5 = m.hexdigest()
132                     server_file_md5 = self.client.recv(1024)
133                     print(\'MD5\', server_file_md5,new_file_md5)
134 
135             else:
136                 print(\'下载的 %s文件不存在\'%filename)
137 
138         else:
139             print(\'输入有误!\')
140 
141 
142     def cmd_dir(self,*arge):
143         cmd_solit = arge[0].split()
144         if len(cmd_solit) > 0:
145             msg_dic = {
146                 \'action\': cmd_solit[0]
147             }
148             self.client.send(json.dumps(msg_dic).encode())
149             cmd_dir = self.client.recv(1024)
150             print(cmd_dir.decode())
151 
152         else:
153             print(\'输入错误!\')
154 
155 
156 
157     def alltime(self,*args):
158         """
159         计算上传、下载时间
160         :param args:
161         :return:
162         """
163         if args:
164             return round(args[0] - args[1])
165         else:
166             return time.time()
167 
168 
169     def processBar(self,num, total):
170         """
171         进度条
172         :param num:文件总大小
173         :param total: 已存入文件大小
174         :return:
175         """
176         rate = num / total
177         rate_num = int(rate * 100)
178         if rate_num == 100:
179             r = \'\\r%s>%d%%\\n\' % (\'=\' * int(rate_num /3), rate_num,)
180         else:
181             r = \'\\r%s>%d%%\' % (\'=\' * int(rate_num /3), rate_num,)
182         sys.stdout.write(r)
183         sys.stdout.flush
184 
185 
186     def authenticate(self):
187         """
188         用户加密认证
189         :return:
190         """
191         username = input(\'输入用户名:>>\')
192         password = input(\'输入密码:>>\')
193         m = hashlib.md5()
194         if len(username) > 0 and len(password) >0:
195             username = \'\'.join(username.split())
196             password = \'\'.join(password.split())
197             m.update(username.encode(\'utf-8\'))
198             m.update(password.encode(\'utf-8\'))
199 
200             m = {
201                 \'username\':username,
202                 \'password\':password,
203                 \'md5\':m.hexdigest()
204             }
205             self.client.send(json.dumps(m).encode(\'utf-8\'))
206             server_user_md5 = self.client.recv(1024).strip()
207             print(server_user_md5.decode())
208             if server_user_md5.decode() == \'success\':
209                 print(\'登录成功!\')
210                 return \'ok\'
211             else:
212                 print(\'用户名密码错误!\')
213         else:
214             print(\'请输入用户名密码\')
215 
216 
217 
218 f = Ftpclient()
219 f.connect(\'localhost\',9999)
220 f.interactive()

2、服务器端

  1 import socketserver,json,os,hashlib,sys,paramiko
  2 
  3 import settings
  4 
  5 class Mysocketserver(socketserver.BaseRequestHandler):
  6 
  7 
  8 
  9     def put(self,*args):
 10         \'\'\'
 11         接受客户端上传文件
 12         :return:
 13         \'\'\'
 14         cmd_dic = args[0]
 15         filename = cmd_dic[\'filename\']  #获取文件名
 16         filesize= cmd_dic[\'size\']       #获取文件大小(字节)
 17 
 18         if os.path.isfile(filename):    #判断文件是否存在
 19             f = open(filename + \'.new\',\'wb\')
 20         else:
 21             f = open(filename, \'wb\')
 22 
 23         self.request.send(b\'200 ok\')    #防止粘包
 24         print(\'%s 文件开始上传\' % self.client_address[0])
 25         received_size = 0
 26         while received_size < filesize:
 27             data = self.request.recv(1024)
 28             f.write(data)
 29             received_size += len(data)
 30         else:
 31             print(\'文件传输完毕\',filename)
 32 
 33 
 34     def get(self, *args):
 35         \'\'\'
 36         客户端下载文件
 37        :return:
 38        \'\'\'
 39         msg_dic = {
 40             \'filename\': \'\',
 41             \'size\': \'\',
 42             \'overridden\': True,
 43             \'action\': \'\',
 44             \'file_exist\': \'\'
 45         }
 46 
 47         cmd_solit = args[0]
 48         filename = cmd_solit[\'filename\']
 49         file_exist = os.path.isfile(filename)
 50         msg_dic[\'file_exist\'] = file_exist
 51         print(file_exist)
 52         if file_exist:
 53             filesize = os.stat(filename).st_size
 54 
 55             msg_dic[\'filename\'] = filename
 56             msg_dic[\'size\'] = filesize
 57             msg_dic[\'action\'] = cmd_solit[\'action\']
 58 
 59             self.request.send(json.dumps(msg_dic).encode(\'utf-8\'))
 60             server_respang = self.request.recv(1024)  #防止粘包
 61             print(\'开始传输\',server_respang)
 62             f = open(filename,\'rb\')
 63             m = hashlib.md5()
 64             for lien in f:
 65                 m.update(lien)
 66                 self.request.send(lien)
 67             else:
 68                 print(\'传输完成\')
 69                 f.close()
 70                 self.request.send(m.hexdigest().encode())
 71         else:
 72             print(\'文件不存在\')
 73             self.request.send(json.dumps(msg_dic).encode(\'utf-8\'))
 74 
 75 
 76 
 77     def client_authentication(self):
 78         """
 79         客户端认证
 80         :return:
 81         """
 82         self.client_user= self.request.recv(1024).strip()
 83         client_xinxi = json.loads(self.client_user.decode(\'utf-8\'))
 84         try:
 85             with open(settings.school_db_file + client_xinxi[\'username\'],\'rb\') as f:
 86                 data = json.load(f)
 87                 if data[\'md5\'] == client_xinxi[\'md5\']: #判断用户输入是否和服务器端MD5是否一致
 88                     print(\'验证成功!\')
 89                     self.request.send(\'success\'.encode())
 90                     return \'success\'
 91                 else:
 92                     self.request.send(\'error\'.encode())
 93         except Exception as e:
 94             print(\'没有此用户\',e)
 95             self.request.send(\'error\'.encode())
 96 
 97 
 98     def dir(self,*args):
 99         """
100         查看目录
101         :param args:
102         :return:
103         """
104         cmd_split = args[0]
105         dd=cmd_split[\'action\']
106         result_os = os.popen(dd).read()
107         self.request.send(result_os.encode())
108 
109 
110 
111     def handle(self):
112         """
113         服务器端入口
114         :return:
115         """
116         while True:
117             try:
118                 success = self.client_authentication()
119                 if success:
120                     self.data=self.request.recv(1024).strip()
121                     cmd_dic = json.loads(self.data.decode(\'utf-8\'))
122                     action = cmd_dic[\'action\']
123                     if hasattr(self,action):
124                         func = getattr(self,action)
125                         func(cmd_dic)
126             except ConnectionResetError as e:
127                 print(\'连接断开\',self.client_address[0])
128                 break
129 
130 
131 
132 if __name__ == \'__main__\':
133 
134     HOST,PORT=\'localhost\',9999
135     server=socketserver.ThreadingTCPServer((HOST,PORT),Mysocketserver)
136     server.serve_forever()

 

settings.py 文件
1 import os
2 
3 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
4 
5 DB_FILE = os.path.join(BASE_DIR, "data\\\\")
6 
7 school_db_file = os.path.join(DB_FILE)
8 print(school_db_file)

data里两个做测试的文件,

1 alex 文件内容:
2 {"username": "alex", "password": "123456", "md5": "94e4ccf5e2749b0bfe0428603738c0f9"}