socket模块

一：创建套接字(socket对象)

# 导入模块：
import socket
​
'''
创建套接字语法格式：
socket.socket([family[, type[, proto]]])
​
参数:
    family:    套接字家族可以使  AF_UNIX  或者  AF_INET
    type:      套接字类型可以根据是面向连接的还是非连接分为  SOCK_STREAM 或 SOCK_DGRAM
    protocol:  一般不填默认为0.
'''
​
#获取tcp/ip套接字
tcpSock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
​
#获取udp/ip套接字
udpSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

 

二：Socket 对象(内建)方法

       1>  服务器端：

s.bind()       #  绑定地址（host,port）到套接字， 在AF_INET下,以元组（host,port）的形式表示地址。
s.listen()     #  开始TCP监听。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了。
s.accept()     #  被动接受TCP客户端连接,(阻塞式)等待连接的到来

 

       2>  客户端：

s.connect()       #  主动初始化TCP服务器连接，。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误。
s.connect_ex()    #  connect()函数的扩展版本,出错时返回出错码,而不是抛出异常

 

 

       3>  公共用途的套接字函数：

s.recv()          # 接收TCP数据，数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略。
s.send()          # 发送TCP数据，将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。
s.sendall()       # 完整发送TCP数据，完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。
s.recvfrom()      # 接收UDP数据，与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。
s.sendto()        # 发送UDP数据，将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。
s.getpeername()   # 返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。
s.getsockname()   # 返回套接字自己的地址。通常是一个元组(ipaddr,port)
s.setsockopt(level,optname,value)       # 设置给定套接字选项的值。
s.getsockopt(level,optname[.buflen])    # 返回套接字选项的值。
s.close()         # 关闭套接字

 

 

       4>  面向锁的套接字方法:

# 设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）
s.settimeout(timeout) 
​
# 返回当前超时期的值，单位是秒，如果没有设置超时期，则返回None。                        
s.gettimeout()
​
# 如果flag为0，则将套接字设为非阻塞模式，否则将套接字设为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引起socket.error异常。
s.setblocking(flag)

 

 

       5> 面向文件的套接字方法:

 

s.fileno()      # 返回套接字的文件描述符。
s.makefile()    # 创建一个与该套接字相关连的文件

 

 

 

三：简单的TCP实例：

 server.py：

import socket
sk = socket.socket()               # 创建socket对象
sk.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)  # 只在测试的时候添加,到正式的生产环境中应该去掉
sk.bind(('192.168.11.229',9000))   # 绑定
sk.listen()                        # 监听
​
conn,addr = sk.accept()            # 建立连接 : 等待有一个客户端来
conn.send('喂'.encode('utf-8'))
msg = conn.recv(1024)              # 接收
print(msg.decode('utf-8'))         # 转码打印
​
conn.close()                       # 关闭客户端链接
sk.close()                         # 关闭socket对象

 

 

 

 client.py

import socket
​
sk = socket.socket()                   # 创建socket对象
sk.connect(('192.168.11.229',9000))    # 请求TCP服务器连接
​
msg = sk.recv(1024)                    # 接收server端数据
print(msg.decode('utf-8'))             # 转码打印
sk.send('你好'.encode('utf-8'))         # 发送数据
​
sk.close()                             # 关闭socket对象

 

 

 

四：升级 TCP 实例：服务端 循环接收多人的链接

 server.py

import time
import socket
​
sk = socket.socket()
sk.bind(('127.0.0.1',9000))
sk.listen()
​
while True:
    conn,addr = sk.accept()  
    while True:
        time.sleep(0.5)
        try:
            msg = conn.recv(1024)
            print(msg)
            conn.send(b'hello')
        except ConnectionAbortedError:
            conn.close()
            break
sk.close()

 

 

 

 client.py

import socket
sk = socket.socket()
sk.connect(('127.0.0.1',9000))
sk.send(b'leihoua')
msg = sk.recv(1024)
print(msg)
sk.close()

 

 

 

五：UDP协议简单使用

server

# server.py
import socket
sk = socket.socket(type = socket.SOCK_DGRAM)
addr = ('127.0.0.1',9000)
sk.bind(addr)
​
while True:
    msg,cli_addr = sk.recvfrom(1024)
    print(msg.decode('utf-8'))
    # message = input('>>>')
    sk.sendto('收到'.encode('utf-8'),cli_addr)
    
sk.close()

 

 

 client

# client.py
import socket
sk = socket.socket(type = socket.SOCK_DGRAM)
​
while True:
    message = input('>>>')
    sk.sendto(message.encode('utf-8'),('127.0.0.1',9000))
    msg,ser_addr = sk.recvfrom(1024)
    print(msg.decode('utf-8'))
​
sk.close()

 

 

 

六：使用UDP建一个简单的QQ聊天程序（待完善，仅参考）

# server.py 服务端代码

import json
import socket
​
sk = socket.socket(type = socket.SOCK_DGRAM)    # UDP
addr = ('127.0.0.1',9000)
sk.bind(addr)
dic = {}          # {qq:addr}
cache_dic = {}    # {to_qq : [msg,msg1]}
​
while True:
    content,addr = sk.recvfrom(1024)
    msg = content.decode('utf-8')
    msg_dic = json.loads(msg)
    if msg_dic['operate'] == 'online':
        qq = msg_dic['qq']
        dic[qq] = addr                # 记录了client端的qq号码和端口号
        if qq in cache_dic:
            for msg in cache_dic[qq]:
                sk.sendto(msg, addr)
    elif msg_dic['operate'] == 'send_msg':
        to_qq = msg_dic['to_qq']
        if dic.get(to_qq) :
            sk.sendto(msg_dic['msg'].encode('utf-8'),dic[to_qq])
        elif to_qq in cache_dic:
            cache_dic[to_qq].append(msg_dic['msg'].encode('utf-8'))
        else:
            cache_dic[to_qq] = [msg_dic['msg'].encode('utf-8')]

　# client.py 客户端代码

import json
import socket
​
sk = socket.socket(type = socket.SOCK_DGRAM)
addr = ('127.0.0.1',9000)
​
# 发送上线通知
info = {'operate':'online','qq':1}
json_info = json.dumps(info)
sk.sendto(json_info.encode('utf-8'),addr)  # 上线之后通知server
​
# 给好友发送消息
info = {'operate':'send_msg','to_qq':2,'msg':'你好'}
json_info = json.dumps(info)
sk.sendto(json_info.encode('utf-8'),addr)
​
# 接收好友消息
msg,addr = sk.recvfrom(1024)
print(msg.decode('utf-8'))