python-- 上传小文件

socket_udp

UDP协议的通信优势：允许一个服务器同时和多个客户端通信

服务端：

import socket

server = socket.socket(type=socket.SOCK_DGRAM) # udp协议
server.bind(('127.0.0.1', 12345))
# 收发
while 1:
msg, addr = server.recvfrom(1024) # 接受来自哪里的消息,addr元组类型，是ip和端口
print(msg.decode('utf-8'))
data = input('>>>')
server.sendto(data.encode('utf-8'), addr) # 发送消息

客户端：

import socket

client = socket.socket(type=socket.SOCK_DGRAM)
while 1:
msg_s = input(">>>")
client.sendto(msg_s.encode('utf-8'), ('127.0.0.1', 12345))
msg, addr = client.recvfrom(1024)
print(msg.decode('utf-8'))

继成 socket 类

My_UDP.py

import socket


class MySocket(socket.socket): # 继承自 socket文件中的socket类,此时socket就是父类
def __init__(self, encoding='utf-8'):
self.encoding = encoding
super(MySocket, self).__init__(type=socket.SOCK_DGRAM) # 执行父类socket中的__init__方法

def my_sendto(self, msg, addr):
return self.sendto(msg.encode(self.encoding), addr) # 调用父类中的sendto方法

def my_recvfrom(self, num):
msg_r, addr = self.recvfrom(num) # 调用父类的recvfrom方法
return

服务端

from My_UDP import MySocket

sk = MySocket()
sk.bind(('127.0.0.1', 8080))
msg, addr = sk.my_recvfrom(1024)
print(msg)
sk.close()

客户端

from My_UDP import MySocket

sk = MySocket()
sk.my_sendto('abcabc中国', ('127.0.0.1', 8080))
sk.close()

上传小文件

服务端

import socket
import json

sk = socket.socket()
sk.bind(('127.0.0.1', 8080))
sk.listen()
conn, addr = sk.accept()
str_dic = conn.recv(9090).decode('utf-8')
dic = json.loads(str_dic)
# dic = {'opt':menu.get(num),'filename':filename,'content':content}
if dic['opt'] == 'upload':
# 上传
filename = '1' + 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)) # 带返回值,如果出错,不会报错,会返回错误的编码
# sk.connect()# 会直接报错 两个都是连接服务器的功能
menu = {'1': 'upload', '2': 'download'}
for k, v in menu.items():
print(k, v)
num = input('请输入功能选项:')
if num == '1':
# 上传功能
# {你要执行的功能, 文件名, 文件内容}
dic = {'opt': menu.get(num), 'filename': None, 'content': None}
file_path = input('请输入一个文件的绝对路径>>>')
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
str_dic = json.dumps(dic)
sk.send(str_dic.encode('utf-8'))

elif num == '2':
# 下载功能
pass
else:
print('错误')
sk.close()