day29 socket编程TCP和UDP

1 发送大文件

2 UDP基本使用

3 UDP实现建议版本的QQ

4 TCP与UDP之间的区别

5 socketserver模块

6 异常处理

 

1 发送大文件

 

通过客户端上传大文件到服务端 

服务端

import socket
import json
import struct

server = socket.socket()
server.bind(('127.0.0.1', 8080))
server.listen(5)


while True:
    conn, addr = server.accept()
    while True:
        try:
            header = conn.recv(4)
            dict_len = struct.unpack('i', header)[0]  # [0]一定要记得加上去
            dict_real = conn.recv(dict_len)
            dict_json = json.loads(dict_real.decode('utf-8'))
            file_size = dict_json.get('file_size')
            file_name = dict_json.get('file_name')
            recv_size = 0
            with open(file_name, 'wb') as f:
                while recv_size < file_size:
                    data = conn.recv(1024)
                    f.write(data)  # 出过错
                    recv_size += len(data)
                print('上传成功')
        except ConnectionResetError:
            break
    conn.close()

注释版

import socket, json, struct

server = socket.socket()
server.bind(('127.0.0.1', 8080))
server.listen(5)

while True:
    conn, addr = server.accept()
    while True:
        try:
            header_len = conn.recv(4)
            # 解析字典报头
            header_len = struct.unpack('i', header_len)[0]
            # 再接收字典数据
            header_dic = conn.recv(header_len)
            real_dic = json.loads(header_dic.decode('utf-8'))
            # 获取数据长度
            total_size = real_dic.get('file_size')
            # 循环接收并写入文件
            recv_size = 0
            with open(real_dic.get('file_name'), 'wb') as f:
                while recv_size < total_size:
                    data = conn.recv(1024)
                    f.write(data)
                    recv_size += len(data)
                print('上传成功')
        except ConnectionResetError as e:
            print(e)
            break
    conn.close()

 

 

客户端

import socket
import os
import json
import struct

client = socket.socket()
client.connect(('127.0.0.1', 8080))
path = r'/Users/yuanjianwei/PycharmProjects/untitled/day28/network'
path_list = os.listdir(path)

while True:
    for index, movie in enumerate(path_list, 1):
        print(index, movie)
    choice = input("请选择上传的电影>>>: ").strip()
    if choice.isdigit() and (int(choice) in range(1, len(path_list) + 1)):  # int(choice)忘了加
        movie_name = path_list[int(choice)]
        movie_path = os.path.join(path, movie_name)
        movie_size = os.path.getsize(movie_path)
        file_dict = {
            'file_name': '性感荷官在线发牌.py',
            'file_size': movie_size,
            'file_info': '要想身体好,多喝健力宝'
        }
        file_json = json.dumps(file_dict).encode('utf-8')
        header = struct.pack('i', len(file_json))  # struct.error: required argument is not an integer 忘了加len()
        client.send(header)
        client.send(file_json)
        with open(movie_path, 'rb') as f:
            for line in f:
                client.send(line)
        print('上传完成')

    else:
        print('请输入正确的数字序号!')

 

注释版

import socket, json, os, struct

client = socket.socket()
client.connect(('127.0.0.1', 8080))

while True:
    # 获取电影列表 循环展示
    MOVIE_DIR = r'D:\python脱产10期视频\day25\视频'
    movie_list = os.listdir(MOVIE_DIR)
    # print(movie_list)
    for i, movie in enumerate(movie_list, 1):
        print(i, movie)
    # 用户选择
    choice = input('please choice movie to upload>>>:')
    # 判断是否是数字且是否在列表范围内
    if choice.isdigit() and (int(choice) in range(0, len(movie_list))):
        choice = int(choice) - 1
        # 获取到用户想上传的文件路径
        path = movie_list[choice]
        # 拼接文件的绝对路径
        file_path = os.path.join(MOVIE_DIR, path)
        # 获取文件大小
        file_size = os.path.getsize(file_path)
        # 定义一个字典
        res_d = {
            'file_name': '性感荷官在线发牌.mp4',
            'file_size': file_size,
            'msg': '注意身体,多喝营养快线'
        }
        # 序列化字典
        json_d = json.dumps(res_d)
        json_bytes = json_d.encode('utf-8')
        # 1.先制作字典格式的报头
        header = struct.pack('i', len(json_bytes))
        # 2.发送字典的报头
        client.send(header)
        # 3.再发字典
        client.send(json_bytes)
        # 4.再发文件数据(打开文件循环发送)
        with open(file_path, 'rb') as f:
            for line in f:
                client.send(line)
    else:
        print('must be a number and in range')

 

 

2 UDP基本使用

UDP通信

# 数据报协议(自带报头),但没有双向通道 通信类似于发短信
1.udp协议客户端允许发空
2.udp协议不会粘包
3.udp协议服务端不存在的情况下，客户端照样不会报错？
4.udp协议支持并发

 

UDP类似于发短信,TCP类似于打电话
并发:看起来像同时运行的
并行:真正意义上的同时运行

服务端:

import socket

