python 基于selectors库实现文件上传与下载

server.py

import selectors
import socket
import os
import time


base_dir =os.path.abspath(os.path.dirname(__file__))

class selectftpserver:
  def __init__(self):
    self.dic = {} # 创建空字典
    self.hasreceived = 0
    self.hassend=0
    self.sel = selectors.defaultselector() # 生成一个select对象
    self.create_socket() #create_socket()是创建socket对象函数完成绑定功能
    self.hanle() #handle()函数完成循环监听

  def create_socket(self):
    sock = socket.socket()
    sock.bind(('127.0.0.1', 8899))
    sock.listen()
    sock.setblocking(false)
    self.sel.register(sock, selectors.event_read, self.accept) # 把刚生成的sock连接对象注册到select连接列表中，并交给accept函数处理
    print("服务端已打开,请连接客户端")

  def hanle(self):
    while true:
      events = self.sel.select() # 默认是阻塞，有活动连接就返回活动的连接列表
      # 这里看起来是select，其实有可能会使用epoll，如果你的系统支持epoll，那么默认就是epoll
      # print("event==",events)
      for key, mask in events:
        callback = key.data # 去调accept函数
        callback(key.fileobj, mask) # key.fileobj就是readable中的一个socket连接对象

  def accept(self,sock, mask):
    conn, addr = sock.accept() # should be ready
    print('accepted', conn, 'from', addr)
    conn.setblocking(false) # 设定非阻塞
    self.sel.register(conn, selectors.event_read, self.read) # 新连接注册read回调函数
    self.dic[conn] = {} # 在空字典里进行了conn赋值,self.dic={conn:{},}

  def read(self, conn, mask): # 接收了conn和mask
    try: # 加异常防止客户端突然断开
      if not self.dic[conn]: # 判断self.dic[conn]里面是否是空字典，如果是空字典，代表第一次进来
        print('====第一次进来')
        data = conn.recv(1024) # conn接收了客户端发来的数据
        print("data==",str(data, encoding='utf-8'))
        cmd, filename,filesize = str(data, encoding='utf-8').split('|') # 把接收到客户端发来的包解开拿到cmd,filename,filesize个信息
        self.dic = {conn: {"cmd": cmd, "filename": filename, "filesize": int(filesize)}} # 把拿到的cmd,filename,filesize信息放到self.dic字典里去后程序返回到handle()函数里的events继续监听
        print(self.dic)
        if cmd == 'put': # 如果接收的信息是put
          conn.send(bytes("ok", encoding='utf8')) # 给客户端返回一条数据
        if self.dic[conn]['cmd'] == 'get':
          file = os.path.join(base_dir, "upload", filename)

          if os.path.exists(file):
            print("文件存在的情况，返回yes给客户端")
            filesize = os.path.getsize(file)
            self.dic[conn]['filesize'] = filesize
            print("self.dic",self.dic)
            send_info = '%s|%s' % ('yes', filesize)
            conn.send(bytes(send_info, encoding='utf8'))
          else:
            print("文件不存在情况下")
            send_info = '%s|%s' % ('no', 0)
            conn.send(bytes(send_info, encoding='utf8'))
            self.dic[conn] = {} #文件不存在的情况下，要将清空字典
      else: # 如果不是空字典代表不是第一次进来
        print('不是第一次来的')
        print(self.dic)
        if self.dic[conn].get('cmd', none): # 对接收的命令进行分发判断是put还是get
          cmd = self.dic[conn].get('cmd')
          if hasattr(self, cmd): # 如果cmd=put调用put函数，如果是cmd=get函数调用get函数
            func = getattr(self, cmd)
            func(conn)
          else:
            print("error cmd!")
            conn.close()
        else:
          print("error cmd!")
          conn.close()
    except exception as e:
      print('断开的客户端信息是：', conn)
      self.sel.unregister(conn) # 如果没有接收到数据做一个关闭解除
      conn.close()

    # put上传函数
  def put(self, conn):
    filename = self.dic[conn]['filename']
    filesize = self.dic[conn]['filesize']
    # print("base_dir",base_dir)
    path = os.path.join(base_dir, "upload", filename) # 拿到要接收的信息
    # print(filename,filesize,path)

    recv_data = conn.recv(1024) # 接收客户端上传的数据1024字节
    self.hasreceived += len(recv_data) # 把接收的数据累加到变量self.hasreceived

    with open(path, 'ab') as f: # 打开文件
      f.write(recv_data) # 把接收的数据写到文件里去

    if filesize == self.hasreceived: # 判断文件大小跟接收大小是否一样
      if conn in self.dic.keys(): # 如果文件大小跟接收大小一样清空字典
        self.dic[conn] = {}
      self.hasreceived = 0 #s上传结束之后，需要将self.hasreceived 重置成功
      print("%s 上传完毕！" % filename)

  def get(self,conn):
    filename = self.dic[conn]['filename']
    file = os.path.join(base_dir, "upload", filename)
    # filesize = os.path.getsize(file)
    filesize=self.dic[conn]['filesize']

    data = conn.recv(1024) # conn接收了客户端发来的数据
    dataok = str(data, encoding='utf-8')

    if dataok == 'ok':
      with open(file, 'rb') as f: # 打开文件
        while filesize > self.hassend: # 循环的发送文件给客户端
          contant = f.read(1024)
          recv_size = len(contant)
          conn.send(contant)
          self.hassend += recv_size
          s = str(int(self.hassend / filesize * 100)) + "%"
          print("正在下载文件： " + filename + " 已经下载：" + s)

      if filesize == self.hassend: # 判断文件大小跟接收大小是否一样
        if conn in self.dic.keys(): # 如果文件大小跟接收大小一样清空字典
          self.dic[conn] = {}
        print("%s 下载完毕！" % filename)
        self.hassend = 0

