Python模拟FTP文件服务器的操作方法

分为服务端和客户端，要求可以有多个客户端同时操作。

客户端可以查看服务器文件库中有什么文件。
客户端可以从文件库中下载文件到本地。
客户端可以上传一个本地文件到文件库。
使用print在客户端打印命令输入提示，引导操作

服务端

# 导入相关模块
from socket import *
from multiprocessing import Process
import signal, os, time
# 绑定IP地址
IP = "127.0.0.1"
# 绑定端口
PORT = 8888
ADDR = (IP, PORT)
# 绑定服务器的指定目录
DIR = "/home/max/ftp"
# 处理查看文件请求
def browse(c):
  # 列表方式查看文件
  list = os.listdir("%s" % DIR)
  # 按通信协议格式组织数据
  msg = "B " + ";".join(list)
  # 发送到客户端
  c.send(msg.encode())
# 处理客户端下载文件请求
def download(c, file):
  # 判断文件是否存在且是否是文件
  if file in os.listdir(DIR) and os.path.isfile("%s/%s" % (DIR, file)):
    # 打开文件
    f = open("%s/%s" % (DIR, file), "rb")
    # 发送下载代码,告知客户端进入"D"
    c.send(("D %s " % file).encode())
    # 等待客户端先进入"D"模式下的recv阻塞函数
    time.sleep(0.1)
    # 循环发送文件
    while True:
      data = f.read(1000)
      if not data:
        # 设置间隔,等待文件传输完整
        time.sleep(0.1)
        # 约定的信息让客户端退出接收循环
        c.send(b"finished")
        break
      c.send(data)
    f.close()
  # 否则按协议格式发送错误代码
  else:
    c.send(b"D fileerror ")
# 处理客户端上传文件请求
def upload(c, file):
  # 创建文件
  f = open("%s/%s" % (DIR, file), "wb")
  # 循环接收文件
  while True:
    data = c.recv(1024)
    # 收到约定的信息退出循环
    if data == b"finished":
      break
    f.write(data)
  f.close()
# 处理客户端退出请求
def login_out(c):
  # 按协议格式组织信息
  msg = "O closed "
  # 发给客户端的recv_msg进程
  c.send(msg.encode())
def deal(c):
  # 发送代码告知客户端连接成功
  c.send(b"K ")
  while True:
    # 循环接收客户端请求,约定通信协议为"* ** ****"格式
    msg = c.recv(1024)
    # 如果客户端崩了,解除该子进程
    if not msg:
      break
    # 解析请求
    req = msg.decode().split(" ", 2)
    # 处理查看文件请求,跳入browse函数
    if req[0] == "B":
      browse(c)
    # 处理客户端下载文件请求,跳入download函数
    elif req[0] == "D":
      download(c, req[1])
    # 处理客户端上传文件请求,跳入upload函数
    elif req[0] == "U":
      upload(c, req[1])
    # 处理客户端退出请求,跳入login_out函数
    elif req[0] == "O":
      login_out(c)
      # 跳出循环,结束子进程
      break
# 主程序,父进程用于接收客户端请求并循环创建子进程,子程序处理请求
def main():
 # 创建tcp套接字
 s = socket()
 # 绑定服务器地址
 s.bind(ADDR)
 # 设置监听套接字
 s.listen()
 # 处理僵尸进程
 signal.signal(signal.SIGCHLD, signal.SIG_IGN)
  while True:
    # 连接客户端
    c, addr = s.accept()
    # 创建子进程,用以处理客户端请求,跳入deal函数
    p = Process(target=deal, args=(c,))
    # 子程序开始执行
    p.start()
if __name__ == '__main__':
  main()

客户端

# 导入相关模块
from socket import *
import os, sys, time
# 绑定服务端IP地址
IP = "127.0.0.1"
# 绑定服务端端口
PORT = 8888
ADDR = (IP, PORT)
# 收到约定的信息退出循环
# 发送消息进程
def send_msg(s):
  # 等待接收进程先运行到"K"分支
  time.sleep(0.1)
  while True:
    try:
      # 输入指令
      data = input(">>>")
    except:
      # 客户端错误,向服务端发送O
      data = "O"
    if data == "B": # 查看目录
      msg = "B "
      s.send(msg.encode())
    elif data == "D": # 下载文件
      # 输入想要下载的文件
      want = input("download ? file:")
      msg = "D %s " % want
      s.send(msg.encode())
    elif data == "U": # 上传文件
      file = input("upload ? file:")
      # 判断文件是否在客户端文件所在的目录且是文件
      if file in os.listdir(os.getcwd()) and os.path.isfile("%s/%s" % (os.getcwd(), file)):
        msg = "U %s " % file
        s.send(msg.encode())
        f = open("%s" % file, "rb")
        # 等待服务端进入upload的recv阻塞函数
        time.sleep(0.1)
        while True:
          data = f.read(1000)
          if not data:
            # 设置间隔,等待文件传输完整
            time.sleep(0.1)
            # 约定的信息让客户端退出接收循环
            s.send(b"finished")
            break
          s.send(data)
        f.close()
        print("upload succussfully")
      # 否则按协议格式显示错误代码
      else:
        print("file not exist\ninput BDUO to forward\n", end="")
    elif data == "O": # 断开连接
      msg = "O "
      s.send(msg.encode())
      sys.exit()
    # 指令错误
    else:
      print("input error")

# 接收消息进程
def recv_msg(s):
  while True:
    data = s.recv(1024)
    # 解析数据
    msg = data.decode().split(" ", 2)
    if msg[0] == "K": # 登录成功反馈
      print(
        "login in successfully\ninput B to browse,D to download,U to upload,O to login out")
    elif msg[0] == "B": # 查看目录反馈
      # 如果文件库不为空
      if msg[2]:
        print("files:", msg[2], "\n>>>", end="")
      else:
        print("files: no files", "\n>>>", end="")
    elif msg[0] == "D":
      # 服务端文件不存在
      if msg[1] == "fileerror":
        print("file not exist\ninput BDUO to forward\n>>>", end="")
        continue
      # 服务端文件存在
      else:
        f = open("%s" % msg[1], "wb")
        while True:
          data = s.recv(1024)
          # 收到约定的信息退出循环
          if data == b"finished":
            break
          f.write(data)
        f.close()
        print("download successfully\n>>>", end="")
    elif msg[0] == "O": # 收到来自发送消息进程发送到服务端的断开请求
      # 进程退出并打印提示
      sys.exit("login out successfully")

# 主程序
def main():
  # 创建tcp套接字
  s = socket()
  # 连接服务端
  s.connect(ADDR)
  # 创建多进程,子进程用于发送消息,父进程用于接收消息
  pid = os.fork()
  if pid < 0:
    print("system error")
  # 子进程
  elif pid == 0:
    send_msg(s)
  # 父进程
  else:
    recv_msg(s)
if __name__ == '__main__':
  main()

总结

以上所述是小编给大家介绍的Python_模拟FTP文件服务器的操作方法，希望对大家有所帮助，也非常感谢大家对脚本之家网站的支持！