python之FTP程序(支持多用户在线)

一、需求

1. 用户加密认证 (完成)
2. 允许同时多用户登录 (完成)
3. 每个用户有自己的家目录 ，且只能访问自己的家目录(完成)
4. 对用户进行磁盘配额，每个用户的可用空间不同(完成)
5. 允许用户在ftp server上随意切换目录cd(完成)
6. 允许用户查看当前目录下文件ls(完成)
7. 允许上传put和下载get文件(完成)，保证文件一致性(此需求不做)
8. 文件传输过程中显示进度条(完成)
附加功能：

 1.新建目录mkdir(完成)

 2.查看当前工作目录的路径pwd(完成)

 3.支持文件的断点续传(未完成)



二、目录结构及模块功能解释

ftp_client

  |----bin(可执行目录)

  |         |----__init__.py

  |         |----ftp_client.py(客户端接口)   

  |----conf(配置文件目录)

  |　　   |----__init__.py

  |　　   |----settings.py(配置文件) 

  |----core(核心代码)

  |　　   |----__init__.py

  |　　   |----auth.py(客户端身份验证)

  |　　   |----cd.py(实现客户端在服务随意切换目录的功能，但只能访问自己的家目录)

  |　　   |----get.py(客户端下载功能)

  |　　   |----interactive.py(用于客户端与服务端的交互/反射)

  |　　   |----ls.py(查看当前目录下的文件(包括目录))

  |　　   |----main.py(主函数，运行被ftp_client.py客户端接口调用)

  |　　　|----mkdir.py(实现用户在当前目录下可创建目录的功能)

  |　　　|----progress_bar.py(进度条:用于显示上传与下载的进度)

  |　　　|----put.py(处理客户端上传功能)

  |　　　|----pwd.py(查看用户当前的目录)

  |----__init__.py




ftp_server

  |----bin

  |   　　|----__init__.py

  |　　   |----ftp_server.py(服务端接口)

  |----core

  |　　   |----__init__.py

  |   　　|----auth.py(用户加密认证，登陆模块)

  |　　   |----db_handle.py(读用户数据与写用户数据--感觉这个模块有点多余~)

  |   　　|----deal_cd.py(处理用户切换目录的功能)

  |　　   |----deal_get.py(处理客户端下载文件的请求)

  |   　　|----deal_ls.py(完成用户显示当前目录下文件(包括目录)的请求)

  |　　   |----deal_mkdir.py(处理用户在当前目录(家目录下)创建目录的请求)

  |   　　|----deal_put.py(处理客户端上传文件的请求)

  |　　   |----deal_pwd.py(用来处理客户端查看当前目录下的请求)

  |   　　|----get_dirisize.py(获取用户家目录的大小(字节))

  |　　   |----main.py(主函数--运行时被ftp_server.py服务端接口调用)

  |----data(用户数据库)

  |　　   |----__init__.py

  |   　　|----Alex.json(Alex用户的数据库)

  |　　   |----zcl.json(zcl用户的数据库)

  |----home(home目录,用来存放各用户的家目录)

  |　　   |----Alex(Alex的家目录)

  |   　　|----zcl(zcl的家目录)

  |　　   |----__init__.py

  |----log(日志--未拓展)

  |　　   |----__init__.py

  |----__init__.py



三、状态码

LOGIN_STATE = {
    "auth_True":"200",   #认证成功
    "auth_False":"400",  #认证失败
    "file_exit":"202",   #文件存在
    "file_no_exit":"402", #文件不存在
    "cmd_right":"201",  #命令正确
    "cmd_error":"401",  #命令错误
    "dir_exit":"203",   #目录已存在
    "dir_no_exit":"403", #目录不存在
    "cmd_success":"204",  #命令成功执行
    "cmd_fail":"404",      #命令执行失败
    "size_enough":"205", #磁盘空间足够
    "size_empty":"405",  #磁盘空间不足
}


四、功能解释

 1.conf目录下settings.py模块记录可操作用户信息，根据用户信息生成用户字典和宿主目录，已经生成的不再新建。
 2.每个用户的宿主目录磁盘空间配额默认为100M，可在settings.py模块里进行修改
 3.程序运行在windows8.1系统上，pycharm 3.4，程序需求除断点续传与保证文件一致性外全部实现。
 4.切换目录: cd .. 返回上一级目录   cd dirname  进入dirname  eg:cd \aa
    用户登陆后默认进入宿主目录，只可在宿主目录下随意切换.
 5.创建目录：mkdir dirname
                 在当前目录下创建目录，如果目录存在则报错，不存在创建.
 6.查看当前目录完整路径: pwd
 7.查看当前路径下的文件名和目录名: ls
 8.下载文件(不可续传)：get filename
     ①、服务端当前目录存在此文件，客户端不存在此文件，直接下载.
        ②、服务端当前目录存在此文件，客户端存在此文件名，之前下载中断，文件不可续传(未实现).
        ③、服务端当前目录存在此文件，客户端存在此文件名，下载，文件名为filename+".new".

 9.上传文件：put  filename
        判断宿主目录磁盘空间是否够用，可以，上传文件；否则，报错.

 1 import os,json
 2 
 3 
 4 def client_get(self,*args):
 5 """
 6 用来处理客户端下载功能
 7 """
 8 cmd_split=args[0].split()
 9 if len(cmd_split)>1:
