一.scoket基本介绍1.scoket简介(以下是来自chatgpt回答)
1)Socket(套接字)是计算机网络中用于描述主机之间通信的一种机制。它定义了一种标准的接口,
使得应用程序可以利用网络传输层提供的服务(如TCP或UDP)进行通信。
2)Socket的作用是在网络应用程序之间提供数据传输服务。通过Socket,应用程序可以将数据发送
到网络上的另一个应用程序,并从网络上的另一个应用程序接收数据。
3)Socket还提供了一种机制,使得应用程序可以接收来自网络上的请求,并对这些请求进行响应。
常见的网络应用程序,如Web服务器、邮件服务器、FTP服务器等都使用Socket技术。此外,许多P2P
(点对点)应用程序,如BitTorrent、eMule等也使用Socket技术。Socket还可以用于实现各种网络应用
程序,如网络游戏、聊天程序等。
2.建立TCP/IP连接
1)创建一个scoket对象
socket_stream = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
第一个参数表示通信协议类型
AF_INET:适用于网络通信
AF_UNIX:适用于本地进程间通信
第二个参数表示创建什么连接
SOCK_STREAM:表示tcp/ip连线
SOCK_DGRAM:表示udp连接
2)与服务端建立连接
# 建立连接ip_port = ("ip", "port") # 连接的服务端ip和端口socket_stream.connect(ip_port) # 建立连接,出错不返回错误码socket_stream.connect_ex(ip_port) # 出错时返回错误码,不抛异常
connect()和connect_ex()选一个即可
3)发送数据
# 发送数据data = 'hello'socket_stream.send(data.encode("utf-8")) # 发送TCP数据,当send在待发送数据量大于己端缓存区剩余空间时,数据丢失,不会发完# orsocket_stream.sendall(data.encode("utf-8")) # 发送完整的TCP数据(循环调用send,sendall在待发送数据量大于己端缓存区剩余空间时,数据不丢失,循环调用send直到发完)
当发送的数据较大时建议使用sendall发送
4)接收TCP数据
# 接收TCP数据。1024(变量)表示每次最多接受1M字节的数据,recv()函数是一个阻塞函数,没有要接收的数据时,会一直等待,直到接收到数据或出现错误才会返回response_data = socket_stream.recv(1024)
5)关闭连接
# 关闭连接socket_stream.close()
3.建立UDP连接
1)创建scoket对象
client_dgram = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
2)发送数据
# 发送数据ip_port = ('127.0.0.1', 8080)data = 'Hello'client_dgram.sendto(data.encode("utf-8"), ip_port)
这里发送用的是sendto()方法
3)接收数据
# 接收UDP数据数据data = client_dgram.recvfrom(4096)
4)关闭连接
# 关闭连接client_dgram.close()
二.使用scoket编写聊天程序
以下部分代码参考:https://www.yuque.com/imhelloworld/nov9az/bffcea259d3c96fb17a130acebc12801
聊天程序的话涉及两个端,既server和click。这里直接贴代码了,相关注释都是代码里
1)server
import socketimport threading# 服务器IP地址和端口号SERVER_IP = 'IP地址'SERVER_PORT = 端口号# 创建一个socket对象server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)# 绑定IP地址和端口号server_socket.bind((SERVER_IP, SERVER_PORT))# 设置最大连接数,超过10后拒绝。0的话表示不接受连接,直接拒绝server_socket.listen(10)# 客户端列表clients = []def handle_client(client_socket): while True: try: # 接收客户端发送的消息 message = client_socket.recv(10240).decode('utf-8') print("打印", message) if message == 'quert': client_socket.send(message.encode('utf-8')) else: # 广播消息给所有客户端 for c in clients: if c != client_socket: c.send(message.encode('utf-8')) except: # 发生异常时关闭连接 client_socket.close() clients.remove(client_socket) print("发送异常了") break# 循环监听客户端连接while True: print('Waiting for client connections...') # 接受客户端连接请求,新的客户端请求时创建一个新的socket,用于处理客户端的请求,原有的socket继续监听其他客户端的连接请求; # 函数是一个阻塞函数,当没有客户端连接请求时,会一直等待,直到有客户端连接请求到达 client_socket, client_address = server_socket.accept() print('客户socket', client_socket, 'ip和端口', client_address) clients.append(client_socket) # 创建一个新的线程处理客户端连接 client_thread = threading.Thread(target=handle_client, args=(client_socket, )) client_thread.start()
2)click
import socketimport threadingimport time# 服务器IP地址和端口号SERVER_IP = 'IP地址'SERVER_PORT = 端口号# 创建一个socket对象client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)# 连接服务器client_socket.connect((SERVER_IP, SERVER_PORT))def receive(): while True: try: # 接收服务器发送的消息 message = client_socket.recv(1024).decode('utf-8') if message == 'quert': client_socket.close() break print('\n对方回答:', message) except: # 发生异常时关闭连接 client_socket.close() breakdef send(): while True: # 获取用户输入的消息 message = input() if message == 'quert': client_socket.close() print('连接断开,通信结束') break else: # 发送消息给服务器 client_socket.send(message.encode('utf-8'))# 创建两个线程,分别用于接收和发送消息receive_thread = threading.Thread(target=receive)send_thread = threading.Thread(target=send)# 启动线程receive_thread.start()send_thread.start()
这里需要注意的是,既然是聊天程序,那么肯定涉及两个click,两个click代码一致即可,运行时先运行server,后运行click,之后就可以在Terminal下建立对话了。
以上是click在同一台电脑上,若不在同一台电脑的,建立对话需在同一个局域网下,ip地址在dos窗口下输入ipconfig\all即可查看,端口号可随便填个,只要不被占用就行。
三.scoket+chatgpt建立搜索对话框
1)获取chatgpt搜索接口,这里就直接贴代码了
def api(params, tokened="token"): url = "https://api.aigcfun.com/api/v1/text?key=" + tokened data = {"messages": [{"role": "system", "content": "请以markdown的形式返回答案"}, {"role": "user", "content": params}], "tokensLength": 32, "model": "gpt-3.5-turbo"} response_data = requests.post(url=url, json=data) return response_data.json()["choices"][0]["text"]
params为用户问的问题,return返回的是chatgpt回答的答案
2)获取到chatgpt的回答内容后返回给服务端
def receive(): while True: try: # 接收服务器发送的消息 message = client_socket.recv(1024).decode('utf-8') data_message = api(message) while data_message == None: # 由于chatgpt有时响应的比较慢,这里需添加个循环判断 continue # if data_message == '您今日使用次数已达上限,请明日再试!': # data_message = api(params=message, tokened=token()) client_socket.send(data_message.encode('utf-8')) # 发送chatgpt回答的问题给服务端 except: # 发生异常时关闭连接 client_socket.close() break
3)目前chatpgt虽是免费的,但每天都有次数限制,所以可以请求获取token的接口来获取新的token。这里就不做演示了(hhhhhh………)