socket 的使用

基于TCP协议的socket

tcp是基于链接的，必须先启动服务端，然后再启动客户端去链接服务端

server端

import socket
sk = socket.socket()
sk.bind(('127.0.0.1',8898))  #把地址绑定到套接字
sk.listen()          #监听链接
conn,addr = sk.accept() #接受客户端链接
ret = conn.recv(1024)  #接收客户端信息
print(ret)       #打印客户端信息
conn.send(b'hi')        #向客户端发送信息
conn.close()       #关闭客户端套接字
sk.close()        #关闭服务器套接字(可选)

client端

import socket
sk = socket.socket()           # 创建客户套接字
sk.connect(('127.0.0.1',8898))    # 尝试连接服务器
sk.send(b'hello!')
ret = sk.recv(1024)         # 对话(发送/接收)
print(ret)
sk.close()            # 关闭客户套接字

问题：有的同学在重启服务端时可能会遇到

解决方法：

#加入一条socket配置，重用ip和端口
import socket
from socket import SOL_SOCKET,SO_REUSEADDR
sk = socket.socket()
sk.setsockopt(SOL_SOCKET,SO_REUSEADDR,1) #就是它，在bind前加
sk.bind(('127.0.0.1',8898))  #把地址绑定到套接字
sk.listen()          #监听链接
conn,addr = sk.accept() #接受客户端链接
ret = conn.recv(1024)   #接收客户端信息
print(ret)              #打印客户端信息
conn.send(b'hi')        #向客户端发送信息
conn.close()       #关闭客户端套接字
sk.close()        #关闭服务器套接字(可选)

基于UDP协议的socket

udp是无链接的，启动服务之后可以直接接受消息，不需要提前建立链接

简单使用

server端

import socket
udp_sk = socket.socket(type=socket.SOCK_DGRAM)   #创建一个服务器的套接字
udp_sk.bind(('127.0.0.1',9000))        #绑定服务器套接字
msg,addr = udp_sk.recvfrom(1024)
print(msg)
udp_sk.sendto(b'hi',addr)                 # 对话(接收与发送)
udp_sk.close()                         # 关闭服务器套接字

client端

import socket
ip_port=('127.0.0.1',9000)
udp_sk=socket.socket(type=socket.SOCK_DGRAM)
udp_sk.sendto(b'hello',ip_port)
back_msg,addr=udp_sk.recvfrom(1024)
print(back_msg.decode('utf-8'),addr)

带颜色和名字的聊天程序

#格式：
　　设置颜色开始 ：\033[显示方式;前景色;背景色m
#说明：
前景色            背景色           颜色
---------------------------------------
30                40              黑色
31                41              红色
32                42              绿色
33                43              黃色
34                44              蓝色
35                45              紫红色
36                46              青蓝色
37                47              白色
显示方式           意义
-------------------------
0                终端默认设置
1                高亮显示
4                使用下划线
5                闪烁
7                反白显示
8                不可见

#例子：
\033[1;31;40m    <!--1-高亮显示 31-前景色红色  40-背景色黑色-->
\033[0m          <!--采用终端默认设置，即取消颜色设置-->

# udp协议的服务器端

# import socket
# dic = {'alex':'\033[0;31;42m','taibai':'\033[1;32;41m'}
# sk = socket.socket(type=socket.SOCK_DGRAM)   #udp协议
# sk.bind(('127.0.0.1',9000))
# while True:
#     msg,addr = sk.recvfrom(1024)
#     name,msg1 = msg.decode('utf-8').split(':')
#     color = dic.get(name.strip(),'')
#     print('%s%s\033[0m' % (color,msg.decode('utf-8')))
#     content = input('content:')
#     sk.sendto(content.encode('utf-8'),addr)
#
# sk.close()

# udp协议的客户端

# import socket
#
# sk = socket.socket(type=socket.SOCK_DGRAM)
# name = input('name:')
# while True:
#     content = input('content:')
#     sk.sendto(('%s:%s' % (name,content)).encode('utf-8'),('127.0.0.1',9000))
#     msg,addr = sk.recvfrom(1024)
#     print(msg.decode('utf-8'))
#
# sk.close()

时间服务器

# udp协议的服务器端

# import time
# import socket
#
# sk = socket.socket(type=socket.SOCK_DGRAM)
# sk.bind(('127.0.0.1',9000))
# while True:
#     msg,addr = sk.recvfrom(1024)
#     sk.sendto(time.strftime(msg.decode('utf-8')).encode('utf-8'),addr)
#
# sk.close()

# udp协议的客户端

# import time
# import socket
#
# sk = socket.socket(type=socket.SOCK_DGRAM)
# while True:
#     sk.sendto('%Y-%m-%d %H-%M-%S'.encode('utf-8'),('127.0.0.1',9000))
#     msg,addr = sk.recvfrom(1024)
#     print(msg.decode('utf-8'))
#     time.sleep(2)
#
# sk.close()

升级版udp协议

#   mysocket.py文件

from socket import *

class Mysocket(socket):
    def __init__(self,coding='utf-8'):
        self.coding = coding
        super().__init__(type=SOCK_DGRAM)
    def my_recv(self,num):
        msg,addr = self.recvfrom(num)
        return msg.decode(self.coding),addr
    def my_send(self,msg,addr):
        return self.sendto(msg.encode(self.coding),addr)

#   server.py文件

from mysocket import Mysocket

sk = Mysocket()
sk.bind(('127.0.0.1',9000))
while True:
    msg,addr = sk.my_recv(1024)
    print(msg,addr)
    content = input('content:')
    sk.my_send(content,addr)

sk.close()

#  client.py文件

from mysocket import Mysocket

sk = Mysocket()
while True:
    content = input('content:')
    sk.my_send(content,('127.0.0.1',9000))
    msg,addr = sk.my_recv(1024)
    print(msg,addr)

sk.close()

socket参数的详解

socket.socket(family=AF_INET,type=SOCK_STREAM,proto=0,fileno=None)

创建socket对象的参数说明：

family	地址系列应为AF_INET(默认值),AF_INET6,AF_UNIX,AF_CAN或AF_RDS。 （AF_UNIX 域实际上是使用本地 socket 文件来通信）
type	套接字类型应为SOCK_STREAM(默认值),SOCK_DGRAM,SOCK_RAW或其他SOCK_常量之一。 SOCK_STREAM 是基于TCP的，有保障的（即能保证数据正确传送到对方）面向连接的SOCKET，多用于资料传送。  SOCK_DGRAM 是基于UDP的，无保障的面向消息的socket，多用于在网络上发广播信息。
proto	协议号通常为零,可以省略,或者在地址族为AF_CAN的情况下,协议应为CAN_RAW或CAN_BCM之一。
fileno	如果指定了fileno,则其他参数将被忽略,导致带有指定文件描述符的套接字返回。 与socket.fromfd()不同,fileno将返回相同的套接字,而不是重复的。 这可能有助于使用socket.close()关闭一个独立的插座。