day18-socket编程

服务端.py

from socket import *
from fileManager import*

server = socket()
server.bind(('10.7.160.92', 1234))
server.listen(512)
picture_nums = Number()
# 让服务器一直运行
while True:
    print('开始监听...')
    connection, address = server.accept()

    # 让通信一直进行
    while True:
        # 接收信息
        re_message = connection.recv(1024).decode(encoding='utf-8')
        if re_message[0] == '1':
            # print('发送文字')
            message = input('请输入：')
            connection.send(message.encode())
        elif re_message[0] == '2':
            # 发送图片
            picture_message = binary_read('./server_files./图片.jpg')
            connection.send(picture_message)
            print('图片发送成功！')
        elif re_message[0] == '3':
            # 发送视频
            video_message = binary_read('./server_files/视频.mp4')
            connection.send(video_message)
            print('视频发送成功！')
        elif re_message[0] == '4':
            # 接收图片
            picture_data = bytes()
            while True:
                picture_message = connection.recv(1024)
                if len(picture_message) < 1024:
                    picture_data += picture_message
                    binary_write('./server_files/图片%d.jpg' % picture_nums.create_num(), picture_data)
                    break
                picture_data += picture_message
            print('图片接收成功！')
        elif re_message[0] == '5':
            break
        else:
            continue
    flag = 0
    while True:
        n = input('是否关闭服务器(n/y):')
        if n == 'y':
            flag = 1
            break
        elif n == 'n':
            break
        else:
            print('输入错误！')
    if flag == 1:
        break

客户端.py

from socket import *
from fileManager import *

client = socket()
client.connect(('10.7.160.92', 1234))
text_nums = Number()
picture_nums = Number()
video_nums = Number()
while True:

    print('1.接收文字 2.接收图片 3.接收视频 4.发送图片 5.退出')
    message = input('请选择：')
    client.send(message.encode())

    if message == '1':
        # 接收文字
        all_message = client.recv(1024).decode(encoding='utf-8')
        text_write('./client_files/聊天记录%d.txt' % text_nums.create_num(), all_message)
    elif message == '2':
        # 接收图片
        picture_data = bytes()
        while True:
            picture_message = client.recv(1024)
            if len(picture_message) < 1024:
                picture_data += picture_message
                binary_write('./client_files/图片%d.jpg' % picture_nums.create_num(), picture_data)
                break
            picture_data += picture_message
    elif message == '3':
        # 接收视频
        all_data = bytes()
        while True:
            data1 = client.recv(1024)
            if len(data1) < 1024:
                all_data += data1
                all_data = binary_write('./client_files/视频%d.mp4' % video_nums.create_num(), all_data)
                break
            all_data += data1
    elif message == '4':
        # 发送图片
        picture_message = binary_read('./server_files./图片.jpg')
        client.send(picture_message)
        print('图片发送成功！')
    elif message == '5':
        break
    else:
        print('输入错误！')

fileManager.py

class Number:
    """产生序号"""
    def __init__(self):
        self._nums = 0

    @property
    def nums(self):
        return self._nums

    def create_num(self):
        self._nums += 1
        return self._nums


def text_write(file_path: str, all_message: str):
    """
    写入文本文件
    :param file_path:
    :return: None
    """
    if not all_message:
        return
    with open(file_path, 'a', encoding='utf-8') as f:
        f.write(all_message+'\n')


def binary_read(file_path: str):
    """
    获得二进制文件
    :param file_path:
    :return:
    """
    try:
        with open(file_path, 'rb') as f:
            return f.read()
    except:
        return None


def binary_write(file_path: str, all_message: bytes):
    """
    写入二进制文件
    :param file_path:
    :param all_message:
    :return:
    """
    if not all_message:
        return
    with open(file_path, 'wb') as f:
        f.write(all_message)