day8--socketserver作业

fileno()文件描述符

handle_request()处理单个请求

server_forever(poll_interval=0.5)处理多个请求,poll_interval每0.5秒检测是否关闭,

作业:开发一个支持多用户在线的FTP程序

要求:

    1.用户加密认证;

    2.允许同时多用户登录;

    3.每个用户有自己的家目录,且只能访问自己的家目录;

    4.对用户进行磁盘配额,每个用户的可用空间不同;

    5.允许用户在ftp.server上随意切换目录;

    6.允许用户查看当前目录下文件;

    7.允许上传和下载文件,保证文件一致性;

    8.文件传输过程中显示进度条;

    9.附加功能,支持文件的断点续传。

发送文件的大致思路:

 

接收文件的大致思路:

遍历空的文件,如果一个文件是空的,Python是不会执行后面缩进的代码的,空的文件。

http://www.cnblogs.com/liujiacai/p/7417953.html

http://www.cnblogs.com/lianzhilei/p/5869205.html

http://www.bubuko.com/infodetail-1758633.html

    由于不是做运维的,对一些系统的操作不是很熟悉,因此首先完成了文件的上传与下载:下面来看看这两个功能:

   

    客户端:

 

import socket,os,json,sys


class Myclient(object):
    '''定义客户端'''
    def __init__(self):
        '''定义socket()实例'''
        self.client = socket.socket()                                         #生成soket()实例

    def connect(self,ip,port):
        """定义连接"""
        self.client.connect((ip,port))

    def interactive(self):
        '''定义和用户交互'''
        while True:
            cmd = input("请输入指令>>:").strip()                              #用户输入指令
            if len(cmd) == 0:
                print("指令不能为空!!!")
                continue                                                      #指令不能为空
            cmd_str = cmd.split()[0]                                          #指令由两种形式
            if hasattr(self,cmd_str):                                         #判断指令是否存在
                func = getattr(self,cmd_str)                                  #指令存在,则获取对应的方法
                func(cmd)
            elif cmd_str == 'q':
                break
            else:
                self.help()
                continue


    def help(self):
        msg = """
        ls
        pwd
        cd
        put   filename
        get   filename
        """
        print("指令名称",msg)

    def put(self,cmd):
        '''定义上传数据接口'''
        cmd_list = cmd.split()
        if len(cmd_list) > 1:
            '''指令是由指令名和文件名构成'''
            filename = cmd_list[1]                                              #获取文件名
            if os.path.isfile(filename):                                        #检测上传的文件名是否存在
                '''文件名存在,则要告诉服务器端上传的文件名,文件大小,并检测服务器是否有相应的方法'''
                filesize = os.stat(filename).st_size
                msg_dic = {
                    "action":"put",
                    "filename":filename,
                    "filesize":filesize
                }
                '''转换为字节码的形式进行传输'''
                self.client.send(json.dumps(msg_dic).encode("utf-8"))           #转换为json格式进行上传,便于服务器接收后处理
                server_response = self.client.recv(1024)                        #等待服务器传回信息,防止粘包
                self.client.send("收到了".encode("utf-8"))
                if int(server_response.decode("utf-8")):
                    '''根据客户端反馈,服务器端是否存在相同的文件名'''
                    file_exist = int(self.client.recv(1024).decode("utf-8"))
                    if file_exist:
                        '''服务器端存在相应的文件'''
                        while True:
                            cover_flag = input("文件已存在,是否覆盖原文件(y/n):")
                            if cover_flag == "y":
                                self.client.send("y".encode("utf-8"))
                                break
                            elif cover_flag == "n":
                                new_filename = input("请输入新的文件名>>:")
                                self.client.send(("n "+new_filename).encode("utf-8"))
                                break
                            else:
                                print("您输入的指令有误,请重新输入!!!")
                                continue
                    else:
                        self.client.send("新创建".encode("utf-8"))

                    '''上面准备工作完毕,开始传输文件'''
                    with open(filename,'rb') as f:
                        for line in f:
                            self.client.send(line)                                  #发送数据
                    '''客户端等待服务器端传来消息,告知上传成功'''
                    response = self.client.recv(1024)
                    print(response.decode("utf-8"))                                 #数据是否上传成功,告知用户
                else:
                    print("服务器端不存在相应的指令!!!")

            else:
                print("要上传的文件名不存在!")


        else:
            print("没有输入要上传的文件名!!!")

    def get(self,cmd):
        '''从服务器端下载数据'''
        cmd_list = cmd.split()                                                    #获取用户输入指令
        if len(cmd_list) > 1:                                                     #是由指令和文件名构成
            filename = cmd_list[1]
            msg_dic = {
                "filename":filename,
                "action":"get"
            }                                                                     #把指令封装成字典发送给服务器
            self.client.send(json.dumps(msg_dic).encode("utf-8"))                 #把指令以json形式传给服务器
            server_response = int(self.client.recv(1024).decode("utf-8"))         #接收服务端发挥的指令,判断是否支持此方法
            self.client.send("收到指令!".encode("utf-8"))
            if server_response:
                file_exists = int(self.client.recv(1024).decode("utf-8"))         #下载文件存在与否标识符
                self.client.send("收到标识符".encode("utf-8"))                     #告知服务器收到标识符,防止粘包
                if file_exists:
                    '''文件存在,则准备接收数据,会接收服务器发送过来的文件大小'''
                    file_mess = json.loads(self.client.recv(3072).decode("utf-8"))
                    self.client.send("文件信息接收成功,开始进行文件接收!".encode("utf-8"))
                    filesize = file_mess["filesize"]
                    while True:
                        '''确定文件名'''
                        if os.path.isfile(filename):
                            '''说明客户端存在文件,就要让用户选择是否覆盖'''
                            choice_cover = input("文件名重复,是否覆盖现有文件(y/n)>>:")
                            if choice_cover == "y":
                                filename = filename
                                break                                       #覆盖文件之后退出循环
                            elif choice_cover == "n":
                                filename = input("请输入新的文件名>>:")
                                '''由于用户输入了新的用户名,我们不知道此文件是否重名,因此重新检测一下,确保不重名'''
                                continue
                        else:
                            filename = filename
                            break                                           #得到文件名之后退出循环
                    recvive_size = 0
                    with open(filename,'wb') as file:
                        while recvive_size < filesize:
                            data = self.client.recv(1024)
                            file.write(data)
                            recvive_size += len(data)
                        print("数据下载完成!!!")
                        self.client.send("接收完毕".encode("utf-8"))
                else:
                    print("要下载的文件不存在!!!")

            else:
                print("服务器不支持此功能!!!")

        else:
            print("对不起,输入有误!")
            self.help()