10     filename=cmd_split[1]
11     msg_dic={                        # 为了可拓展性，用字典形式
12     "action":"get",        #发送给服务端的指令
13     "filename":filename,
14     "overridden":True
15     }
16 self.client.send(json.dumps(msg_dic).encode())
17 #防止粘包，等服务器确认
18 #可优化，确认同时服务端看客户端是否有权限等404403(状态码)
19 server_response=self.client.recv(1024).decode()
20 print(server_response,type(server_response))
21 self.client.send("客户端已准备好下载".encode())

 1 import os,json
 2 from conf import settings
 3 
 4 def server_deal_get(self,*args):
 5     """处理客户端下载文件的请求"""
 6     cmd_dic=args[0]
 7     filename=cmd_dic["filename"]
 8     if os.path.isfile(filename):
 9         file_size=os.stat(filename).st_size#服务端文件大小
10         msg_dic={
11         "file_size":file_size,#服务端将发给客户端的文件的大小
12         "file_exit":settings.LOGIN_STATE["file_exit"]
13         }
14         self.request.send(json.dumps(msg_dic).encode())
15         #防止粘包，服务端与客户端再进行一次交互
16         client_response=self.request.recv(1024)
17         print(client_response.decode())
18     else:
19 self.request.send(json.dumps(settings.LOGIN_STATE["file_no_exit"]).encode())

C:\Python34\python3.exe C:/Users/Administrator/PycharmProjects/laonanhai/ftp/ftp_client/bin/ftp_client.py
Username:zcl
Password:abc
******************Welcome Login*******************
>>>:get oldboy-25.avi
{"file_exit": "402", "file_size": 180251848} <class ‘str‘>
>>>:get aa
"403" <class ‘str‘>
>>>:get oldboy-25.avi
Traceback (most recent call last):
  File "C:/Users/Administrator/PycharmProjects/laonanhai/ftp/ftp_client/bin/ftp_client.py", line 15, in <module>
    main.run()
  File "C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_client\core\main.py", line 53, in run
    ftp_client.interactive()
  File "C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_client\core\main.py", line 31, in interactive
    interactive.interactive(self)
  File "C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_client\core\interactive.py", line 14, in interactive
    func(cmd)
  File "C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_client\core\main.py", line 38, in cmd_get
    get.client_get(self, *args)
  File "C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_client\core\get.py", line 20, in client_get
    server_response = self.client.recv(1024).decode()
ConnectionAbortedError: [WinError 10053] 你的主机中的软件中止了一个已建立的连接。

Process finished with exit code 1

1 s=b‘\xe5\xae\xa2\xe6\x88\xb7\xe7\xab\xaf\xe5\xb7\xb2\xe5\x87\x86\xe5\xa4\x87\xe5\xa5\xbd\xe4\xb8\x8b\xe8\xbd\xbd‘
2 print(type(s))
3 s1=str(s,encoding="utf-8")
4 print(s1)
5 
6 ss="客户端已准备好下载"
7 b=bytes(ss,encoding="utf-8")
8 print(b)

C:\Python34\python3.exe C:/Users/Administrator/PycharmProjects/laonanhai/ftp/ftp_server/core/test.py
<class ‘bytes‘>
客户端已准备好下载
b‘\xe5\xae\xa2\xe6\x88\xb7\xe7\xab\xaf\xe5\xb7\xb2\xe5\x87\x86\xe5\xa4\x87\xe5\xa5\xbd\xe4\xb8\x8b\xe8\xbd\xbd‘

Process finished with exit code 0

 驱动器 C 中的卷没有标签。
 卷的序列号是 000C-3580

 C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server 的目录

2017/01/10 周二  上午 01:02    <DIR>          .
2017/01/10 周二  上午 01:02    <DIR>          ..
2017/01/10 周二  上午 12:43    <DIR>          bin
2017/01/10 周二  下午 08:54    <DIR>          conf
2017/01/10 周二  下午 10:09    <DIR>          core
2017/01/10 周二  上午 01:31    <DIR>          data
2017/01/10 周二  上午 11:45    <DIR>          home
2016/11/02 周三  下午 09:42    <DIR>          log
2016/11/02 周三  下午 09:41                 0 __init__.py
               1 个文件              0 字节
               8 个目录 33,612,865,536 可用字节


Process finished with exit code 0

 1 """
 2 客户端接口
 3 """
 4 
 5 import os,sys
 6 
 7 
 8 dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 9 sys.path.insert(0, dir)