server = socket.socket(type=socket.SOCK_DGRAM)  # UDP协议
server.bind(('127.0.0.1', 8080))
# UDP不需要设置半连接池 它也没有半连接池的概念

# 因为没有双向通道  不需要accept  直接就是通信循环
while True:
    data, addr = server.recvfrom(1024)
    print('数据:', data)  # 客户端发来的消息
    print('地址:', addr)  # 客户端的地址
    server.sendto(data.upper(), addr)

 

客户端:

import socket

client = socket.socket(type=socket.SOCK_DGRAM)

server_addr = ('127.0.0.1', 8080)

while True:
    client.sendto(b'hello', server_addr)
    data, addr = client.recvfrom(1024)
    print('服务端发来的数据', data)
    print('服务端的地址', addr)

 

 

3 UDP实现建议版本的QQ

服务器

import socket

server = socket.socket(type=socket.SOCK_DGRAM)
server.bind(('127.0.0.1', 8080))

while True:
    data, addr = server.recvfrom(1024)
    print(data.decode('utf-8'))
    msg = input('>>>')
    server.sendto(msg.encode('utf-8'), addr)

客户端1

import socket

client = socket.socket(type=socket.SOCK_DGRAM)
addr = ('127.0.0.1', 8080)

while True:
    msg = input('>>>>')
    msg = f"来自客户端1的消息{msg}"
    client.sendto(msg.encode('utf-8'), addr)
    data, addr = client.recvfrom(1024)
    print(data.decode('utf-8'))

客户端2

import socket

client = socket.socket(type=socket.SOCK_DGRAM)
addr = ('127.0.0.1', 8080)

while True:
    msg = input('>>>>')
    msg = f"来自客户端2的消息{msg}"
    client.sendto(msg.encode('utf-8'), addr)
    data, addr = client.recvfrom(1024)
    print(data.decode('utf-8'))

 

4 TCP与UDP之间的区别

服务端

import socket

server = socket.socket(type=socket.SOCK_DGRAM)  # 需要type把默认的TCP套接字转化成UDP的
server.bind(('127.0.0.1', 8080))

data, addr = server.recvfrom(1024) # recv变成recvfrom
print(data)
data, addr1 = server.recvfrom(1024)
print(data)
data, addr2 = server.recvfrom(1024)
print(data)

 

客户端

import socket

client = socket.socket(type=socket.SOCK_DGRAM)
server_address = ('127.0.0.1', 8080)
# while True:
# msg = input('>>>:')
client.sendto(b'hello', server_address)  # send变成sendto
client.sendto(b'hello', server_address)
client.sendto(b'hello', server_address)
# data, server_addr = client.recvfrom(1024)
# print(data)

 

 

5 socketserver模块(支持TCP的并发)

服务端

import socketserver

class MyServer(socketserver.BaseRequestHandler):  # 请求处理器（RequestHandler）
    def handle(self):  # 处理器
        # print('来啦 老弟')
        while True:
            data = self.request.recv(1024)  # 接受传输的内容
            print(self.client_address)  # 客户端地址
            print(data.decode('utf-8'))
            self.request.send(data.upper())  # request 请求

if __name__ == '__main__':
    """只要有客户端连接  会自动交给自定义类中的handle方法去处理"""
    server = socketserver.ThreadingTCPServer(('127.0.0.1', 8080), MyServer)  # 创建一个基于TCP的对象  thread(穿过通过)
    server.serve_forever()  # 启动该服务对象,让server永远运行下去,除非强制停止程序

    # 设置allow_reuse_address允许服务器重用地址
    socketserver.TCPServer.allow_reuse_address = True
    # 创建一个server, 将服务地址绑定到127.0.0.1:9999
    server = socketserver.TCPServer((HOST, PORT),Myserver)

 

 

客户端

import socket

client = socket.socket()
client.connect(('127.0.0.1', 8080))

while True:
    client.send(b'hello')
    data = client.recv(1024)
    print(data.decode('utf-8'))

 

 

6 异常处理

什么是异常?

程序在运行过程中出现了不可预知的错误
并且该错误没有对应的处理机制,那么就会以异常的形式表现出来
造成的影响就是整个程序无法再正常运行

 

异常的结构

# 1.异常的类型:NAMEERROR
# 2.异常的信息:name 'fdsdfsdf' is not defined
# 3.异常的位置:Traceback (most recent call last):
File "D:/python脱产10期视频/day29/01 异常处理.py", line 1, in <module>
fdsdfsdf

 

异常的种类(两大类)

# 1.语法错误
是你程序立刻就能解决的,这种错误是不能被容忍的
语法上的错误 发现之后应该立刻解决

# 2.逻辑错误
这种错是可以被容忍的 因为一眼看不出来
针对逻辑上的错误 可以采用异常处理机制进行捕获

 

常见的错误类型

# NAMERROR 名字错误
# SyntaxError 语法错误
# KeyError 键不存在
# ValueError 值错误
# IndexError 索引错误

 

如何避免
异常处理
在你认为可能会出现bug的代码块上方try一下:注意try内部的代码块越少越好

try:
可能出错的代码
except 出错的类型 as e: # 将报错信息赋值给变量e
出错之后的处理机制