if __name__ == '__main__':
  selectftpserver()

client.py

import socket
import os,sys
base_dir=os.path.dirname(os.path.abspath(__file__))

class selectftpclient:
  def __init__(self):
    self.args=sys.argv               #sys.argv在命令行输入的参数，第一个参数默认文件名，第二个参数跟ip地址和端口
    if len(self.args)>1:              #如果大于1把第二个参数俩个值赋值给port
      self.port=(self.args[1],int(self.args[2]))
    else:
      self.port=("127.0.0.1",8899)        #如果没有第二个参数默认取这个
    self.create_socket()               #
    self.command_fanout()              #进行命令分发
    self.mainpath = os.path.join(base_dir, 'filename') # 获取该客户端下的filename路径

  #create_socket函数创建socket对象连接服务端
  def create_socket(self):
    try:
      self.sk = socket.socket()
      self.sk.connect(self.port)
      print('连接ftp服务器成功！')
    except exception as e:
      print("eroor:",e)

  #command_fanout()函数进行命令分发
  def command_fanout(self):
    while true:
      try:
        print("----------------welcome to ftp client-------------------")
        self.help_info()
        cmd_info = input('>>>请输入操作命令:').strip() # put 12.png images
        if not cmd_info:
          continue
        cmd,file = cmd_info.split() ##按照空格分隔
        # print("命令是什么", cmds)
        if cmd == "quit":
          break
        if hasattr(self, cmd):
          func = getattr(self, cmd)
          func(cmd,file)
          tag = input("是否继续进入ftp clinet,请选择y/n:").strip()
          if tag.upper() == 'y':
            continue
          else:
            break
        else:
          print('no such command ,please try again')
      except exception as e: # server关闭了
        print('%s' % e)
        break

  def help_info(self):
    print ('''
       get + (文件名）  表示下载文件
       put + (文件名）  表示上传文件
       quit       表示退出登录
    ''')

  #put()上传函数
  def put(self,cmd,file):
    if os.path.isfile(file):              #判断本地文件是否存在
      filename = os.path.basename(file)        #取出文件的名字
      filesize = os.path.getsize(file)         #取出文件的大小
      fileinfo = '%s|%s|%s'%(cmd,filename,filesize) #给文件名字大小打包成fileinf
      self.sk.send(bytes(fileinfo, encoding='utf8')) #调用send方法把fileinf发给服务端
      recvstatus = self.sk.recv(1024)         #接收服务端返回的ok内容
      print('recvstatus' , recvstatus)
      hassend = 0
      if str(recvstatus, encoding='utf8') == "ok":  #如果接收到服务端返回的ok
        with open(file, 'rb') as f:        #打开文件
          while filesize > hassend :       #循环的去上传文件
            contant = f.read(1024)
            recv_size = len(contant)
            self.sk.send(contant)
            hassend += recv_size
            s=str(int(hassend/filesize*100))+"%"
            print("正在上传文件： "+filename+" 已经上传：" +s)
        print('%s文件上传完毕' % (filename,))
    else:
      print('要上传的文件不存在')

  #get()下载函数
  def get(self,cmd,filename):
    path = os.path.join(base_dir, "download", filename) # 拿到要接收的信息
    filesize=0
    fileinfo = '%s|%s|%s' % (cmd, filename, filesize) # 给文件名字大小打包成fileinf
    print(fileinfo)
    self.sk.send(bytes(fileinfo, encoding='utf8')) # 调用send方法把fileinfo发给服务端

    recvdata = self.sk.recv(1024) # 接收服务端返回的是否存在文件内容
    recvstatus, filesize = str(recvdata, encoding='utf-8').split('|')
    print("recvstatus==",recvstatus,filesize)
    filesize = int(filesize)

    hasreceived = 0
    if recvstatus == "yes": # 如果接收到服务端返回的yes
      self.sk.send(bytes('ok', encoding='utf8')) # 通知服务端可以正常下载了

      while filesize > hasreceived: # 循环的发送文件给客户端
        recv_data = self.sk.recv(1024) # 接收客户端上传的数据1024字节
        hasreceived += len(recv_data) # 把接收的数据累加到变量self.hasreceived
        print("hasreceived",hasreceived)

        with open(path, 'ab') as f: # 打开文件
          f.write(recv_data) # 把接收的数据写到文件里去

        if filesize == hasreceived: # 判断文件大小跟接收大小是否一样
          print("%s 下载完毕！" % filename)
          recvstatus = 'yess'
    else:
       print('要下载的文件不存在')


if __name__=='__main__':
  selectftpclient()

以上就是python 基于selectors库实现文件上传与下载的详细内容，更多关于python 上传下载的资料请关注其它相关文章！