13python入门——网络基础案例

socket案例1

服务端与客户的链接、收发数据
服务端

import socket

sk = socket.socket()

#绑定端口
sk.bind(('127.0.0.1',8080))
sk.listen()

#接受链接、地址
cnn,addr = sk.accept()
#接受消息
res = cnn.recv(1024)
#给发送消息
cnn.send(b'I love girl')
print(res)
cnn.send(('星光不付赶路人').encode('utf-8'))

cnn.close()
sk.close()

用户端

import socket

sk = socket.socket()

#链接外部端口
sk.connect(('127.0.0.1',8080))
#发送信息
sk.send(b'Hello')
#接受信息1024个字节
res = sk.recv(1024)
print(res)
res1 = sk.recv(1024)
print(res1.decode())

sk.close()

socket案例2

实现一直链接，输入q退出
服务端

import socket

sk = socket.socket()
sk.bind(('127.0.0.1',8080))
sk.listen()
while True:
    # 建立连接
    conn,addr = sk.accept()
    print(addr)
    while True:
        inp = input('setver>>>')
        if inp == 'q':
            conn.send(inp.encode('utf-8'))
            break
        conn.send(inp.encode('utf-8'))
        mesg = conn.recv(1024).decode()
        if mesg == 'q':break
        print(mesg)
    conn.close()
sk.close()

客户端1

import socket

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

while True:
    mesg = sk.recv(1024).decode()
    print(mesg)
    if mesg == 'q':break
    inp = '1>>>'+input('client1>>>')
    if inp == 'q':
        sk.send(inp.encode('utf-8'))
        break
    sk.send(inp.encode('utf-8'))

sk.close()

客户端2

import socket

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

while True:
    mesg = sk.recv(1024).decode()
    print(mesg)
    if mesg == 'q':break
    inp = '2>>>'+input('client1>>>')
    if inp == 'q':
        sk.send(inp.encode('utf-8'))
        break
    sk.send(inp.encode('utf-8'))

sk.close()

黏包解决案例

借助struct实现，先把发送数据的大小发给你，在接受指定大小数据
服务端

import struct
import socket

sk = socket.socket()
sk.bind(('127.0.0.1',8080))
sk.listen()

conn, addr = sk.accept()
while True:
    inp = input('>>>')
    if inp == 'q':
        conn.send(b'q')
        break
    conn.send(inp.encode('utf-8'))
    num = conn.recv(4)
    num = struct.unpack('i',num)[0]
    res = conn.recv(int(num)).decode('gbk')
    print(res)

conn.close()
sk.close()

客户端

import struct
import socket
import subprocess

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

while True:
    cmd = sk.recv(1024).decode('utf-8')
    if cmd == 'q':
        break
    res = subprocess.Popen(cmd,shell=True,
                           stdout=subprocess.PIPE,
                           stderr=subprocess.PIPE
    )
    std_out = res.stdout.read()
    std_err = res.stderr.read()
    len_num = len(std_out)+len(std_err)
    num = struct.pack('i',len_num)
    sk.send(num)
    sk.send(std_out)
    sk.send(std_err)
sk.close()

文件传输案例

服务端

import json
import socket
import struct

sk = socket.socket()
sk.bind(('127.0.0.1',8090))
sk.listen()
buffer = 1024

conn, addr = sk.accept()

head_len = conn.recv(4)
head_len = struct.unpack('i',head_len)[0]
print(head_len)
json_head = sk.recv(head_len).decode('utf-8')
head = json.loads(json_head)
print(head)
filesize = head['filesize']

with open(head['filename'],'wb') as f:
    while filesize:
        if filesize >= buffer:
            cotent = conn.recv(buffer)
            f.write(cotent)
            filesize -= buffer
        else:
            cotent = conn.recv(filesize)
            f.write(cotent)
            break

conn.close()
sk.close()

客户端

import socket
import json
import os
import struct

# 建立连接
sk = socket.socket()
sk.connect(('127.0.0.1',8090))
buffer = 1024

head = {
    'filepath':r'E:\BaiduYunDownload',
    'filename':'111.mp4'
}
file_path = os.path.join(head['filepath'],head['filename'])
filesize = os.path.getsize(file_path)
head['filesize'] = filesize
print(head)
# 转换为json数据，再转换为byte数据
json_head = json.dumps(head).encode('utf-8')
print(json_head)
# 对数据长度进行转换
head_len = len(json_head)
print(head_len)
num = struct.pack('i',head_len)
sk.send(num)
sk.send(json_head)


with open(file_path,'rb') as f:
    while filesize:
        if filesize >= buffer:
            content = f.read(buffer)
            sk.send(content)
            filesize -= buffer
        else:
            content = f.read(filesize)
            sk.send(content)
            break

sk.close()