一、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通讯原理:
socket函数
python 3.x下发送的内容必须是byte类型,2.x是字符串
二、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))
运行结果:
以上是一个最简单的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基本应用。如有不当之处欢迎指正。