10 
11 from core import main
12 
13 if __name__ == "__main__":
14     main.run()

 1 """
 2 客户端配置
 3 """
 4 
 5 LOGIN_STATE = {
 6     "auth_True":"200",   #认证成功
 7     "auth_False":"400",  #认证失败
 8     "file_exit":"202",   #文件存在
 9     "file_no_exit":"402", #文件不存在
10     "cmd_right":"201",  #命令正确
11     "cmd_error":"401",  #命令错误
12     "dir_exit":"203",   #目录已存在
13     "dir_no_eixt":"403", #目录不存在
14     "cmd_success": "204",  # 命令成功执行
15     "cmd_fail": "404",  # 命令执行失败
16     "size_enough": "205",  # 磁盘空间足够
17     "size_empty": "405",  # 磁盘空间不足
18 }

 1 """
 2 客户端身份验证
 3 """
 4 from conf import settings
 5 import hashlib
 6 
 7 def send_auth(self):
 8     """
 9     发送用户名:密码到服务端，进行登陆验证
10     :param self:
11     :return: 返回400(成功)或200(失败)
12     """
13     user_name = input("Username:")
14     password = input("Password:")
15     #客户端发送用户名与密码(列表)
16     password_hash = hash(password)
17     user_data = "%s:%s" % (user_name, password_hash)
18     self.client.send(user_data.encode())
19     #接收服务端返回，认证成功True，认证失败False
20     auth_recv = self.client.recv(1024).decode()
21     if auth_recv == settings.LOGIN_STATE["auth_True"]:
22         print("Welcome Login".center(50, "*"))
23         return  auth_recv
24     elif auth_recv == settings.LOGIN_STATE["auth_False"]:
25         print("usename or password not exist")
26         return auth_recv
27 
28 
29 
30 def hash(data):
31     #对密码进行md5加密
32     m = hashlib.md5()
33     m.update(data.encode())
34     #返回加密后的数据
35     return m.hexdigest()

 1 import json
 2 from conf import settings
 3 
 4 
 5 def client_cd(self, *args):
 6     """实现客户端在服务随意切换目录的功能，但只能访问自己的家目录"""
 7     cmd_split = args[0].split()
 8     if len(cmd_split) > 1:
 9         cd_dir = cmd_split[1]
10         msg_dic = {             # 为了可拓展性，用字典形式
11             "action": "cd",   # 发送给服务端的指令
12             "cd_dir": cd_dir,
13         }
14         self.client.send(json.dumps(msg_dic).encode())
15         server_response = self.client.recv(1024)
16         print(server_response.decode())
17     else:
18         print("%s:命令错误" % settings.LOGIN_STATE["cmd_error"])

 1 import os,json
 2 from conf import settings
 3 from core import progress_bar
 4 
 5 def client_get(self, *args):
 6     """
 7     用来处理客户端下载功能
 8     """
 9     cmd_split = args[0].split()
10     if len(cmd_split) > 1:
11         filename = cmd_split[1]
12         msg_dic = {             # 为了可拓展性，用字典形式
13             "action": "get",  # 发送给服务端的指令
14             "filename": filename,
15             "overridden": True
16         }
17         self.client.send(json.dumps(msg_dic).encode())
18         # 防止粘包，等服务器确认
19         # 可优化，确认同时服务端看客户端是否有权限等404 403(标准码)
20         server_response = json.loads(self.client.recv(1024).decode())
21         print(server_response,type(server_response))
22         if server_response["file_exit"] == settings.LOGIN_STATE["file_exit"]:
23             self.client.send("客户端已准备好下载".encode())
24             if os.path.isfile(msg_dic["filename"]):    #文件已经存在
25                 f = open(filename + ".new", "wb")
26             else:
27                 f = open(filename, "wb")
28             receive_size = 0
29             while receive_size < server_response["file_size"]:
30                 data = self.client.recv(1024)
31                 receive_size += len(data)
32                 #调用progress_bar模块的方法
33                 progress_bar.progress_bar(self, receive_size, server_response["file_size"])
34                 f.write(data)
35             else:
36                 print("download from server success")
37 
38         elif server_response["file_exit"] == settings.LOGIN_STATE["file_no_exit"]:
39             print("%s:请求文件不存在" % server_response["file_exit"])

 1 def interactive(self):
 2     """
 3     本模块用于客户端与服务端的交互
 4     """
 5     while True:
 6         cmd = input(">>>:").strip()
 7         if len(cmd) == 0:
 8             continue
 9         cmd_str = cmd.split()[0]  # 指令
10         if hasattr(self, "cmd_%s" % cmd_str):  # 反射
11             func = getattr(self, "cmd_%s" % cmd_str) #获得方法对应的内存地址
12             func(cmd)
13         else:
14             self.help()

 1 import json
 2 
 3 
 4 def client_ls(self, *args):
 5     """查看当前目录下的文件(包括目录)"""
 6     cmd_split = args[0].split()
 7     if len(cmd_split) == 1 and cmd_split[0] == "ls":
 8         msg_dic = {
 9             "action":"ls",
10         }
11         self.client.send(json.dumps(msg_dic).encode())
12         server_response = self.client.recv(1024)
13         print(server_response.decode())

 1 import socket
 2 import hashlib
 3 import os
 4 import json
 5 from core import interactive
 6 from core import put
 7 from core import get
 8 from core import auth
 9 from conf import settings