if __name__ == "__main__":
    try:
        client = Myclient()
        client.connect("localhost",9998)
        client.interactive()
    except KeyboardInterrupt as e:
        print("客户端断开!!!",e)

 

    上面客户端只实现了上传和下载功能,没有实现其他系统操作有关的功能,后续将逐步完善;

    服务器端:

'''定义服务器端'''
import socketserver,json,os

class MyTcpServer(socketserver.BaseRequestHandler):
    '''定义服务器端,socketserver,继承的是BaseRequestHandler类'''
    def handle(self):
        '''服务器端所有的功能都在handle()里面进行处理,因此接收数据也是在handle()中,只是处理的时候,调用下面模块而已'''
        while True:
            data = self.request.recv(3072)
            if len(data) == 0:
                print("客户端断开")
                break
            msg_dic = json.loads(data.decode("utf-8"))            #服务器端接收指令
            cmd = msg_dic["action"]
            if hasattr(self,cmd):                                                    #判断服务器是否有相应的指令
                self.request.send("1".encode("utf-8"))
                self.request.recv(1024)
                func = getattr(self,cmd)
                func(msg_dic)
            else:
                self.request.send("0".encode("utf-8"))
                self.request.recv(1024)

    def put(self,msg_dic):
        '''定义上传的服务器代码'''
        filename = msg_dic['filename']
        filesize = msg_dic["filesize"]
        '''判断服务器端是否存在对应的文件名'''
        recvive_size = 0
        if os.path.isfile(filename):
            '''如果服务器存在相应的文件名,则询问客户端是否覆盖'''
            self.request.send("1".encode("utf-8"))                              #文件存在,循环客户端是否覆盖
            response = self.request.recv(1024).decode("utf-8")
            response = response.split()
            '''如果用户让覆盖则覆盖,否则重新创建'''

            if response[0] == "y":
                '''覆盖'''
                with open(filename,"wb") as f1:
                   while recvive_size < filesize:
                       data = self.request.recv(1024)
                       f1.write(data)
                       recvive_size += len(data)
                self.request.send("文件接收成功!".encode("utf-8"))

            else:
                '''不覆盖,新创建'''
                new_filename = response[1]
                with open(new_filename,'wb') as f:
                    while recvive_size < filesize:
                        data1 = self.request.recv(1024)
                        f.write(data1)
                        recvive_size += len(data1)
                self.request.send("文件接收成功!".encode("utf-8"))

        else:
            '''如果服务器不存在相应文件名,则创建'''
            self.request.send("0".encode("utf-8"))
            self.request.recv(1024)
            with open(filename,'wb') as f2:
                while recvive_size < filesize:
                    data1 = self.request.recv(1024)
                    f2.write(data1)
                    recvive_size += len(data1)
            self.request.send("文件接收成功!".encode("utf-8"))

    def get(self,msg_dict):
        '''基础交流结束,服务器支持文件下载功能,开始文件下载操作'''
        filename = msg_dict["filename"]
        if os.path.isfile(filename):
            self.request.send("1".encode("utf-8"))                             #文件存在告知客户端
            self.request.recv(1024)
            filesize = os.stat(filename).st_size
            '''首先告知客户端文件大小,让这边做好接收准备'''
            file_mess = {
                "filesize":filesize,
            }
            self.request.send(json.dumps(file_mess).encode("utf-8"))           #以json格式发送给客户端
            self.request.recv(1024)                                            #防止粘包,客户端回应服务器接收完毕
            with open(filename,"rb") as f_obj:
                for line in f_obj:
                    self.request.send(line)
            self.request.recv(1024)

        else:
            self.request.send("0".encode("utf-8"))                             #文件不存在,告知客户端
            self.request.recv(1024)

