基于 tcp协议完成的上传下载

ps: 下面两个版本全部是 基于 tcp协议完成的.

简单版(无注释):

　　服务器(server)代码:

from socket import socket
sk = socket()
sk.bind(('127.0.0.1', 8585))
sk.listen()

def display(): return '1, 上传\n2, 下载'


def upload(word,chose_path):
    # chose_path = input('>>>请输入要上传的文件路径:')
    with open(chose_path.decode('utf-8'), 'wb') as f:
        f.write(word)


def download():
    name = '__init__.py'
    with open(name, 'rb') as f:
        word = f.read()
    return name.encode('utf-8'), word


def main():
    while 1:
        conn, addr = sk.accept()
        while 1:
            conn.send(display().encode('utf-8'))
            msg_r = conn.recv(1024).decode('utf-8')
            if msg_r == 'q':continue
            if msg_r == '1':
                word = conn.recv(1024*10)
                chose_path = conn.recv(1024 * 10)
                upload(word, chose_path)
                conn.send('上传成功'.encode('utf-8'))
            elif msg_r == '2':
                name, word = download()
                conn.send(name)
                conn.send(word)

main()

　　客户端(client) 代码:

from socket import socket

sk = socket()
sk.connect(('127.0.0.1', 8585))
flag = True

def upload():
    chose_path = input('>>>请输入要上传的文件路径:')
    with open(chose_path, 'rb') as f:
        word = f.read()
        return chose_path.encode('utf-8'), word


def download(name, word):
    with open(name,'wb') as f:
        f.write(word)
    return '下载成功'


def main():
    global flag
    msg_r = sk.recv(1024).decode('utf-8')
    print(msg_r)
    choise = input('>>>请选择功能ID(输入"q"退出程序):').strip().encode('utf-8')
    sk.send(choise)
    if choise == 'q'.encode('utf-8'): flag = False
    if choise == '1'.encode('utf-8'):
        chose_path, word = upload()
        sk.send(word)
        sk.send(chose_path)
        print(sk.recv(1024).decode('utf-8'))
    elif choise == '2'.encode('utf-8'):
        name = sk.recv(1024).decode('utf-8')
        word = sk.recv(1024)
        print(download(name, word))

while flag:
    main()
sk.close()

　　

升级版 (有注释,此程序会出现个别粘包现象):

　　服务器(server):

from socket import socket
import os
import pickle

sk = socket()
sk.bind(('127.0.0.1', 8589))
sk.listen()

def display(): return '1, 上传\n2, 下载'     #  此函数用来发送第一个选择


def update(name): # 更名
    chose = conn.recv(1024).decode('utf-8')  # 接收客户的选择
    if chose == '1':pass   # 如果选择 1 覆盖 则不用操作
    elif chose == '2':      # 如果选择 2 更改 接收客户端的消息
        name = conn.recv(1024).decode('utf-8')      #  新文件的名字
    return name


def upload(word,chose_path):    # 上传函数
    chose_path = chose_path.decode('utf-8')  # 把接收的文件名 解码

    # 判断 该文件 是否已经存在与 服务器 当前工作目录下:
    if chose_path in os.listdir('E:/老男孩工具/13期课后作业及视频/day33/myself_test/new_upload_download/updata_server_path'):    # 路径 按照自己的文件夹来自定义
        # 如果在, 向客户端发消息, 并且询问 用户的选择
        conn.send('服务器已存在该文件:选择覆盖还是更名:\n1, 覆盖\n2, 更名'.encode('utf-8'))
        name = update(chose_path)     # 调用 update() 更改函数 获得新文件的名字 或 老文件的名字
        if not name:chose_path = name  # 如果 name 不为None  给文件进行改名:
    with open(chose_path, 'wb') as f:  # 打开文件 进行写入
        f.write(word)


def download(): # 下载
    # 把服务器工作目录下 所有文件 装到列表里
    path_list = os.listdir(r'E:\老男孩工具\13期课后作业及视频\day33\myself_test\new_upload_download\updata_server_path')
    lis = [{x:y} for x,y in enumerate(path_list,1)] # 生成一个 有序的元素为字典的 内容列表
    pic = pickle.dumps(lis)    # 序列化一下
    conn.send(pic)          # 发送给 客户端, 然后询问 客户要下载那个文件的id
    file_id = conn.recv(1024).decode('utf-8')  #   接收到 id
    name = lis[int(file_id)-1][int(file_id)]    # 取得 相对应的文件名
    with open(name, 'rb') as f:         # 打开文件, 读到内容
        word = f.read()
    return name.encode('utf-8'), word    # 返回文件名字, 和 内容