10 from core import pwd
11 from core import mkdir
12 from core import ls
13 from core import cd
14 
15 class FtpClient(object):
16     def __init__(self):
17         self.client = socket.socket()
18 
19     def help(self):
20         msg = """
21         ls
22         pwd
23         cd ../..
24         get filename
25         put filename
26         """
27         print(msg)
28 
29     def connect(self, ip, port):
30         self.client.connect((ip, port))
31 
32 
33     def interactive(self):   #交互
34         interactive.interactive(self
35 )
36 
37     def cmd_put(self, *args):  #*args是为了将来参数的拓展
38         put.client_put(self, *args)
39 
40     def cmd_get(self, *args):
41         get.client_get(self, *args)
42 
43     def send_auth_data(self):
44         #返回用户帐号名与密码
45         login_state = settings.LOGIN_STATE["auth_False"]
46         while login_state != settings.LOGIN_STATE["auth_True"]:
47             login_state = auth.send_auth(self)
48 
49     def cmd_pwd(self, *args):
50         pwd.client_pwd(self, *args)
51 
52     def cmd_mkdir(self, *args):
53         mkdir.client_mkdir(self, *args)
54 
55     def cmd_ls(self, *args):
56         ls.client_ls(self, *args)
57 
58     def cmd_cd(self, *args):
59         cd.client_cd(self, *args)
60 
61 
62 def run():
63     ftp_client = FtpClient()
64     ftp_client.connect("localhost", 8787)
65     #身份验证
66     ftp_client.send_auth_data()
67     #客户端与服务端交户
68     ftp_client.interactive()

 1 import json
 2 from conf import settings
 3 
 4 
 5 def client_mkdir(self, *args):
 6     """实现用户在当前目录下可创建目录的功能"""
 7     cm_split = args[0].split()
 8     if len(cm_split) > 1:
 9         new_dir = cm_split[1]
10         msg_dic = {
11             "action":"mkdir",
12             "new_dir":new_dir,  #将新建的目录
13             "overriden":False, #已存在的目录不可覆盖
14         }
15         self.client.send(json.dumps(msg_dic).encode())
16         server_response = self.client.recv(1024)
17         print(json.loads(server_response.decode()))
18 
19     else:
20         print("%s:命令错误" % settings.LOGIN_STATE["cmd_error"])

 1 import sys
 2 
 3 
 4 def progress_bar(self, num, total):
 5     """
 6     进度条:用于显示上传与下载的进度
 7   :return: 无
 8     """
 9     rate = num/total
10     rate_num = int(rate * 100)
11     r = "\r%s%d%%" % ("|" * rate_num, rate_num)  #/r表示重新回到当前行输出
12     sys.stdout.write(r)   #输出没有换行符
13     sys.stdout.flush()   #清空缓存

 1 import os,json
 2 from conf import settings
 3 from core import progress_bar
 4 
 5 
 6 def client_put(self, *args):
 7     """
 8     用于处理客户端上传功能
 9     """
10     cmd_split = args[0].split()  # 列表
11     if len(cmd_split) > 1:
12         filename = cmd_split[1]
13         if os.path.isfile(filename):  # 判断是否存在文件
14             file_size = os.stat(filename).st_size
15             msg_dic = {             # 为了可拓展性，用字典形式
16                 "action": "put",  # 发送给服务端的指令
17                 "filename": filename,
18                 "file_size": file_size,
19                 "overridden": True
20             }
21             self.client.send(json.dumps(msg_dic).encode())
22             # 防止粘包，等服务器确认
23             # 可优化，确认同时服务端看客户端是否有权限等404 403(标准码)
24             server_response = json.loads(self.client.recv(1024).decode())
25             print(server_response)
26             if server_response == settings.LOGIN_STATE["file_exit"] or27                             server_response == settings.LOGIN_STATE["file_no_exit"]:
28                 f = open(filename, "rb")
29                 for line in f:  # 上传文件一行一行
30                     self.client.send(line)
31                     send_size = f.tell()   #获取当前指针位置（字节）
32                     progress_bar.progress_bar(self, send_size, file_size)
33                 else:
34                     print("file upload success")
35                     f.close()
36             #如果磁盘空间不足
37             elif server_response == settings.LOGIN_STATE["size_empty"]:
38                 print("server_response:磁盘空间不足")
39 
40         else:
41             print(filename, "is not exist")

 1 import json
 2 from conf import settings
 3 
 4 def client_pwd(self, *args):
 5     """用来查看用户当前的目录"""
 6     cmd_split = args[0].split()
 7     if len(cmd_split) == 1 and cmd_split[0] == "pwd":
 8         msg_dic = {
 9             "action":"pwd",
10         }
11         self.client.send(json.dumps(msg_dic).encode())
12         server_response = json.loads(self.client.recv(1024).decode())
13         print(server_response)
14         print(server_response["current_path"])
15     else:
16         print("%s:命令错误" % settings.LOGIN_STATE["cmd_error"])

 1 import os,sys
 2 
 3 dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 4 sys.path.insert(0, dir)
 5 print(sys.path)
 6 
 7 from core import main
 8 
 9 if __name__ == "__main__":
