Python--网络编程-----通过socket收发文件软件开发--面向对象开发

服务端实现

import socket
import struct
import json
import os


class MYTCPServer:
    address_family = socket.AF_INET
    socket_type = socket.SOCK_STREAM
    allow_reuse_address = False
    max_packet_size = 8192
    coding = 'utf-8'
    request_queue_size = 5
    server_dir = r'C:\Users\xudachen\PycharmProjects\Python全栈开发\第三模块\网络编程\通过socket发送文件\服务端\file_upload'

    def __init__(self, server_address, bind_and_active=True):
        """Constructor. May be extended, do not override"""
        self.server_address = server_address
        self.socket = socket.socket(self.address_family, self.socket_type)

        if bind_and_active:
            try:
                self.server_bind()
                self.server_activate()
            except:
                self.server_close()
                raise

    def server_bind(self):
        """Called by constructor to bind the socket"""
        if self.allow_reuse_address:
            self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.socket.bind(self.server_address)
        self.server_address = self.socket.getsockname()

    def server_activate(self):
        """Called by constructor to activate the server"""
        self.socket.listen(self.request_queue_size)

    def server_close(self):
        """Called to clean-up the server"""
        self.socket.close()

    def get_request(self):
        """Get the request and client address from the socket"""
        return self.socket.accept()

    def close_request(self, request):
        """Called to clean-up an individual request"""
        request.close()

    def run(self):
        while True:
            self.conn, self.client_addr = self.get_request()
            print('from client', self.client_addr)
            while True:
                try:
                    head_struct = self.conn.recv(4)
                    if not head_struct:
                        break

                    head_len = struct.unpack('i', head_struct)[0]
                    head_json = self.conn.recv(head_len).decode(self.coding)
                    head_dic = json.loads(head_json)

                    print(head_dic)
                    cmd = head_dic['cmd']
                    if hasattr(self, cmd):
                        func = getattr(self, cmd)
                        func(head_dic)
                except Exception:
                    break

    def put(self, args):
        file_path = os.path.normpath(os.path.join(self.server_dir, args['filename']))

        filesize = args['filesize']
        recv_size = 0
        print('---->', file_path)
        with open(file_path, 'wb') as f:
            while recv_size < filesize:
                recv_data = self.conn.recv(self.max_packet_size)
                f.write(recv_data)
                recv_size += len(recv_data)
                print('recvsize:%s filesize:%s' % (recv_size, filesize))


tcpserver1 = MYTCPServer(('127.0.0.1', 8080))
tcpserver1.run()

客户端实现

import socket
import struct
import json
import os


class MYTCPClient:
    address_family = socket.AF_INET
    socket_type = socket.SOCK_STREAM
    allow_reuse_address = False
    max_packet_size = 8192
    coding = 'utf-8'
    request_queue_size = 5

    def __init__(self, server_address, connect=True):
        self.server_address = server_address
        self.socket = socket.socket(self.address_family, self.socket_type)

        if connect:
            try:
                self.client_connect()
            except:
                self.client_close()
                raise

    def client_connect(self):
        self.socket.connect(self.server_address)

    def client_close(self):
        self.socket.close()

    def run(self):
        while True:
            inp = input(">>:").strip()  # put test.py
            if not inp:
                continue
            l = inp.split()  # ['put','test.py']
            cmd = l[0]  # put
            if hasattr(self, cmd):
                func = getattr(self, cmd)
                func(l)

    def put(self, args):  # ['put','test.py']
        cmd = args[0]  # put
        filename = args[1]  # test.py
        if not os.path.isfile(filename):
            print('file:%s is not exists' % filename)
            return
        else:
            filesize = os.path.getsize(filename)

        head_dic = {'cmd': cmd,
                    'filename': os.path.basename(filename),
                    'filesize': filesize}
        print(head_dic)
        head_json = json.dumps(head_dic)
        head_json_bytes = bytes(head_json, encoding=self.coding)

        head_struct = struct.pack('i', len(head_json_bytes))
        self.socket.send(head_struct)
        self.socket.send(head_json_bytes)
        send_size = 0
        with open(filename, 'rb') as f:
            for line in f:
                self.socket.send(line)
                send_size += len(line)
                print(send_size)
            else:
                print('upload successful')


client = MYTCPClient(('127.0.0.1', 8080))
client.run()

先启动服务端，再启动客户端，在客户端输入put test.py

客户端运行结果为;

>>:put test.py
{'cmd': 'put', 'filename': 'test.py', 'filesize': 1014}
17
20
61
83
86
128
140
142
187
201
241
253
255
268
294
313
328
354
386
398
400
413
416
440
443
477
489
504
507
536
593
632
651
687
723
779
805
807
822
856
877
879
894
896
955
988
1014
upload successful
>>:

服务端运行结果：

from client ('127.0.0.1', 55771)
{'cmd': 'put', 'filename': 'test.py', 'filesize': 1014}
----> C:\Users\xudachen\PycharmProjects\Python全栈开发\第三模块\网络编程\通过socket发送文件\服务端\file_upload\test.py
recvsize:1014 filesize:1014

实际效果：

注意，在windows系统中使用pycharm，上传成功后，并不会立即在上传文件夹中显示，需要找到文件夹路径，点进去，然后才会在pycharm中看到，