且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

python中socket与socketserver模块简单入门使用

更新时间:2022-08-22 09:50:38

一、socket模块
socket又叫套接字,是网络编程中的一个基本组件,是两个端点的程序之间的“信息通道”程序可分布在不同的计算机上(通过网络连接),通过socket套接字相互发送信息。python中的大多数的网络编程都 隐藏了socket模块的基本细节。
python中通过socket模块完成网络编程的套接字实现,一个套接字就是socket模块中的socket类的一个实例。socket实例化需要三个参数分别是family(ipv4,ipv6,unix)其中默认是ipv4 "socket.AF_INET",第二个参数是流,默认是socket.SOC_STREAM表示tcp,或socket.SOCK_DGRAM,第三个参数是协议,默认是0,使用默认即可。因此实际上实例化出一个套接字,只需要二个参数。

socket通讯原理:
python中socket与socketserver模块简单入门使用

socket函数
python 3.x下发送的内容必须是byte类型,2.x是字符串
python中socket与socketserver模块简单入门使用
python中socket与socketserver模块简单入门使用

二、socket简单通讯
环境说明:ubuntu 16.04 python 3.5.2 模拟的服务器和客户端均在一台机器上(可以在不同机器上,只要保障网络连接)

先来看一个简的服务器与客户端的socket通讯
功能:服务器侦听等待连接,客户端连接后发送连接信息,之后客户端发送的任何信息,服务器端原样返回客户端。
代码如下:
1、srv.py

import socket
sk = socket.socket()
#print(sk)
HOST = '127.0.0.1'
PORT = 2222
BUFSIZ = 1024
ADDR = (HOST,PORT)

sk.bind(ADDR)
sk.listen(5)
print("waiting...")

while True:
    conn,addr = sk.accept()
    ip,port = addr
    print("Got connection from",addr)
    conn.send(bytes("connected from %s:%s." %addr,"utf-8"))
    while True:
        data = conn.recv(BUFSIZ)
        if not data:
            break                                     #直接回车退出本次连接
        print(data)
        conn.send(data)

conn.close()

cli.py

import socket
HOST = '127.0.0.1'
PORT = 2222
BUFSIZ = 1024
ADDR = (HOST,PORT)
sk = socket.socket()
sk.connect(ADDR)
print(sk.recv(BUFSIZ))
while True:
    inp = input(">>>")
        if not inp:                                       #忽略空格回车
            continue
    sk.send(bytes(inp,"utf-8"))
    print(sk.recv(BUFSIZ))

运行结果:
python中socket与socketserver模块简单入门使用
python中socket与socketserver模块简单入门使用

以上是一个最简单的socket网络通讯。以上可以发现双方通讯时一发一收,或一收一发,总是成对出现,但有一个缺点,就是同时只能给一个客户端提供服务(通讯)其他的客户端只能等待。后面通过socketserver来解决一服务器多客户端同时连接的情况。

三、socket上传文件
在完成基本的通讯后,我们来做一个文件上传的实例。
server端

#coding:utf-8
import socket
import os
sk = socket.socket()
#print(sk)
HOST = '127.0.0.1'            #指定ip
PORT = 2222                     #侦听端口
BUFSIZ = 1024                  #缓存大小
ADDR = (HOST,PORT)

sk.bind(ADDR)                   #套接字ip端 口进行绑定
sk.listen(5)                           #连接等待数,默认5
print("waiting....")
BASE_DIR = os.path.dirname(os.path.abspath(__file__))

while True:
    conn, addr = sk.accept()
    print(addr)
    while True:
        data = conn.recv(1024)
        cmd,filename,filesize = (str(data,"utf-8")).split("|")
        path = os.path.join(BASE_DIR,'yuan',filename)
        filesize = int(filesize)

        f = open(path,"ab")
        has_receive = 0
        while has_receive != filesize:
            data = conn.recv(1024)
            f.write(data)
            has_receive+=len(data)
        f.close()

sk.close()

client端

#coding:utf-8
import socket
import os

HOST = '127.0.0.1' 
PORT = 2222
BUFSIZ = 1024
ADDR = (HOST,PORT)
sk = socket.socket()
sk.connect(ADDR)

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
while True:
    inp = input(">>>").strip() #post|12.png
    cmd,path = inp.split("|")
    path = os.path.join(BASE_DIR,path)  ##完整文件路径xxx/12.png
    filename = os.path.basename(path)    ##获取文件名
    file_size = os.stat(path).st_size    #获取文件大小
    file_info = "post|%s|%s" %(filename,file_size)
    sk.sendall(bytes(file_info,"utf-8"))
    f = open(path,'rb')
    has_sent = 0
    while  has_sent != file_size:
        data = f.read(BUFSIZ)
        sk.sendall(data)
        has_sent+=len(data)
    f.close()
    print("上传成功~")
sk.close()

总结:socket通讯的关键是一发一收,无论是服务端先发还是客户端先发,必须是一端发送另一端接收,如果连续发送会出现粘包现象。

四、socketserver实例多并发通讯

server.py

#coding:utf-8
import socketserver    #能实现并发
#并发聊天
class MyServer(socketserver.BaseRequestHandler):
    print("服务端启动...")
    def handle(self):
        while True:
            conn = self.request
            print (self.client_address)
            while True:
                client_data=conn.recv(1024)
                print (str(client_data,"utf8"))
                print ("waiting...")
                server_response=input(">>>")
                conn.sendall(bytes(server_response,"utf8"))
            conn.close()

if __name__ == '__main__':
    server = socketserver.ThreadingTCPServer(('127.0.0.1',2223),MyServer)
    server.serve_forever()

client.py

#coding:utf-8
import socket

ip_port = ('127.0.0.1',2223)
sk = socket.socket()
sk.connect(ip_port)
print ("客户端启动:")
while True:
    inp = input('>>>')
    sk.sendall(bytes(inp,"utf-8"))
    if inp == 'exit':
        break
    server_response=sk.recv(1024)
    print (str(server_response,"utf-8"))
sk.close()

执行结果服务器端通过多线程完成并发多客户端“同时”聊天效果。

以上为socket socketserver基本应用。如有不当之处欢迎指正。










本文转自 dyc2005 51CTO博客,原文链接:http://blog.51cto.com/dyc2005/2044166,如需转载请自行联系原作者