if __name__ == "__main__":
    try:
        IP,PORT = "localhost",9998
        server = socketserver.ThreadingTCPServer((IP,PORT),MyTcpServer)              #定义socketserver实例
        server.serve_forever()
    except KeyboardInterrupt as e:
        print("关闭服务器")

    上面代码是服务器端,客户端与服务器端其实是相呼应的,两者实现数据的交换;其实就是收发数据,中间掺杂我们想要实现的功能。

    下面来看几个容易犯错的地方:

    1、BrokenPipeError: [Errno 32] Broken pipe

 

  客户端代码:

 

import socket,os,json,sys


class Myclient(object):
    '''定义客户端'''
    def __init__(self):
        '''定义socket()实例'''
        self.client = socket.socket()                                         #生成soket()实例

    def connect(self,ip,port):
        """定义连接"""
        self.client.connect((ip,port))

    def interactive(self):
        '''定义和用户交互'''
        while True:
            cmd = input("请输入指令>>:").strip()                              #用户输入指令
            if len(cmd) == 0:
                print("指令不能为空!!!")
                continue                                                      #指令不能为空
            cmd_str = cmd.split()[0]                                          #指令由两种形式
            if hasattr(self,cmd_str):                                         #判断指令是否存在
                func = getattr(self,cmd_str)                                  #指令存在,则获取对应的方法
                func(cmd)
            elif cmd_str == 'q':
                break
            else:
                self.help()
                continue


    def help(self):
        msg = """
        ls
        pwd
        cd
        put   filename
        get   filename
        """
        print("指令名称",msg)

    def put(self,cmd):
        '''定义上传数据接口'''
        cmd_list = cmd.split()
        if len(cmd_list) > 1:
            '''指令是由指令名和文件名构成'''
            filename = cmd_list[1]                                              #获取文件名
            if os.path.isfile(filename):                                        #检测上传的文件名是否存在
                '''文件名存在,则要告诉服务器端上传的文件名,文件大小,并检测服务器是否有相应的方法'''
                filesize = os.stat(filename).st_size
                msg_dic = {
                    "action":"put",
                    "filename":filename,
                    "filesize":filesize
                }
                '''转换为字节码的形式进行传输'''
                self.client.send(json.dumps(msg_dic).encode("utf-8"))           #转换为json格式进行上传,便于服务器接收后处理
                server_response = self.client.recv(1024)                        #等待服务器传回信息,防止粘包
                self.client.send("收到了".encode("utf-8"))
                if int(server_response.decode("utf-8")):
                    '''根据客户端反馈,服务器端是否存在相同的文件名'''
                    file_exist = int(self.client.recv(1024).decode("utf-8"))
                    if file_exist:
                        '''服务器端存在相应的文件'''
                        while True:
                            cover_flag = input("文件已存在,是否覆盖原文件(y/n):")
                            if cover_flag == "y":
                                self.client.send("y".encode("utf-8"))
                                break
                            elif cover_flag == "n":
                                new_filename = input("请输入新的文件名>>:")
                                self.client.send(("n "+new_filename).encode("utf-8"))
                                break
                            else:
                                print("您输入的指令有误,请重新输入!!!")
                                continue
                    else:
                        self.client.send("新创建".encode("utf-8"))

                    '''上面准备工作完毕,开始传输文件'''
                    with open(filename,'rb') as f:
                        for line in f:
                            self.client.send(line)                                  #发送数据
                    '''客户端等待服务器端传来消息,告知上传成功'''
                    response = self.client.recv(1024)
                    print(response.decode("utf-8"))                                 #数据是否上传成功,告知用户
                else:
                    print("服务器端不存在相应的指令!!!")

            else:
                print("要上传的文件名不存在!")


        else:
            print("没有输入要上传的文件名!!!")

    def get(self,cmd):
        '''从服务器端下载数据'''
        cmd_list = cmd.split()                                                    #获取用户输入指令
        if len(cmd_list) > 1:                                                     #是由指令和文件名构成
            filename = cmd_list[1]
            msg_dic = {
                "filename":filename,
                "action":"get"
            }                                                                     #把指令封装成字典发送给服务器
            self.client.send(json.dumps(msg_dic).encode("utf-8"))                 #把指令以json形式传给服务器
            server_response = int(self.client.recv(1024).decode("utf-8"))         #接收服务端发挥的指令,判断是否支持此方法
            self.client.send("收到指令!".encode("utf-8"))
            if server_response:
                file_exists = int(self.client.recv(1024).decode("utf-8"))         #下载文件存在与否标识符
                self.client.send("收到标识符".encode("utf-8"))                     #告知服务器收到标识符,防止粘包
                if file_exists:
                    '''文件存在,则准备接收数据,会接收服务器发送过来的文件大小'''
                    file_mess = json.loads(self.client.recv(3072).decode("utf-8"))
                    self.client.send("文件信息接收成功,开始进行文件接收!".encode("utf-8"))
                    filesize = file_mess["filesize"]
                    while True:
                        '''确定文件名'''
                        if os.path.isfile(filename):
                            '''说明客户端存在文件,就要让用户选择是否覆盖'''
                            choice_cover = input("文件名重复,是否覆盖现有文件(y/n)>>:")
                            if choice_cover == "y":
                                filename = filename
                                break                                       #覆盖文件之后退出循环
                            elif choice_cover == "n":
                                filename = input("请输入新的文件名>>:")
                                '''由于用户输入了新的用户名,我们不知道此文件是否重名,因此重新检测一下,确保不重名'''
                                continue
                        else:
                            filename = filename
                            break                                           #得到文件名之后退出循环
                    recvive_size = 0
                    with open(filename,'wb') as file:
                        while recvive_size < filesize:
                            data = self.client.recv(1024)
                            file.write(data)
                            recvive_size += len(data)
                        print("数据下载完成!!!")
                        self.client.send("接收完毕".encode("utf-8"))
                else:
                    print("要下载的文件不存在!!!")

            else:
                print("服务器不支持此功能!!!")

        else:
            print("对不起,输入有误!")
            self.help()

