socket 网络编程

一、服务端

import socket
sk = socket.socket()
sk.bind(('127.0.0.1',9999,))
sk.listen(5)

while True:
    conn,address = sk.accept()
    conn.sendall(bytes('欢迎致电10086',encoding='utf-8'))
    while True:
        ret_bytes = conn.recv(1024)
        ret_str = str(ret_bytes,encoding='utf-8')
        if ret_str =='q':
            break
        conn.sendall(bytes(ret_str + 'ok',encoding='utf-8'))

二、客户端

import socket
obj = socket.socket()
obj.connect(('127.0.0.1',9999,))

rec_bytes = obj.recv(1024)
rec_str = str(rec_bytes,encoding='utf-8')
print(rec_str)
while True:
    inp = input('请输入要发送的内容：')
    if inp =='q':
        obj.sendall(bytes(inp,encoding='utf-8'))
        break
    else:
        obj.sendall(bytes(inp, encoding='utf-8'))
        ret = str(obj.recv(1024),encoding='utf-8')
        print(ret)
obj.close()

三、服务端传输文件

#！/usr/bin/env python
# -*- coding:utf-8 -*-
import socket
sk = socket.socket()
sk.bind(('127.0.0.1',9999,))
sk.listen(5)

while True:
    conn,address = sk.accept()
    conn.sendall(bytes('欢迎登陆FTP系统',encoding='utf-8'))
    # 先接受文件大小
    file_size = str(conn.recv(1024),encoding='utf-8')
    # 与客户端交互，发送响应，解决缓冲区粘包的问题
    conn.sendall(bytes('传输ok', encoding='utf-8'))
    total_size = int(file_size)
    has_recv = 0
    f = open('new.jpg','wb')
    # 接收文件直到完毕
    while True:
        if total_size ==has_recv:
            break
        data = conn.recv(1024)
        f.write(data)
        has_recv +=len(data)
    f.close()

四、客户端发送文件

#！/usr/bin/env python
# -*- coding:utf-8 -*-

import socket
import os
obj = socket.socket()
obj.connect(('127.0.0.1',9999,))

rec_bytes = obj.recv(1024)
rec_str = str(rec_bytes,encoding='utf-8')
print(rec_str)

# 发送当前文件大小
size = os.stat('file.jpg').st_size
obj.sendall(bytes(str(size),encoding='utf-8'))
# 与服务端交互，解决缓冲区粘包的问题
obj.recv(1024)
# 发送文件
with open('file.jpg','rb') as f:
    for line in f:
        obj.sendall(line)
obj.close()