def main(): ## 服务器
    while 1:
        conn.send(display().encode('utf-8'))  # 发送 第一条消息:  '1, 上传\n2, 下载'
        msg_r = conn.recv(1024).decode('utf-8') #  接收 客户端的选择信息
        if msg_r == 'q':break               # 如果是q 退出程序
        if msg_r == '1':                # 如果 选择 1  此时需要在客户端 向用户索要 文件信息.
            word = conn.recv(1024*10)    # 接收 文件内容
            chose_path = conn.recv(1024 * 10)   #   接收文件名字
            upload(word, chose_path)        # 调用上传 函数 并且把文件内容 和名字 传进去
            conn.send('上传成功'.encode('utf-8'))   # 如果程序成功运行到了这里 就把消息发过去.
        elif msg_r == '2':              # 如果选择2 下载
            name, word = download()     # 调用 下载函数 download() 把返回值 分别 赋值 name, word
            conn.send(name)                # 发送 文件名字.
            conn.send(word)         # 最后一条消息: 文件内容
    return


if __name__ == '__main__':
    while 1:
        conn, addr = sk.accept()
        main()
        conn.close()

　　客户端(client) 代码:

import pickle
from socket import socket
import os

sk = socket()
sk.connect(('127.0.0.1', 8589))
flag = True

def upload():
    chose_path = input('>>>请输入要上传的文件路径:')
    with open(chose_path, 'rb') as f:
        word = f.read()                         #  读取文件内容
    chose_path = os.path.basename(chose_path)   # 取得 路径的 文件名
    return chose_path.encode('utf-8'), word    # 把word  name  返回


def download(name, word):
    path = input('>>>请选择工作路径:')
    new_path = os.path.join(path, name)
    with open(new_path,'wb') as f:
        f.write(word)
    return '下载成功'


def update(): # 更名
    chose = input('>>>请输入您的选择:').strip().encode('utf-8')
    sk.send(chose) # 把选择发给 服务器, 让服务器做判断
    if chose == '1'.encode('utf-8'):pass   # 如果是1 则不作任何操作
    elif chose == '2'.encode('utf-8'):      # 如果是2 则更名
        new_name = input('>>>请输入新的名字:').encode('utf-8')
        sk.send(new_name)        # 把新名字 给服务器 发过去
        print(sk.recv(1024).decode('utf-8')) # 接收 服务器 发送的 成功的消息 并且打印出来


def main():   # 客户端
    global flag
    msg_r = sk.recv(1024).decode('utf-8')   #  接收第一条消息,   '1, 上传\n2, 下载'
    print(msg_r)
    choise = input('>>>请选择功能ID(输入"q"退出程序):').strip().encode('utf-8')
    sk.send(choise)  # 把 客户选择的功能发送给 服务器
    if choise == 'q'.encode('utf-8'): flag = False
    if choise == '1'.encode('utf-8'):
        chose_path, word = upload()  # 调用upload() 上传函数  返回 word 和 文件名
        sk.send(word) #         把 word发送给服务器
        sk.send(chose_path) #       把文件名发送给 服务器
        ret = sk.recv(1024).decode('utf-8')     # 接收 服务器传输过来的 判断是否 存在的语句
        print(ret)
        if ret == '服务器已存在该文件:选择覆盖还是更名:\n1, 覆盖\n2, 更名':      # 如果是这句话就 调用下面的函数, 如果不是这句话 会打印 上传成功
            update()   # 此函数的功能 和 服务器里面的 功能一样 :更名

    elif choise == '2'.encode('utf-8'):    # 当选择 2 下载 时
        pic = pickle.loads(sk.recv(1024))     #  把服务器发来的 pickle 字典 反序列化了
        print('下面是所有子文件:')
        for i in pic:
            print(i)                        #  一个一个打印出来
        file_id = input('>>>请输入想要下载的文件的ID:').strip().encode('utf-8')
        sk.send(file_id)                    # 把 id 发给 服务器
        name = sk.recv(1024).decode('utf-8')   # 接收 服务器发来的 文件的名字
        word = sk.recv(1024*1000)   #  最后一条消息: 接收文件名
        download(name, word)         # 此函数 是用来把内容写入新文件
        print('下载成功')


if __name__ == '__main__':
    while flag:
        main()
    sk.close()