开发简单的FTP:
1. 用户登陆
2. 上传/下载文件
3. 不同用户家目录不同
4. 查看当前目录下文件
5. 充分使用面向对象知识
REDMAE
1 用户登陆 2 3 1、查看用户目录文件 4 2、上传文件, 5 3、下载方件 6 4、退出 7 8 程序结构: 9 socket_server_client/#程序目录 10 |- - -clients/#client程序主目录 11 | |- - -__init__.py 12 | |- - -bin/#启用目录 13 | | |- - - __init__.py 14 | | |- - -socket_client.py#客户端启动 15 | | 16 | |- - -cfg/#配置文件目录 17 | | |- - - __init__.py 18 | | |- - -config.py#配置文件 19 | | 20 | |- - -core/#主要程序目录 21 | | |- - - __init__.py 22 | | |- - -client_func.py#主要函数 23 | | 24 | |- - -home/#客户端下载文件目录 25 | 26 |- - -servers/#server程序主目录 27 | |- - -__init__.py 28 | |- - -bin/#启用目录 29 | | |- - - __init__.py 30 | | |- - -registration.py#用户注册 31 | | |- - -socket_server.py#服务端启动 32 33 | | 34 | |- - -cfg/#配置文件目录 35 | | |- - - __init__.py 36 | | |- - -config.py#配置文件 37 | | 38 | |- - -core/#主要程序目录 39 | | |- - - __init__.py 40 | | |- - -server_classc.py#主要函数 41 | | 42 | |- - -db/#用户上传文件主目录 43 | |- - -user_file/#用户上传目录 44 | |- - -user_names#注册用户文件 45 |
服务端
servers/
bin/
registration.py
![](https://images.cnblogs.com/OutliningIndicators/ExpandedBlockStart.gif)
1 #!usr/bin/env python 2 #-*-coding:utf-8-*- 3 # Author calmyan 4 import socket,os,json,sys,pickle 5 6 BASE_DIR=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#获取相对路径转为绝对路径赋于变量 7 sys.path.append(BASE_DIR)#增加环境变量 8 from cfg import config 9 print('用户注册'.center(60,'=')) 10 while True: 11 user_=input('请输入您要注册的用户名:').strip() 12 user_dir=os.path.join(config.USER_DIR,user_)#拼接用户目录路径 13 if os.path.isdir(user_dir):# 判断一个目录是否存在 14 print('用户已经存在请重输!') 15 continue 16 else: 17 pwd_=input('请输入密码:').strip() 18 pwd_two=input('请确认密码:').strip() 19 if pwd_==pwd_two: 20 try: 21 os.mkdir(user_dir)#创建目录 22 except Exception as e: 23 print(e) 24 continue 25 finally: 26 file_dir=user_dir+'\user'#用户目录下的用户名文件 27 if not os.path.isfile(config.USER_FILE): 28 with open(config.USER_FILE,'w',encoding='utf-8') as f: 29 f.write('{}') 30 with open(config.USER_FILE,'r+',encoding='utf-8') as f: 31 data=eval(f.readline()) 32 data[user_]=pwd_ 33 f.seek(0) 34 f.write(str(data)) 35 print('用户[%s]注册成功!'%user_) 36 exit()
socket_server.py
1 #!usr/bin/env python 2 #-*-coding:utf-8-*- 3 # Author calmyan 4 import socket,os,json 5 import sys 6 BASE_DIR=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#获取相对路径转为绝对路径赋于变量 7 sys.path.append(BASE_DIR)#增加环境变量 8 9 from core.server_class import listen_func 10 s=socket.socket()#定义 11 s.bind(('localhost',9000))#绑定要监听的 端口 12 s.listen(5)#对列5 13 print('正在监听中') 14 listen_func(s)
cfg/
config.py
![](https://images.cnblogs.com/OutliningIndicators/ExpandedBlockStart.gif)
1 #!usr/bin/env python 2 #-*-coding:utf-8-*- 3 # Author calmyan 4 import os ,sys 5 BASE_DIR=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#获取相对路径转为绝对路径赋于变量 6 sys.path.append(BASE_DIR)#增加环境变量 7 8 9 USER_DIR=BASE_DIR+'db\user_file\'#定义用户目录文件路径变量 10 11 USER_FILE=BASE_DIR+'db\user_names'#定义用户文件路径变量
core/
server_class.py
![](https://images.cnblogs.com/OutliningIndicators/ExpandedBlockStart.gif)
1 #!usr/bin/env python 2 #-*-coding:utf-8-*- 3 # Author calmyan 4 import socket,os,json,sys,pickle 5 6 BASE_DIR=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#获取相对路径转为绝对路径赋于变量 7 sys.path.append(BASE_DIR)#增加环境变量 8 9 from cfg import config 10 11 12 #用户名检测函数 13 def open_file_list(name,pas):#传入当前类 14 with open(config.USER_FILE,'r',encoding='utf-8') as f: 15 data=eval(f.readline()) 16 if name in data and pas==data[name]: 17 return True 18 else: 19 return False 20 21 22 #连接类 23 class socket_server(object): 24 '''连接类''' 25 file_path=config.USER_DIR#用户路经变量 26 def __init__(self,data,conn):#传入用户名,密码 27 self.DATA=data 28 self.conn=conn 29 30 31 def show_process(self,lens): 32 received_size=0#定义大小 33 current_percent=0#当前大小百分比 34 while received_size<lens: 35 if int((received_size/lens)*100)>current_percent: 36 print('#',end='',flush=True) 37 current_percent=int((received_size/lens)*100) 38 new_size=yield 39 received_size+=new_size 40 41 def ret_l(self): 42 ret=socket_server.login(self.DATA["name"],self.DATA['pwd'],self.conn)#用户名检测 43 return ret 44 def open_f(self,ret):#打开目录 45 file_dir=os.path.join(socket_server.file_path,ret['data']['user'])#用户目录 46 file_name=os.listdir(file_dir)#目录文件列表 47 f=file_dir+self.DATA['filename']##上传的文件名 48 return file_dir,file_name,f#返回 49 50 def ls_file(self,data):#查看文件 51 self.conn.send(json.dumps(data[1]).encode()) 52 53 def send_file(self,data): 54 55 if self.DATA['filename'] in data[1]: 56 f=data[0]+'/'+self.DATA['filename'] 57 file_obj=open(f,"rb")#打开文件 58 name=file_obj.name.split('/')[-1]#文件名 59 sez=os.path.getsize(f)#获取文件大小 60 print(sez) 61 data_header={ 62 "action":"put", 63 "filename":name, 64 "size":sez 65 } 66 self.conn.send(json.dumps(data_header).encode())#用json 序列化后,发送相关 信息 67 for line in file_obj: 68 self.conn.send(line)#发送数据 69 70 elif self.DATA['filename'].isdigit(): 71 num=int(self.DATA['filename'])#转为数字 72 try: 73 f=data[0]+'/'+data[1][num]# 74 file_obj=open(f,"rb")#打开文件 75 name=file_obj.name.split('/')[-1]#文件名 76 sez=os.path.getsize(f)#获取文件大小 77 print(sez) 78 data_header={ 79 "action":"put", 80 "filename":name, 81 "size":sez 82 } 83 self.conn.send(json.dumps(data_header).encode())#用json 序列化后,发送相关 信息 84 for line in file_obj: 85 self.conn.send(line)#发送数据 86 self.conn.send(json.dumps(f).encode())#发送文件 87 except Exception as e: 88 data={'filename':False} 89 self.conn.send(json.dumps(data).encode()) 90 else: 91 data={'filename':False} 92 self.conn.send(json.dumps(data).encode()) 93 def put_file(self,data):#上传文件 94 file_obj=open(data[2],'wb')#打开新建 这个文件 95 rece_size=0#定义 文件大小值 96 #prten=socket_server.show_process(self.DATA["size"]) 97 #prten.__next__() 98 while rece_size<self.DATA["size"]:#小于接收的文件大小时, 99 recv_data=self.conn.recv(4096) 100 file_obj.write(recv_data)#写入文件 101 rece_size+=len(recv_data)#增加文件大小计算 102 103 else: 104 print("文件[%s]接收完毕!"%self.DATA["filename"]) 105 file_obj.flush() 106 file_obj.close()#关闭文件 107 108 109 @staticmethod 110 def login(name,pas,conn):#用户检测 函数 111 try: 112 if open_file_list(name,pas): 113 tag=True 114 error='' 115 datas={'user':name} 116 data={'mag':'用户认证通过','tag':True} 117 print(json.dumps(data).encode()) 118 conn.send(json.dumps(data).encode()) 119 else: 120 raise Exception(' 33[41;1m用户名或密码错误 33[0m' %name) 121 except Exception as e: 122 tag=False 123 error=str(e) 124 datas='' 125 data={'mag':'用户或密码错误','tag':False} 126 print('发送数据%s'%data) 127 conn.send(json.dumps(data).encode()) 128 return {'tag':tag,'error':error,'data':datas} 129 130 #监听函数 131 def listen_func(s): 132 while True: 133 conn,client_addr=s.accept()#端口监听中....返回两个值 ,联接编号对象 , ip 134 print('获取到新连接:',client_addr) 135 while True: 136 data=conn.recv(4096)#接收数据 指令 137 print('接收的数据:',data) 138 data= json.loads(data.decode())#反序列 139 if len(data)==0: 140 break 141 if data['action']=='user':#如果是用户名,进行认证 142 serv=socket_server(data,conn) 143 ret=serv.ret_l() 144 if ret['tag']: 145 pass 146 else: 147 continue 148 149 print(data) 150 if data['action']=="put":#如果接收的字典中是put,就是进行接收 151 serv=socket_server(data,conn) 152 serv.put_file(serv.open_f(ret))#调对象方法 153 elif data['action']=='get':#下载 154 serv=socket_server(data,conn)#实例化 155 serv.send_file(serv.open_f(ret))#调 用方法 156 elif data['action']=='ls':#查看 157 serv=socket_server(data,conn) 158 serv.ls_file(serv.open_f(ret)) 159 continue
客户端
clients/
bin/
socket_client.py
![](https://images.cnblogs.com/OutliningIndicators/ExpandedBlockStart.gif)
1 #!usr/bin/env python 2 #-*-coding:utf-8-*- 3 # Author calmyan 4 5 import socket,os,json,sys 6 BASE_DIR=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#获取相对路径转为绝对路径赋于变量 7 sys.path.append(BASE_DIR)#增加环境变量 8 from core.client_func import user_pwd 9 #from core.client_func import show_process 10 from cfg import config 11 12 #进度条 13 def show_process(lens): 14 received_size=0#定义大小 15 current_percent=0#当前大小百分比 16 while received_size<lens: 17 if int((received_size/lens)*100)>current_percent: 18 print('#',end='',flush=True) 19 current_percent=int((received_size/lens)*100) 20 new_size=yield 21 received_size+=new_size 22 23 24 client=socket.socket() 25 client.connect(('localhost',9000)) 26 while True: 27 data_d=user_pwd(client) 28 if data_d['tag']:#运行#用户名登陆成功 29 while True: 30 print('''=====指令提示==== 31 查看目录文件: ls 32 下载文件: get 文件名 或 文件编号 如: get test.txt 或 get 1 33 上传方件: put 路径/文件名 如 put e:/test.txt 34 退出:exit 35 ''') 36 cho=input('指令 >>:').strip() 37 if len(cho)==0:continue 38 if cho=='exit':exit()#退出指令 39 cmd_list=cho.split() 40 if cmd_list[0]=='put':#如果等于下载指令 41 if len(cmd_list)==1: 42 print('没有输入相关文件名') 43 continue 44 filename=cmd_list[1] 45 if os.path.isfile(filename):#如果文件存在 46 file_obj=open(filename,"rb")#打开文件 47 name=file_obj.name.split('/')[-1]#文件名 48 #name=filename.split("\")[-1]#文件名 49 sez=os.path.getsize(filename)#获取文件大小 50 data_header={ 51 "action":"put", 52 "filename":name, 53 "size":sez 54 } 55 client.send(json.dumps(data_header).encode())#用json 序列化后,发送相关 信息 56 57 print("文件[%s]发送中...."%data_header["filename"]) 58 for line in file_obj: 59 client.send(line) 60 else: 61 print("文件[%s]发送完毕!"%data_header["filename"]) 62 else: 63 print('该文件不存在') 64 continue 65 elif cmd_list[0]=='get':#如查等 于上传指令 66 if len(cmd_list)==1: 67 print('没有输入相关文件名') 68 continue 69 filename=cmd_list[1] 70 print(filename) 71 data_header={ 72 "action":"get", 73 "filename":filename, 74 "size":'' 75 } 76 client.send(json.dumps(data_header).encode())#用json 序列化后,发送相关 信息 77 datas=client.recv(4096)#接收数据 指令 78 data_l= json.loads(datas.decode())#反序列 79 if not data_l['filename']: 80 print('文件不存在') 81 continue 82 file_dir=config.USER_DIR+data_l["filename"] 83 file_obj=open(file_dir,'wb')#打开新建 这个文件 84 rece_size=0#定义 文件大小值 85 prten=show_process(data_l["size"]) 86 prten.__next__() 87 while rece_size<data_l["size"]:#小于接收的文件大小时, 88 recv_data=client.recv(4096) 89 file_obj.write(recv_data)#写入文件 90 rece_size+=len(recv_data)#增加文件大小计算 91 try: 92 prten.send(len(recv_data)) 93 except StopIteration as e: 94 print('100%') 95 96 else: 97 print("文件[%s]接收完毕!"%data_l["filename"]) 98 file_obj.flush() 99 file_obj.close()#关闭文件 100 elif cmd_list[0]=='ls':#查看目录文件 101 data_header={ 102 "action":"ls", 103 "filename":'', 104 "size":'' 105 } 106 client.send(json.dumps(data_header).encode())#用json 序列化后,发送相关 信息 107 datas=client.recv(4096)#接收数据 指令 108 data_l= json.loads(datas.decode())#反序列 109 for k,v in enumerate(data_l): 110 print('编号: %s 文件名:%s'%(k,v)) 111 112 else: 113 print(data_d['mag'])
cfg/
config.py
1 #!usr/bin/env python 2 #-*-coding:utf-8-*- 3 # Author calmyan 4 5 import os ,sys 6 BASE_DIR=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))#获取相对路径转为绝对路径赋于变量 7 sys.path.append(BASE_DIR)#增加环境变量 8 9 10 USER_DIR=BASE_DIR+'home\'#定义用户目录文件路径变量
core/
client_func.py
![](https://images.cnblogs.com/OutliningIndicators/ExpandedBlockStart.gif)
1 #!usr/bin/env python 2 #-*-coding:utf-8-*- 3 # Author calmyan 4 import socket,os,json,sys 5 #用户名登陆函数 6 def user_pwd(client): 7 user_=input('请输入用户名:').strip() 8 pwd_=input('请输入密码:').strip() 9 data_header={ 10 "action":"user", 11 "name":user_, 12 "pwd":pwd_ 13 } 14 client.send(json.dumps(data_header).encode())#用json 序列化后,发送相关 信息 15 data=client.recv(4096)#接收数据 指令 16 data_s=json.loads(data.decode('utf-8'))#反序列 17 return data_s 18 19 20 #进度条 21 def show_process(lens): 22 received_size=0#定义大小 23 current_percent=0#当前大小百分比 24 while received_size<lens: 25 if int((received_size/lens)*100)>current_percent: 26 print('#',end='',flush=True) 27 current_percent=int((received_size/lens)*100) 28 new_size=yield 29 received_size+=new_size