服务端：

1 import socket  2 import socketserver  3 import struct  4 import json  5 import os  6 import time  7 import subprocess  8   9 #os.path.abspath(__file__) 显示当前文件的绝对路径 10 #os.path.dirname           显示当前文件/目录 所在的目录 11 #os.path.normpath()        规范路径，在Linux平台自动转换路径 12  13  14 flag="FBI" 15  16 class Ftp_server(socketserver.BaseRequestHandler): 17     code="utf-8" 18     def handle(self): 19         while True: 20             pwd=os.getcwd() 21             os.chdir(pwd) 22  23             recv_head_len=struct.unpack("i",self.request.recv(4))[0] #接收 客户端发来固定的报头长度数字（字节格式） 24             recv_head_data=json.loads(self.request.recv(recv_head_len)) #接收报头信息 (字典类型) 25                                                  #根据客户端发来的报头，执行响应的方法，fla=FBI登录 26             if recv_head_data.get("flag")==flag:# print(recv_head_data)    # d1 = {"name": name, "pwd": pwd,"cmd":cmd,"flag":flag} 27                 self.login(recv_head_data) 28                 print(recv_head_data) 29             if hasattr(self,recv_head_data["cmd"]):#其他情况 就是 put 和 get操作 30                 cmd=recv_head_data["cmd"] 31                 file=recv_head_data['file_name'] 32                 print(cmd,file) 33                 func=getattr(self,cmd) 34                 func(cmd,file,recv_head_data) 35  36  37     def put(self,cmd,file,head_data):  #上传 38         data_size =head_data["data_size"] 39         print(data_size) 40         res =head_data 41  42         file_path = os.path.normpath(os.path.join(r"E:\full_stack\day38"  #拼接用户路径 43                                                   'put', 44                                                   res["name"], 45                                                   res['pwd'])) 46  47                                                 #姓名+密码=用户目录 48         if os.path.exists(file_path) == True:  # 如果用户存在 49             os.chdir(file_path)                 #判断用户目录下的文件是否存在 50             os.path.exists(file) 51             if os.path.exists(file):  # 如果文件存在 就代表可以断定续传,打开这个文件，先记录一下末尾的tell 52                 print(file) 53                 with open(file, 'rb') as f: 54                     f.seek(0, 2) 55                     sek = f.tell() 56                     print("文件的sek%s" % sek) 57  58                 if head_data["data_size"] ==sek:  #如果末尾的位置 和文件的大小 相等 就代表已经存在了 无需上传 59                     self.request.send(struct.pack('i',sek)) 60                     print("上传的文件已经完成。") 61  62  63                 else:                               #如果文件不完整，就代表断点续传 64                     self.request.send(struct.pack('i',sek)) 65                     with open(file, 'wb') as f: 66                         f.seek(sek) 67                         recv_size = 0 68                         while recv_size < data_size:  # 开写数据 69                             data = self.request.recv(1024) 70                             f.write(data) 71                             recv_size += len(data) 72  73  74             else:                                   # 如果文件不存在，从头开始写 75                 self.request.send(struct.pack('i', 0)) 76                 with open(file, 'wb') as f: 77                     recv_size = 0 78                     while recv_size < data_size:  # 开写数据 79                         data = self.request.recv(1024) 80                         f.write(data) 81                         recv_size += len(data) 82                 self.request.send("上传完成".encode("utf-8")) 83                 time.sleep(1) 84         else: #如果用户目录不存在 85             os.makedirs(file_path) 86             os.chdir(file_path) 87             self.request.send(struct.pack('i',0)) 88             with open(file, 'wb') as f: 89                 recv_size = 0 90                 while recv_size < data_size:  # 开写数据 91                     data = self.request.recv(1024) 92                     f.write(data) 93                     recv_size += len(data) 94             self.request.send("上传完成".encode("utf-8")) 95             time.sleep(1) 96  97     def get(self,cmd,file,head_data):  #下载方法 98         filesize=os.path.getsize(file) 99         d3={
   "name":file,"data_size":filesize}100         bytes_d1 = json.dumps(d3).encode("utf-8")101         len_d1 = len(bytes_d1)102         struct_len = struct.pack("i",len_d1)103         self.request.send(struct_len)104         self.request.send(bytes_d1)105         with open(file,'rb') as f:106             for i in f:107                 self.request.send(i)108             else:f.close()109         time.sleep(2)110         self.request.send("接收完毕".encode("utf-8"))111 112     def login(self,recv_head_data):#{'file_path': 'alex\\123', 'cmd': 'login', 'file_name': 'c'}113 114         file_path=os.path.normpath(os.path.join(r"E:\full_stack\day38"115                                       'put',116                                       recv_head_data["name"],117                                       recv_head_data['pwd']))118         os.chdir(file_path)119         cmd = recv_head_data['cmd']120         print(cmd)121         act_res = subprocess.Popen(cmd, shell=True,122                                    stdout=subprocess.PIPE,123                                    stdin=subprocess.PIPE,124                                    stderr=subprocess.PIPE)125         act_err = act_res.stderr.read()126         if act_err:127             ret = act_err128         else:129             ret = act_res.stdout.read()130         bytes_len = len(ret)131         self.request.send(struct.pack("i",bytes_len))132         self.request.send(ret)133 134 135 if __name__ == '__main__':136     obj=socketserver.ThreadingTCPServer(("127.0.0.1",8080),Ftp_server)137     obj.serve_forever()

 

 