if __name__ == "__main__":
    try:
        client = Myclient()
        client.connect("localhost",9998)
        client.interactive()
    except KeyboardInterrupt as e:
        print("客户端断开!!!",e)

 

    服务器端:

'''定义服务器端'''
import socketserver,json,os

class MyTcpServer(socketserver.BaseRequestHandler):
    '''定义服务器端,socketserver,继承的是BaseRequestHandler类'''
    def handle(self):
        '''服务器端所有的功能都在handle()里面进行处理,因此接收数据也是在handle()中,只是处理的时候,调用下面模块而已'''
        # while True:
        data = self.request.recv(3072)
        if len(data) == 0:
            print("客户端断开")
                # break
        msg_dic = json.loads(data.decode("utf-8"))            #服务器端接收指令
        cmd = msg_dic["action"]
        if hasattr(self,cmd):                                                    #判断服务器是否有相应的指令
            self.request.send("1".encode("utf-8"))
            self.request.recv(1024)
            func = getattr(self,cmd)
            func(msg_dic)
        else:
            self.request.send("0".encode("utf-8"))
            self.request.recv(1024)

    def put(self,msg_dic):
        '''定义上传的服务器代码'''
        filename = msg_dic['filename']
        filesize = msg_dic["filesize"]
        '''判断服务器端是否存在对应的文件名'''
        recvive_size = 0
        if os.path.isfile(filename):
            '''如果服务器存在相应的文件名,则询问客户端是否覆盖'''
            self.request.send("1".encode("utf-8"))                              #文件存在,循环客户端是否覆盖
            response = self.request.recv(1024).decode("utf-8")
            response = response.split()
            '''如果用户让覆盖则覆盖,否则重新创建'''

            if response[0] == "y":
                '''覆盖'''
                with open(filename,"wb") as f1:
                   while recvive_size < filesize:
                       data = self.request.recv(1024)
                       f1.write(data)
                       recvive_size += len(data)
                self.request.send("文件接收成功!".encode("utf-8"))

            else:
                '''不覆盖,新创建'''
                new_filename = response[1]
                with open(new_filename,'wb') as f:
                    while recvive_size < filesize:
                        data1 = self.request.recv(1024)
                        f.write(data1)
                        recvive_size += len(data1)
                self.request.send("文件接收成功!".encode("utf-8"))

        else:
            '''如果服务器不存在相应文件名,则创建'''
            self.request.send("0".encode("utf-8"))
            self.request.recv(1024)
            with open(filename,'wb') as f2:
                while recvive_size < filesize:
                    data1 = self.request.recv(1024)
                    f2.write(data1)
                    recvive_size += len(data1)
            self.request.send("文件接收成功!".encode("utf-8"))

    def get(self,msg_dict):
        '''基础交流结束,服务器支持文件下载功能,开始文件下载操作'''
        filename = msg_dict["filename"]
        if os.path.isfile(filename):
            self.request.send("1".encode("utf-8"))                             #文件存在告知客户端
            self.request.recv(1024)
            filesize = os.stat(filename).st_size
            '''首先告知客户端文件大小,让这边做好接收准备'''
            file_mess = {
                "filesize":filesize,
            }
            self.request.send(json.dumps(file_mess).encode("utf-8"))           #以json格式发送给客户端
            self.request.recv(1024)                                            #防止粘包,客户端回应服务器接收完毕
            with open(filename,"rb") as f_obj:
                for line in f_obj:
                    self.request.send(line)
            self.request.recv(1024)

        else:
            self.request.send("0".encode("utf-8"))                             #文件不存在,告知客户端
            self.request.recv(1024)