10     main.run()

 1 """
 2 服务器配置
 3 """
 4 import os
 5 
 6 PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 7 HOME_PATH = os.path.join(PATH, "home")
 8 
 9 #帐号名与密码
10 USER_DATA = {
11     "zcl":"abc",
12     "Alex":"123"
13 }
14 
15 LOGIN_STATE = {
16     "auth_True":"200",   #认证成功
17     "auth_False":"400",  #认证失败
18     "file_exit":"202",   #文件存在
19     "file_no_exit":"402", #文件不存在
20     "cmd_right":"201",  #命令正确
21     "cmd_error":"401",  #命令错误
22     "dir_exit":"203",   #目录已存在
23     "dir_no_exit":"403", #目录不存在
24     "cmd_success":"204",  #命令成功执行
25     "cmd_fail":"404",      #命令执行失败
26     "size_enough":"205", #磁盘空间足够
27     "size_empty":"405",  #磁盘空间不足
28 }
29 
30 
31 #磁盘配额:每个用户默认最多100M
32 MAX_SIZE = 102400000

 1 """
 2 用户加密认证，登陆模块
 3 """
 4 import os,sys,hashlib
 5 import json
 6 from conf import settings
 7 from core import db_handle
 8 
 9 
10 def auth_login(self):
11     """用户登陆时调用"""
12     recv_data = self.request.recv(1024).strip()
13     recv_data = recv_data.decode()
14     print(recv_data,type(recv_data))
15     recv_list = recv_data.split(":")
16     print(recv_list)
17     # 登陆后用户当前目录, 即用户的家目录
18     self.current_path = os.path.join(settings.HOME_PATH, recv_list[0])
19     # 用户宿主目录
20     self.user_home_path = os.path.join(settings.HOME_PATH, recv_list[0])
21 
22     user_path = "%s/data/%s.json" % (settings.PATH, recv_list[0])
23     print(user_path)
24     if os.path.isfile(user_path):
25         print("user(file) exist")
26         file_data = db_handle.user_load(user_path)
27         print(file_data)
28         if file_data["password"] == recv_list[1]:
29             print("login success")
30             #发送状态码给客户端
31             self.request.send(settings.LOGIN_STATE["auth_True"].encode())
32             print("send login_state")
33             #认证成功的状态码
34             return settings.LOGIN_STATE["auth_True"]
35         else:
36             #发送状态码给客户端
37             self.request.send(settings.LOGIN_STATE["auth_False"].encode())
38             print("send login_state")
39             #认证失败的状态码
40             return settings.LOGIN_STATE["auth_False"]
41     else:
42         # 发送状态码给客户端
43         self.request.send(settings.LOGIN_STATE["auth_False"].encode())
44         print("send login_state")
45         print("False,please registe")
46         return settings.LOGIN_STATE["auth_False"]
47 
48 
49 def create_user():
50     #服务端初始化时，先创建两个用户Alex,zcl
51     path = settings.PATH
52     for key in settings.USER_DATA:
53         print(settings.USER_DATA)
54         user_path = "%s/data/%s.json" % (path, key)
55         if not os.path.isfile(user_path):
56             password_hash = hash(settings.USER_DATA[key])
57             user_data = {
58                 "username":key,
59                 "password":password_hash,
60                 "user_path":os.path.join(settings.HOME_PATH, key),  #创建同时添加用户个人目录
61                 "max_size": settings.MAX_SIZE      #磁盘配额100M
62             }
63             #json.dump(user_data, open(user_path,"w",encoding="utf-8"))
64             db_handle.user_dump(user_path, user_data)
65     user_mkdir()
66 
67 
68 def user_mkdir():
69     """创建用户个人目录，在home目录下"""
70     for key in settings.USER_DATA:
71         user_home_path = os.path.join(settings.HOME_PATH, key)
72         if not os.path.isdir(user_home_path):
73             os.popen("mkdir %s" % user_home_path)
74 
75 
76 def hash(data):
77     m = hashlib.md5()
78     m.update(data.encode())
79     #返回加密后的数据
80     return m.hexdigest()

 1 """
 2 读用户数据与写用户数据
 3 """
 4 import json
 5 from conf import settings
 6 
 7 
 8 def user_load(user_path):
 9     """
10     读用户数据
11     :param user_path:读出的路径
12     :return: 用户数据字典
13     """
14     user_data = json.load(open(user_path, "r", encoding="utf-8"))
15     return user_data
16 
17 
18 def user_dump(user_path, user_data):
19     """
20     将数据写到用户数据库
21     :param user_path:写入的路径
22     :param user_data: 要写入的数据
23     :return:
24     """
25     json.dump(user_data, open(user_path, "w", encoding="utf-8"))

 1 import json,os
 2 from conf import settings
 3 
 4 def server_deal_cd(self, *args):
 5     """处理用户切换目录的功能"""
 6     cmd_dic = args[0]
 7     cd_dir = cmd_dic["cd_dir"]
 8     dir_path = self.current_path + r"%s" % cd_dir
 9     if cd_dir == ".." and len(self.current_path) > len(self.user_home_path):