客户端

1 import socket  2 import struct  3 import json  4 import os  5 import mode  6 flag="FBI"  7 user_dict={
   "alex":'123','egon':'456'}     #DB  8   9 l=["put 文件名", 'get 文件名', "login 登录" ] 10  11 #类的 __init__方法在对象实例化时自动执行，所以实例化之后对象就执行了__init__中定义的东西 12 class Client(): 13     def __init__(self,Addr_server): 14         self.addr_server=Addr_server 15         self.socket = socket.socket() 16         self.client_connect() 17  18     def client_connect(self): 19         self.socket.connect(self.addr_server) 20  21     def run(self): 22         while True: 23             name_inpu = input("请输入用户名: ") 24             psw_inpu = input("请输入密码:").strip() 25             if user_dict.get(name_inpu) == psw_inpu: 26                     for i in l: 27                         print(i.center(50)) 28             else: 29                 print("用户名或者密码错误!".center(50)) 30                 continue 31  32             flie_inpu = input("请输入FTP操作:  ").strip() 33             if flie_inpu=="login": 34                 self.login(name_inpu,psw_inpu) 35             else: 36                 cmd, file = flie_inpu.split() 37                 print(cmd,file) 38                 if hasattr(mode, cmd):  # 根据不同的请求，设置不同的报头的格式。 39                     func = getattr(mode, cmd) 40                     struct_len,bytes_d1,d1 = func(name_inpu,psw_inpu,cmd,file) 41                 else: 42                     continue 43                 self.socket.send(struct_len) # 44                 self.socket.send(bytes_d1) 45                 if hasattr(self, cmd): #根据不同的报头格式，执行不同的方法 46                     func = getattr(self, cmd) 47                     func(d1) 48                 else: 49                     continue 50  51     def get(self,d1): #下载操作 52         server_head_len=struct.unpack("i",self.socket.recv(4))[0]   #接收 服务端报头的长度 53         server_heaa_data=json.loads(self.socket.recv(server_head_len))#接收服务端报头信息 54         print(server_heaa_data) 55         server_data_size=server_heaa_data['data_size'] 56         server_data_name=server_heaa_data['name'] 57         with open(server_data_name,'wb') as f: 58             recv_size=0 59             while recv_size < server_data_size:  # 开下载数据 60                         data = self.socket.recv(1024) 61                         f.write(data) 62                         recv_size += len(data) 63             else: 64                 print(self.socket.recv(1024).decode("utf-8")) 65  66     def put(self,d1):  #上传操作 67         print(type(d1)) 68         struct_sek=self.socket.recv(4) 69         sek=struct.unpack('i',struct_sek)[0] 70         print("传来的sek%s"% sek) 71         print(d1[ 'data_size']) 72         if sek==d1[ 'data_size']: 73             print("已经上传") 74         else: 75             with open(d1["file_name"],'rb') as f:  # 上传 76                 f.seek(sek) 77                 for i in f: 78                     self.socket.send(i) 79  80     def login(self,name,pwd):               # 登录 81         while True:                       #循环发送报头给服务端，让服务端做判断，返回执行结果； 82             cmd=input('请输入命令-------->').strip() 83             d1={
   "name": name, "pwd": pwd,"cmd":cmd,"flag":flag} 84             bytes_d1 = json.dumps(d1).encode("utf-8") 85             len_dl = len(bytes_d1) 86             struct_len = struct.pack("i", len_dl) 87             self.socket.send(struct_len) 88             self.socket.send(bytes_d1) 89             data_len=struct.unpack("i",self.socket.recv(4))[0] 90             res=self.socket.recv(data_len).decode("gbk") 91             print(res) 92  93  94  95  96  97  98  99 100 101 102 103 104 105 c=Client(("127.0.0.1",8080))106 c.run()