if __name__ == "__main__":
    try:
        IP,PORT = "localhost",9998
        server = socketserver.ThreadingTCPServer((IP,PORT),MyTcpServer)              #定义socketserver实例
        server.serve_forever()
    except KeyboardInterrupt as e:
        print("关闭服务器")

    上面代码是我遇到的坑,客户端没有变,只变了服务器端接收的代码,因为socketserver.serve_forever()持续的响应链接。结果客户端输入命令如下:

请输入指令>>:gcx
指令名称 
        ls
        pwd
        cd
        put   filename
        get   filename
        
请输入指令>>:put gcx
文件已存在,是否覆盖原文件(y/n):y
文件接收成功!
请输入指令>>:put gcx
Traceback (most recent call last):
  File "/home/zhuzhu/day8作业/FTP/Ftpclient/ftp_client.py", line 154, in <module>
    client.interactive()
  File "/home/zhuzhu/day8作业/FTP/Ftpclient/ftp_client.py", line 24, in interactive
    func(cmd)
  File "/home/zhuzhu/day8作业/FTP/Ftpclient/ftp_client.py", line 59, in put
    self.client.send("收到了".encode("utf-8"))
BrokenPipeError: [Errno 32] Broken pipe

    上面客户端执行结果可以看出,第一次执行没有问题,程序成功的接收到了服务器下载的数据,但是第二次操作的时候,开始报错,提示BrokenPipeError,这是什么错误呢?
  