10         #返回上一级目录
11         self.request.send(json.dumps(settings.LOGIN_STATE["cmd_success"]).encode())
12         self.current_path = os.path.dirname(self.current_path)
13     elif os.path.isdir(dir_path):   #切换目录
14         if cd_dir != "." and cd_dir != "..":
15             self.request.send(json.dumps(settings.LOGIN_STATE["cmd_success"]).encode())
16             self.current_path = self.current_path + r"%s" % cd_dir
17             print(self.current_path)
18         else:
19             self.request.send(json.dumps(settings.LOGIN_STATE["cmd_fail"]).encode())
20     else:   #切换的目录不存在
21         self.request.send(json.dumps(settings.LOGIN_STATE["dir_no_exit"]).encode())

 1 import os,json
 2 from conf import settings
 3 
 4 def server_deal_get(self, *args):
 5     """处理客户端下载文件的请求"""
 6     cmd_dic = args[0]
 7     filename = cmd_dic["filename"]
 8     if os.path.isfile(filename):
 9         file_size = os.stat(filename).st_size #服务端文件大小
10         msg_dic = {
11             "file_size":file_size,  #服务端将发给客户端的文件的大小
12             "file_exit":settings.LOGIN_STATE["file_exit"]
13         }
14         self.request.send(json.dumps(msg_dic).encode())
15         #防止粘包，服务端与客户端再进行一次交互
16         client_response = self.request.recv(1024)
17         print(client_response.decode())
18         f = open(filename, "rb")
19         for line in f:
20             self.request.send(line)
21         else:
22             print("server:file upload to client success")
23     else:
24         msg_dic = {"file_exit":settings.LOGIN_STATE["file_no_exit"]}
25         self.request.send(json.dumps(msg_dic).encode())

1 import os,json
2 
3 def server_deal_ls(self, *args):
4     """完成用户显示当前目录下文件(包括目录)的请求"""
5     cmd_dic = args[0]
6     r = os.popen("dir %s" % self.current_path)
7     dir_message = r.read()
8     self.request.send(dir_message.encode())

 1 import json,os
 2 from conf import settings
 3 
 4 
 5 def server_deal_mkdir(self, *args):
 6     """处理用户在当前目录(家目录下)创建目录的请求"""
 7     cmd_dir = args[0]
 8     new_dir = cmd_dir["new_dir"]  #当前目录在将创建的目录
 9     new_dir_path = os.path.join(self.current_path, new_dir)
10     print(new_dir_path)
11     if not os.path.isdir(new_dir_path):
12         #不存在目录，则创建
13         print("new_dir no exit")
14         os.popen("mkdir %s" % new_dir_path)
15         msg_dic = {
16             "cmd_state":settings.LOGIN_STATE["cmd_success"],
17         }
18         self.request.send(json.dumps(msg_dic).encode())
19     else:
20         msg = "%s:目录已存在，请先删除再创建" % settings.LOGIN_STATE["dir_exit"]
21         self.request.send(json.dumps(msg).encode())

 1 import os,json
 2 from conf import settings
 3 from core import get_dirsize
 4 
 5 
 6 def server_deal_put(self, *args):
 7     """处理客户端上传文件的请求"""
 8     cmd_dic = args[0]       #字典格式
 9     filename = cmd_dic["filename"]
