socket实现ftp上传下载

socket实现ftp文件的上传和下载

server端代码：

import socket

import json

import struct

import os

soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

soc.bind(('127.0.0.1', 8021))

soc.listen(5)

# 上传函数

def uploading_file():

    while True:

        try:

            ftp_dir = r'F:\shpython11\pycharmwork\first_project\study_start\day32\ftp_dir'

            if not os.path.isdir(ftp_dir):  # 这个是作为我们上传了之后的文件放在这个位置,你改成自己本地的

                os.mkdir(ftp_dir)

            head_bytes_len = conn.recv(4)  # 拿到struct后的头长度

            head_len = struct.unpack('i', head_bytes_len)[0]  # 取出真正的头长度

            # 拿到真正的头部内容

            head_bytes = conn.recv(head_len)

            # 反序列化后取出头

            head = json.loads(head_bytes)

            file_name = head['file_name']

            file_path = os.path.join(ftp_dir,file_name)

            data_len = head['data_len']

            with open(file_path, 'wb') as fw:

                while data_len > 1024:

                    fw.write(conn.recv(1024))

                    data_len -= 1024

                else:

                    fw.write(conn.recv(data_len))

            conn.send(f'上传文件 {file_name} 成功'.encode('utf8'))

        except Exception:

            break

while True:

    print('等待客户端连接。。。')

    conn, addr = soc.accept()

    print('客户端已连接：', addr)

    choice = conn.recv(1).decode('utf8')

    if choice == 'q':

        break

    if choice == '1':

        print('客户端选择了ftp上传服务')

        uploading_file()

    elif choice == '2':

        print('客户端选择了ftp下载服务')

        while True:

            try:

                ftp_dir = r'F:\shpython11\pycharmwork\first_project\study_start\day32\ftp_dir'

                if not os.path.isdir(ftp_dir):  # 这个是作为我们上传了之后的文件放在这个位置,你改成自己本地的

                    os.mkdir(ftp_dir)

                file_list = os.listdir(ftp_dir)

                head = {'file_list': file_list}

                head_bytes = json.dumps(head).encode('utf8')

                head_bytes_len = struct.pack('i', len(head_bytes))

                conn.send(head_bytes_len)

                conn.send(head_bytes)

                client_head_bytes_len = conn.recv(4)

                client_head_len = struct.unpack('i', client_head_bytes_len)[0]

                client_head_bytes = conn.recv(client_head_len)

                client_head = json.loads(client_head_bytes)

                choice = client_head['choice']

                file_name = file_list[int(choice)]

                file_path = os.path.join(ftp_dir, file_name)

                with open(file_path, 'rb') as fr:

                    data = fr.read()

                server_head = {'data_len': len(data), 'file_name': file_name}  # 自定义头

                server_head_bytes = json.dumps(server_head).encode('utf8')

                server_head_bytes_len = struct.pack('i', len(server_head_bytes))

                conn.send(server_head_bytes_len)

                conn.send(server_head_bytes)

                conn.send(data)

                conn.send(f'下载文件 {file_name} 成功'.encode('utf8'))

            except Exception:

                break

    conn.close()

soc.clone()

client端代码：

import socket

import os

import struct

import json

client_soc = socket.socket()

client_soc.connect(('127.0.0.1',8021))

# 获取文件夹下的文件

def get_file_list(file_path):

    if os.path.exists(file_path):

        if os.path.isfile(file_path):

            return True,file_path

        else:

            file_list = os.listdir(file_path)

            if file_list==[]:

                return False,'当前文件夹为空，请重新选择'

            else:

                return True,file_list

    else:

        return False,'你输入的文件路径不对'

# 上传函数

def uploading_file():

    print('欢迎进入ftp文件上传系统')

    while True:

        dir_path = input('请输入文件所在的文件夹路径或文件路径(输入q退出):')

        if dir_path == 'q':

            print('你选择了退出上传系统')

            break

        flag, dir_list = get_file_list(dir_path)

        if flag:

            if os.path.isfile(dir_path):

                file_name = dir_path.split('\\')[-1]

                file_path = dir_path

            else:

                for ind, file in enumerate(dir_list):

                    print(f'文件编号：{ind} 对应的文件名为：{file}')

                choice = input('请输入文件编号进行上传：').strip()

                choice = int(choice)

                file_name = dir_list[choice]

                file_path = os.path.join(dir_path, file_name)

            with open(file_path,'rb') as fr:

                data = fr.read()

            head = { 'data_len': len(data),'file_name':file_name}  # 自定义头

            head_bytes = json.dumps(head).encode('utf8')

            head_bytes_len = struct.pack('i',len(head_bytes))

            client_soc.send(head_bytes_len)

            client_soc.send(head_bytes)

            client_soc.send(data)

            msg = client_soc.recv(1024)

            print(msg.decode('utf8'))

        else:

            print('你输入的文件路径不存在')

# 下载函数

def download():

    print('欢迎来到ftp下载系统')

    head_bytes_len = client_soc.recv(4)

    head_len = struct.unpack('i', head_bytes_len)[0]

    head_bytes = client_soc.recv(head_len)

    head = json.loads(head_bytes)

    file_list = head['file_list']

    if file_list == []:

        print('当前ftp中无文件可下载，等待上传中')

    else:

        print('当前ftp中有如下文件')

        for ind,file in enumerate(file_list):

            print(f'文件编号：{ind} 对应的文件名为：{file}')

        choice = input('请选择要下载文件的编号：')

        choice_head = {'choice':choice}  # 自定义头

        choice_head_bytes = json.dumps(choice_head).encode('utf8')

        choice_head_bytes_len = struct.pack('i', len(choice_head_bytes))

        client_soc.send(choice_head_bytes_len)

        client_soc.send(choice_head_bytes)

        # 接收用户传递过来的文件

        client_head_bytes_len = client_soc.recv(4)  # 拿到struct后的头长度

        client_head_len = struct.unpack('i', client_head_bytes_len)[0]  # 取出真正的头长度

        # 拿到真正的头部内容

        client_head_bytes = client_soc.recv(client_head_len)

        # 反序列化后取出头

        client_head = json.loads(client_head_bytes)

        file_name = client_head['file_name']

        data_len = client_head['data_len']

        with open(file_name, 'wb') as fw:

            while data_len > 1024:

                fw.write(client_soc.recv(1024))

                data_len -= 1024

            else:

                fw.write(client_soc.recv(data_len))

        msg = client_soc.recv(1024)

        print(msg.decode('utf8'))

while True:

    msg = '''

    1.文件上传

    2.文件下载

    q.退出系统

    '''

    print(msg)

    choice = input('请做出你的选择：').strip()

    client_soc.send(choice.encode('utf8'))

    if choice == 'q':

        print('你选择了退出系统')

        break

    if choice == '1':

        uploading_file()

    elif choice == '2':

        download()