python--基于socket网络编程
Python 提供了两个基本的 socket 模块。
第一个是 Socket,它提供了标准的 BSD Sockets API。
第二个是 SocketServer, 它提供了服务器中心类,可以简化网络服务器的开发。
1、Scoket类型
套接字格式:
socket(family,type[,protocal]) 使用给定的地址族、套接字类型、协议编号(默认为0)来创建套接字。
1.1、socket.AF_UNIX #只能够用于单一的Unix系统进程间通信
1.2、socket.AF_INET #服务器之间网络通信
1.3、socket.AF_INET6 #IPv6
1.4、socket.SOCK_STREAM #流式socket , for TCP
1.5、socket.SOCK_DGRAM #数据报式socket , for UDP
1.6、socket.SOCK_RAW #原始套接字,普通的套接字无法处理ICMP、IGMP等网络报文,而SOCK_RAW可以;其次,SOCK_RAW也可以处理特殊的IPv4报文;此外,利用原始套接字,可以通过IP_HDRINCL套接字选项由用户构造IP头。
1.7、socket.SOCK_SEQPACKET #可靠的连续数据包服务
1.8、创建TCP Socket: s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
1.9、创建UDP Socket: s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
2、Socket 函数
注意点:
1)TCP发送数据时,已建立好TCP连接,所以不需要指定地址。UDP是面向无连接的,每次发送要指定是发给谁。
2)服务端与客户端不能直接发送列表,元组,字典。需要字符串化repr(data)。
2.1、服务端socket函数
s.bind(address) #将套接字绑定到地址, 在AF_INET下,以元组(host,port)的形式表示地址.
s.listen(backlog) #开始监听TCP传入连接。backlog指定在拒绝连接之前,操作系统可以挂起的最大连接数量。该值至少为1,大部分应用程序设为5就可以了。
s.accept() #接受TCP连接并返回(conn,address),其中conn是新的套接字对象,可以用来接收和发送数据。address是连接客户端的地址。
2.2、客户端socket函数
s.connect(address) #连接到address处的套接字。一般address的格式为元组(hostname,port),如果连接出错,返回socket.error错误。
s.connect_ex(adddress) #功能与connect(address)相同,但是成功返回0,失败返回errno的值。
2.3、公共socket函数
s.recv(bufsize[,flag]) #接受TCP套接字的数据。数据以字符串形式返回,bufsize指定要接收的最大数据量。flag提供有关消息的其他信息,通常可以忽略。
s.send(string[,flag]) #发送TCP数据。将string中的数据发送到连接的套接字。返回值是要发送的字节数量,该数量可能小于string的字节大小。
s.sendall(string[,flag]) #完整发送TCP数据。将string中的数据发送到连接的套接字,但在返回之前会尝试发送所有数据。成功返回None,失败则抛出异常。
s.recvfrom(bufsize[.flag]) #接受UDP套接字的数据。与recv()类似,但返回值是(data,address)。其中data是包含接收数据的字符串,address是发送数据的套接字地址。
s.sendto(string[,flag],address) #发送UDP数据。将数据发送到套接字,address是形式为(ipaddr,port)的元组,指定远程地址。返回值是发送的字节数。
s.close() #关闭套接字。
s.getpeername() #返回连接套接字的远程地址。返回值通常是元组(ipaddr,port)。
s.getsockname() #返回套接字自己的地址。通常是一个元组(ipaddr,port)
s.setsockopt(level,optname,value) #设置给定套接字选项的值。
s.getsockopt(level,optname[.buflen]) #返回套接字选项的值。
s.settimeout(timeout) #设置套接字操作的超时期,timeout是一个浮点数,单位是秒。值为None表示没有超时期。一般,超时期应该在刚创建套接字时设置,因为它们可能用于连接的操作(如connect())
s.gettimeout() #返回当前超时期的值,单位是秒,如果没有设置超时期,则返回None。
s.fileno() #返回套接字的文件描述符。
s.setblocking(flag) #如果flag为0,则将套接字设为非阻塞模式,否则将套接字设为阻塞模式(默认值)。非阻塞模式下,如果调用recv()没有发现任何数据,或send()调用无法立即发送数据,那么将引起socket.error异常。
s.makefile() #创建一个与该套接字相关连的文件
3、socket编程思路
3.1、TCP服务端:
1 创建套接字,绑定套接字到本地IP与端口
# socket.socket(socket.AF_INET,socket.SOCK_STREAM) , s.bind()
2 开始监听连接 #s.listen()
3 进入循环,不断接受客户端的连接请求 #s.accept()
4 然后接收传来的数据,并发送给对方数据 #s.recv() , s.sendall()
5 传输完毕后,关闭套接字 #s.close()
3.2、TCP客户端:
1 创建套接字,连接远端地址
# socket.socket(socket.AF_INET,socket.SOCK_STREAM) , s.connect()
2 连接后发送数据和接收数据 # s.sendall(), s.recv()
3 传输完毕后,关闭套接字 #s.close()
4、模拟聊天机器人实例
4.1、socket_server.py 服务端
#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
Socket服务器端
"""
import socket
sk = socket.socket()
sk.bind(('127.0.0.1', 8080,))
sk.listen(5)
#接收客户端请求
#连接,客户端地址信息
while True:
conn, address = sk.accept() #accept()阻塞
conn.sendall(bytes('欢迎致电10086', encoding='utf-8'))
while True:
ret_byte = conn.recv(1024)
ret_str = str(ret_byte, encoding='utf-8')
if ret_str == 'q':
break
conn.sendall(bytes('工号10086回复您:' + ret_str, encoding='utf-8'))
4.2、socket_client.py 客户端
#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
socket客户端
"""
import socket
ck = socket.socket()
ck.connect(('127.0.0.1', 8080))
ret_byte = ck.recv(1024)
ret_str = str(ret_byte, encoding='utf-8')
print(ret_str)
while True:
inp = input("请输入内容,按q退出 \n >>>")
if inp == 'q':
ck.sendall(bytes(inp, encoding='utf-8'))
break
else:
ck.sendall(bytes(inp, encoding='utf-8'))
ret_str = str(ck.recv(1024), encoding='utf-8')
print(ret_str)
ck.close()
5、基于socket实现文件上传实例
5.1、ftp_server.py
#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
模拟发送文件的服务器端
"""
import socket
sk = socket.socket()
sk.bind(('127.0.0.1', 8080,)) #绑定套接字到本地IP与端口
sk.listen(5) #开始监听连接
#接收客户端请求
#连接,客户端地址信息
while True:
conn, address = sk.accept() # accept()阻塞
conn.sendall(bytes('欢迎访问我的ftp', encoding='utf-8'))
#先接收文件大小,然后再接收文件
ret_size = str(conn.recv(1024), encoding='utf-8')
conn.sendall(bytes('start', encoding='utf-8')) #给客户端发送一个确认信息,通知客户端开始发送文件,解决数据粘包的问题。
total_size = int(ret_size) #文件总大小
has_recv = 0 #已经接收的文件大小
f = open('new.png', 'wb')
#接收文件内容,直到获取完成
while True:
if total_size == has_recv:
break
data = conn.recv(1024)
f.write(data)
has_recv += len(data)
f.close()
5.2、ftp_client.py
#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
模拟发送文件的客户端
"""
import socket
import os
ck = socket.socket()
ck.connect(('127.0.0.1', 8080))
ret_byte = ck.recv(1024) #接收服务器端返回的字节数据
ret_str = str(ret_byte, encoding='utf-8') #将字节数据转换成字符串
print(ret_str)
#首先向服务器发送文件的大小
file_size = os.stat('file1.png').st_size
ck.sendall(bytes(str(file_size), encoding='utf-8'))
ck.recv(1024) #接收一下服务器端的确认信息,再开始发送文件,解决数据粘包引起的系统异常问题
#发送文件
with open('file1.png', 'rb') as f:
for line in f:
ck.sendall(line)
ck.close()
6、socketserver模块实现并发操作
使用:
1、创建类,必须继承socketserver.BaseRequestHandler
2、定义handle方法
3、server_forever
socketserver模块实现并发操作实例:
#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
socketserver模块实现并发操作
"""
import socketserver
class MyServer(socketserver.BaseRequestHandler):
def handle(self):
#self.request, self.client_address, self.server
conn = self.request
conn.sendall(bytes('欢迎致电10086', encoding='utf-8'))
while True:
ret_byte = conn.recv(1024)
ret_str = str(ret_byte, encoding='utf-8')
if ret_str == 'q':
break
conn.sendall(bytes('工号10086回复您:' + ret_str, encoding='utf-8'))
if __name__ == '__main__':
server = socketserver.ThreadingTCPServer(('127.0.0.1', 8080), MyServer)
server.serve_forever()