10     file_size = cmd_dic["file_size"]
11     file_path = os.path.join(self.current_path, filename)
12     print("AA")
13     dir_size = get_dirsize.get_dirsize(self.user_home_path)
14     print("BB")
15     print("当前用户磁盘空间大小:%s" % dir_size)
16     #如果用户家目录下的大小加上本次将上传文件的大小仍小于最大的磁盘配额，则可以继续上传
17     if dir_size+file_size < settings.MAX_SIZE:
18 
19         if os.path.isfile(file_path):     # 如果文件已经存在
20             f = open(file_path + ".new", "wb")
21             # 交互，防止粘包
22             self.request.send(json.dumps(settings.LOGIN_STATE["file_exit"]).encode())
23         else:  # 如果不存在，就上传
24             f = open(file_path, "wb")
25             self.request.send(json.dumps(settings.LOGIN_STATE["file_no_exit"]).encode())
26 
27         #self.request.send(b"200, OK")  # 可优化成字典json，状态码
28         # 开始接收数据
29         received_size = 0
30         while received_size < file_size:
31             data = self.request.recv(1024)
32             received_size += len(data)
33             f.write(data)
34         else:  # 文件上传完成
35             print("file [%s] has uploaded..." % filename)
36 
37     else:
38         #磁盘配额不足
39         self.request.send(json.dumps(settings.LOGIN_STATE["size_empty"]).encode())

 1 import json
 2 from conf import settings
 3 
 4 
 5 def server_deal_pwd(self, *args):
 6     """ 用来处理客户端查看当前目录下的请求"""
 7     cmd_dic = args[0]  # 字典格式
 8     msg_dic = {
 9         "current_path":self.current_path,   #发送当前目录
10         "cmd_state":settings.LOGIN_STATE["cmd_right"] #发送命令状态
11     }
12     self.request.send(json.dumps(msg_dic).encode())

 1 import os
 2 
 3 
 4 def get_dirsize(dir):
 5     """
 6     获取目录的大小
 7     :param dir: 目录的路径
 8     :return: 大小(字节)
 9     """
10     size = 0
11     for root, dirs, files in os.walk(dir):
12         size += sum([os.path.getsize(os.path.join(root, name)) for name in files])
13     return size

 1 import socketserver
 2 import json,os
 3 from core import deal_put
 4 from core import deal_get
 5 from core import deal_pwd
 6 from core import deal_mkdir
 7 from core import deal_cd
 8 from core import deal_ls
 9 from core import auth
10 from conf import settings
11 
12 
13 class MyTCPHandler(socketserver.BaseRequestHandler):  #继承父类
14     def put(self, *args):
15         deal_put.server_deal_put(self, *args)
16 
17     def get(self, *args):
18         deal_get.server_deal_get(self, *args)
19 
20     def pwd(self, *args):
21         deal_pwd.server_deal_pwd(self, *args)
22 
23     def mkdir(self, *args):
24         deal_mkdir.server_deal_mkdir(self, *args)
25 
26     def ls(self, *args):
27         deal_ls.server_deal_ls(self, *args)
28 
29     def cd(self, *args):
30         deal_cd.server_deal_cd(self, *args)
31 
32     def handle(self):
33         while True:
34             #用户认证
35             login_state = auth.auth_login(self)
36             #如果认证成功
37             if login_state == settings.LOGIN_STATE["auth_True"]:
38                 while True:
39                     try:
40                         self.data = self.request.recv(1024).strip()
41 
42                         print("{} wrote:".format(self.client_address[0]))
43                         print(self.data)
44                         cmd_dic = json.loads(self.data.decode())   #字典格式
45                         action = cmd_dic["action"]      #获取客户端指令
46 
47                         if hasattr(self, action):   #反射
48                             func = getattr(self, action)
49                             func(cmd_dic)   #参数为字典格式
50                         else:
51                             print("服务端反射调用失败")
52 
53                     except ConnectionResetError as e:
54                         print("客户端断开",e)
55                         break
56             elif login_state == settings.LOGIN_STATE["auth_False"]:
57                 continue
58 
59 
60 def run():
61     #程序初始化时创建用户
62     auth.create_user()
63     #用户登陆时身份验证
64     HOST, PORT = "localhost", 8787
65     server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)  #实例化
66     server.serve_forever()

C:\Python34\python3.exe C:/Users/Administrator/PycharmProjects/laonanhai/ftp/ftp_client/bin/ftp_client.py
Username:zcl
Password:abc
******************Welcome Login*******************
>>>:get video.mp4
{‘file_exit‘: ‘202‘, ‘file_size‘: 31491127} <class ‘dict‘>
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||100%download from server success
>>>:cd \aa\ff
"204"
>>>:pwd
{‘current_path‘: ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server\\home\\zcl\\aa\\ff‘, ‘cmd_state‘: ‘201‘}
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\zcl\aa\ff
>>>:put jita.mp4
402
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||100%file upload success
>>>:get video.mp4
{‘file_exit‘: ‘202‘, ‘file_size‘: 31491127} <class ‘dict‘>
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||100%download from server success
>>>:pwd
{‘current_path‘: ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server\\home\\zcl\\aa\\ff‘, ‘cmd_state‘: ‘201‘}
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\zcl\aa\ff
>>>:cd ..
"204"
>>>:pwd
{‘current_path‘: ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server\\home\\zcl\\aa‘, ‘cmd_state‘: ‘201‘}
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\zcl\aa
>>>:ls
 驱动器 C 中的卷没有标签。
 卷的序列号是 000C-3580

 C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\zcl\aa 的目录

2017/01/23 周一  下午 12:48    <DIR>          .
2017/01/23 周一  下午 12:48    <DIR>          ..
2017/01/22 周日  下午 01:24    <DIR>          ee
2017/01/23 周一  下午 01:02    <DIR>          ff
2017/01/23 周一  下午 12:40    <DIR>          gg
               0 个文件              0 字节
               5 个目录 34,504,105,984 可用字节

