欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-03-10 11:43:19
一、Socketserver实现FTP,文件上传、下载 目录结构 1、socketserver实现ftp文件上传下载,可以同时多用户登录、上传、下载 效果图: 二、上面只演示了下载,上传也是一样的,来不及演示了,上代码 1、客户端 1 import socket,hashlib,os,json,sy ......

一、socketserver实现ftp,文件上传、下载

 

  目录结构

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

 

 

 

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

  效果图:

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

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

Python-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"}
kml123456文件内容:
{"username": "kml123456", "password": "123456","md5": "a791650e70ce08896e3dafbaa7598c26"}

 

到这里差不多就没了,