python 网络编程 03 socket模块 TCP协议 小型文件的上传

客户端向服务器上传文件

# 服务器端
import socket
import json

sk = socket.socket()
sk.bind(('127.0.0.1', 8080))  # 127.0.0.1 回环地址 自己
sk.listen()  # 监听

conn, addr = sk.accept()  # 等待接受客户端的连接 阻塞等待 返回两个值

# 通信
str_dic = conn.recv(10240).decode('utf-8')  # 接收字节数
dic = json.loads(str_dic)  # 将已编码的 JSON 字符串解码为 Python 对象
# dic = {'opt':menu.get(num),'filename':filename,'content':content}
if dic['opt'] == 'upload':
    # 上传
    filename = "new" + dic['filename']
    with open(filename, 'w', encoding='utf-8') as f:  # 写文件
        f.write(dic['content'])

elif dic['opt'] == 'download':
    # 下载
    pass

conn.close()
sk.close()

# 客户端
import socket
import os
import json

sk = socket.socket()
sk.connect_ex(('127.0.0.1', 8080))  # connect_ex带返回值,如果出错,不会报错,会返回错误的编码
# sk.connect()# connect如果出错会直接报错   connect与connect_ex 两个都是连接服务器的功能

menu = {'1': 'upload', '2': 'download'}
for k, v in menu.items():
    print(k, v)

num = input('输入功能选项:')

if num == '1':
    # 上传功能
    # {'opt'你要执行的功能, 'filename'文件名, 'content'文件内容}
    dic = {'opt': menu.get(num), 'filename': None, 'content': None}
    file_path = input('请输入一个文件的绝对路径>>>')
    # E:\XUAOWEN_python\123.txt
    filename = os.path.basename(file_path)  # 通过绝对路径获取文件名
    with open(file_path, 'r', encoding='utf-8') as f:  # 读文件
        content = f.read()

    dic['filename'] = filename  # 将数据写入字典
    dic['content'] = content
    # print(dic)
    str_dic = json.dumps(dic)  # 将 Python 对象编码成 JSON 字符串
    sk.send(str_dic.encode('utf-8'))  # 发送


elif num == '2':
    # 下载功能
    pass

sk.close()