day 7作业 简单的FTP

第七天作业：简单FTP程序
作者：keven

使用说明：运行f_s,运行f_c ，输入用户名，密码，例如zhangsan,  zhangsang

作业要求：
1. 用户登陆
2. 上传/下载文件
3. 不同用户家目录不同
4. 查看当前目录下文件
5. 充分使用面向对象知识

目录：

ftp_server  ，如下：

import os,socketserver,json,sys
basepath=os.path.dirname(os.path.dirname(__file__))
# print(basepath)
sys.path.append(basepath)
# path=os.path.join(basepath,'db','zhangsan')
# print(path)
# print(os.path.isdir(path))

class MyTCPHandler(socketserver.BaseRequestHandler):
    '''
    FTP server class
    '''

    def authenticate(self,*args):
        #print(type(*args))
        #print(*args)
        cmd_dic=args[0]
        #print(args[0])
        user=cmd_dic["username"]
        passwd=cmd_dic["password"]
        #print(user,passwd)
        with open('userinfo',"r") as f:
            for line in f:
                line=line.split(",")
                if user == line[1] and passwd == line[2]:
                    path=os.path.join(basepath,'db',user)
                    ack_dic={
                        'authenticated':'Y',
                        'path':path
                    }
                    #print(ack_dic)
                    self.request.send(json.dumps(ack_dic).encode())
                    break
            else:
                ack_dic={
                        'authenticated':'N'
                    }
                #print(ack_dic)
                self.request.send(json.dumps(ack_dic).encode())



    def put(self,*args):
        '''接受客户端文件'''
        cmd_dic = args[0]
        path = cmd_dic["path"]
        filename=os.path.join(path,cmd_dic["filename"])
        filesize=cmd_dic["size"]
        if os.path.isfile(filename):
            f = open(filename,+"new","wb")
        else:
            f =open(filename,"wb")

        self.request.send(b"200 ok")
        received_size= 0
        while received_size < filesize:
            data = self.request.recv(1024)
            f.write(data)
            received_size += len(data)
        else:
            print("file {} has upload".format(filename))

    def ls(self,*args):
        '''查看用户目录文件'''
        cmd_dic=args[0]
        path=cmd_dic["path"]
        res=os.listdir(path)
        self.request.send(json.dumps(res).encode())

    def get(self,*args):
        '''客户端下载文件'''
        cmd_dic=args[0]
        filename=os.path.join(cmd_dic["path"],cmd_dic["filename"])
        if os.path.isfile(filename):
            size=os.stat(filename).st_size
            ack_dic={
                "file_status":"Y",
                "size":size
            }
            self.request.send(json.dumps(ack_dic).encode())

            with open(filename,"rb") as f:
                for line in f:
                    self.request.send(line)
                else:
                    print("File download complete ")
        else:
            ack_dic={
                "file_status":"N"
            }
            self.request.send(json.dumps(ack_dic).encode())




    def handle(self):
        while True:
            try:
                self.data = self.request.recv(1024).strip()
                #print("{} wrote:".format(self.client_addrss[0]))
                print(self.data)
                cmd_dic = json.loads(self.data.decode())
                action = cmd_dic["action"]
                if hasattr(self,action):
                    func =getattr(self,action)
                    func(cmd_dic)
            except ConnectionRefusedError as e:
                print("error",e)
                break


if __name__ == '__main__':
    HOST,PORT ='localhost',9001
    server=socketserver.ThreadingTCPServer((HOST,PORT),MyTCPHandler)
    server.serve_forever()

ftp_client

import os,socket,json



class FtpClient(object):
    '''
    client class
    '''
    def __init__(self):
        self.client =socket.socket()


    def help(self):
        msg='''
        ls       
        put filename
        get filename
        '''
        print(msg)

    def connect(self,host,port):
        self.client.connect((host,port))

    def authenticate(self):
        '''验证用户名，密码是否正确，正确返回验证状态及用户目录'''
        user=input("pls input your username:")
        passwd=input("pls input your password:")
        msg_dic={"action":"authenticate",
                'username':user,
                 "password":passwd}
        self.client.send(json.dumps(msg_dic).encode())
        return self.client.recv(1024).decode()


    def interactive(self):
        '''判断用户输入命令，根据命令分发操作'''
        auth_res=self.authenticate()
        auth_res=eval(auth_res)
        #print(type(auth_res))
        if auth_res['authenticated'] == "Y":
            path=auth_res["path"]
            while True:
                cmd=input(">>>:").strip()
                if len(cmd) == 0:continue
                cmd_str = cmd.split()
                #print(cmd_str)
                #print(cmd_str[0])
                if hasattr(self,cmd_str[0]):
                    func=getattr(self,cmd_str[0])
                    func(cmd,path)
                else:
                    self.help()
        elif auth_res['authenticated'] == "N":
            print("用户名，密码错误")

    def ls(self,*args):
        '''显示用户下的文件'''
        #print("1",args[0],args[1],type(args[0]))
        cmd=args[0].split()
        path=args[1]

        msg_dic={
            "action":'ls',
            "path":path
        }
        #print(msg_dic)
        self.client.send(json.dumps(msg_dic).encode())
        ack_dic=json.loads(self.client.recv(1024).decode())
        print(ack_dic)


    def get(self,*args):
        '''下载文件'''
        cmd_split =  args[0].split()
        path=args[1]
        #print(args[1])
        if len(cmd_split) > 1:
            filename=cmd_split[1]
            #print(type(filename))
            msg_dic={
                "action":"get",
                "filename":filename,
                "path":path
                }
            #print(msg_dic)
            self.client.send(json.dumps(msg_dic).encode())
            ack_dic=json.loads(self.client.recv(1024).decode())
            #print(ack_dic)
            file_status=ack_dic.get("file_status")
            if file_status == "Y":
                file_size=ack_dic.get("size")
                if os.path.isfile(filename):
                    #print(type(filename))
                    f = open(filename +"new", "wb")
                else:
                    f = open(filename, "wb")
                received_size = 0
                while received_size < file_size:
                    data = self.client.recv(1024)
                    f.write(data)
                    received_size += len(data)
                else:
                    print("file {} has download".format(filename))

            elif file_status == "N":
                print("下载的文件不存在")
        else:
            print("参数不正确，请查看帮助信息")

    def put(self,*args):
        '''上传文件'''
        cmd_split = args [0].split()
        print(args[1])
        path=args[1]
        if len(cmd_split) > 1:
            filename = cmd_split[1]
            if os.path.isfile(filename):
                filesize = os.stat(filename).st_size
                msg_dic = {
                    "action":"put",
                    "filename":filename,
                    "size":filesize,
                    'overridden':True,
                    'path':path
                }
                self.client.send(json.dumps(msg_dic).encode())
                #防止粘包，等待服务器确认
                server_response = self.client.recv(1024)
                f=open(filename,"rb")
                for line in f:
                    self.client.send(line)
                else:
                    print("file upload success...")
                    f.close()
            else:
                print(filename,"is not exitst")
        else:
            print("参数不正确，请查看帮助信息")