>>>:mkdir hh
{‘cmd_state‘: ‘204‘}
>>>:ls
 驱动器 C 中的卷没有标签。
 卷的序列号是 000C-3580

 C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\zcl\aa 的目录

2017/01/23 周一  下午 01:06    <DIR>          .
2017/01/23 周一  下午 01:06    <DIR>          ..
2017/01/22 周日  下午 01:24    <DIR>          ee
2017/01/23 周一  下午 01:02    <DIR>          ff
2017/01/23 周一  下午 12:40    <DIR>          gg
2017/01/23 周一  下午 01:06    <DIR>          hh
               0 个文件              0 字节
               6 个目录 34,503,962,624 可用字节

>>>:cd ..
"204"
>>>:owd

        ls
        pwd
        cd ../..
        get filename
        put filename
        
>>>:pwd
{‘current_path‘: ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server\\home\\zcl‘, ‘cmd_state‘: ‘201‘}
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\zcl
>>>:cd ..
"404"
>>>:

C:\Python34\python3.exe C:/Users/Administrator/PycharmProjects/laonanhai/ftp/ftp_client/bin/ftp_client.py
Username:Alex
Password:123
******************Welcome Login*******************
>>>:pwd
{‘cmd_state‘: ‘201‘, ‘current_path‘: ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server\\home\\Alex‘}
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\Alex
>>>:

C:\Python34\python3.exe C:/Users/Administrator/PycharmProjects/laonanhai/ftp/ftp_server/bin/ftp_server.py
[‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server‘, ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server\\bin‘, ‘C:\\Python34\\lib\\site-packages\\pip-8.1.2-py3.4.egg‘, ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai‘, ‘C:\\Windows\\SYSTEM32\\python34.zip‘, ‘C:\\Python34\\DLLs‘, ‘C:\\Python34\\lib‘, ‘C:\\Python34‘, ‘C:\\Python34\\lib\\site-packages‘]
{‘zcl‘: ‘abc‘, ‘Alex‘: ‘123‘}
{‘zcl‘: ‘abc‘, ‘Alex‘: ‘123‘}
zcl:900150983cd24fb0d6963f7d28e17f72 <class ‘str‘>
[‘zcl‘, ‘900150983cd24fb0d6963f7d28e17f72‘]
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server/data/zcl.json
user(file) exist
{‘max_size‘: 102400000, ‘user_path‘: ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server\\home\\zcl‘, ‘password‘: ‘900150983cd24fb0d6963f7d28e17f72‘, ‘username‘: ‘zcl‘}
login success
send login_state
127.0.0.1 wrote:
b‘{"filename": "video.mp4", "overridden": true, "action": "get"}‘
客户端已准备好下载
server:file upload to client success
127.0.0.1 wrote:
b‘{"cd_dir": "\\\\aa\\\\ff", "action": "cd"}‘
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\zcl\aa\ff
127.0.0.1 wrote:
b‘{"action": "pwd"}‘
127.0.0.1 wrote:
b‘{"filename": "jita.mp4", "file_size": 8386449, "overridden": true, "action": "put"}‘
AA
BB
当前用户磁盘空间大小:25159347
file [jita.mp4] has uploaded...
127.0.0.1 wrote:
b‘{"filename": "video.mp4", "overridden": true, "action": "get"}‘
客户端已准备好下载
server:file upload to client success
127.0.0.1 wrote:
b‘{"action": "pwd"}‘
127.0.0.1 wrote:
b‘{"cd_dir": "..", "action": "cd"}‘
127.0.0.1 wrote:
b‘{"action": "pwd"}‘
127.0.0.1 wrote:
b‘{"action": "ls"}‘
127.0.0.1 wrote:
b‘{"new_dir": "hh", "overriden": false, "action": "mkdir"}‘
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server\home\zcl\aa\hh
new_dir no exit
127.0.0.1 wrote:
b‘{"action": "ls"}‘
127.0.0.1 wrote:
b‘{"cd_dir": "..", "action": "cd"}‘
127.0.0.1 wrote:
b‘{"action": "pwd"}‘
127.0.0.1 wrote:
b‘{"cd_dir": "..", "action": "cd"}‘
Alex:202cb962ac59075b964b07152d234b70 <class ‘str‘>
[‘Alex‘, ‘202cb962ac59075b964b07152d234b70‘]
C:\Users\Administrator\PycharmProjects\laonanhai\ftp\ftp_server/data/Alex.json
user(file) exist
{‘max_size‘: 102400000, ‘user_path‘: ‘C:\\Users\\Administrator\\PycharmProjects\\laonanhai\\ftp\\ftp_server\\home\\Alex‘, ‘password‘: ‘202cb962ac59075b964b07152d234b70‘, ‘username‘: ‘Alex‘}
login success
send login_state
127.0.0.1 wrote:
b‘{"action": "pwd"}‘