ThreadingHTTPServer 实现的 http 服务,如果客户端在服务器返回前,主动断开连接则服务器端会报 [Errno 32] Broken pipe 错,并导致处理线程 crash.

    上面说明,我们交互一次之后,服务器端已经断开了,不是说socketserver()的客户端一直处在活跃状态吗?是处在活跃状态没错,但是上一个链接其实已经断开了,这是socket的特性,如果没有程序接收数据,接收一次就会断开,因此要修改接收的代码,如下:

class MyTcpServer(socketserver.BaseRequestHandler):
    '''定义服务器端,socketserver,继承的是BaseRequestHandler类'''
    def handle(self):
        '''服务器端所有的功能都在handle()里面进行处理,因此接收数据也是在handle()中,只是处理的时候,调用下面模块而已'''
        while True:
            data = self.request.recv(3072)
            if len(data) == 0:
                print("客户端断开")
                break
            msg_dic = json.loads(data.decode("utf-8"))            #服务器端接收指令
            cmd = msg_dic["action"]
            if hasattr(self,cmd):                                                    #判断服务器是否有相应的指令
                self.request.send("1".encode("utf-8"))
                self.request.recv(1024)
                func = getattr(self,cmd)
                func(msg_dic)
            else:
                self.request.send("0".encode("utf-8"))
                self.request.recv(1024)

    上面只需修改handle()中的代码,让服务器端能够持续不断的接收数据,如上面代码所示,加上一个while循环,让服务器能够不断接收此链接的信息,除非这个链接断开,否则是不会终端这个链接与服务端的交互,socketserver()实现的功能时多个客户端与服务端能够同时进行交互,使其相互之间不受影响,本质上单个客户端与服务器之间的交互还是基础的socket实现的,如果不持续的接收,那么交互一次就会断开。

    上面的交互如下所示:

请输入指令>>:put gcx
文件已存在,是否覆盖原文件(y/n):y
文件接收成功!
请输入指令>>:put zmz
文件已存在,是否覆盖原文件(y/n):y
文件接收成功!
请输入指令>>:put gcx
文件已存在,是否覆盖原文件(y/n):y
文件接收成功!
请输入指令>>:客户端断开!!! 

    从上面代码可以看出,我们实现了持续的数据交换情况,没有发生太大的问题。

    2、反射的时候,反射是判断类中是否具有某种方法,hasattr(),getattr()是最常使用的方法;

    3、while a < b:

      with open(filename) as f:

        for line in f

    与

  wiht open(filename) as f:

    while a < b:

      for line in f

    上面两种写法是不一样的,为什么说呢,第一种方法,当文件遍历玩的时候,不管条件成立与否,文件都是要关闭的,这是因为with打开文件具有自动关闭文件的特性,当文件遍历完成之后,就会自动关闭文件。

  建议使用第二种方式,第二种方式就会杜绝这种情况,让程序能够即完成比较,由遍历完成文件。

    4、socketserver与socket的区别,其实两者本质都是一样的,原则上都是交互一次结束,只是socketserver封装了很多方法,能够实现不相互干扰下的交互;

    5、文件传输,要告知文件大小,方便接收,json的使用,序列化,让彼此接受的时候,容易得到之前存储的结果;

    6、规范化封装;   

posted @ 2017-09-03 21:19  (野生程序员)  阅读(464)  评论(0编辑  收藏  举报