从零开始用Python搭建Socket服务器(初篇)

本文假设你学过Python方面的知识

文章目录

前言

Socket

1.基本结构

2.高级应用

---

前言

如何加密自己的信息?如何基本安全的发送自己的信息?

Python的Socket库搭建的服务器可以完全胜任这种任务!

---

一、Socket简介

Socket其实就是Python其中关于网络套接字运用方面的库。

二、Socket

1.基本结构

服务端代码如下：

#coding:utf-8
def TCP_he(ip="",port=9999):
    import socket #导入库
    import threading

    server=socket.socket(socket.AF_INET,socket.SOCK_STREAM) #实例化Socket对象
    server.bind((ip,port)) #绑定Ip和端口
    server.listen(5) #开始监听
    print("[*] Listen on %s:%d"%(ip,port))
    def chuli(a):
        requst=a.recv(1024) #获取信息
        print("[*] Received: "+str(requst))
        a.send("I'm already") #往回发送字符串"I'm already"
        a.close() #关闭与客户端的链接
    while True:
        client,addr=server.accept() #如果有客户端链接
        print("[*]成功建立连接于 %s:%d" % (addr[0],addr[1]))
        handler=threading.Thread(target=chuli,args=(client,)) #开启线程处理连接
        handler.start()
try:
    TCP_he()
except:
    print("[*]Error!")

客户端代码如下(来源:堡塔插件：堡塔应用管理器 使用说明 - Linux面板 - 宝塔面板论坛 (bt.cn)https://www.bt.cn/bbs/thread-54473-1-1.html

#!/usr/bin/python
# coding: utf-8
import socket
socCli = socket.socket() #实例化Socket
try:
    socCli.connect(('IP', port)) #连接指定服务器 (需更改IP和port)

    while True:
        data = input("# 输入内容:") #正常输入
        socCli.send(data.encode("utf-8")) #向服务器发送转码后的信息
        reply = socCli.recv(1024) #接收回传信息
        if reply:
            print("接收到信息:")
            print(reply.decode("utf-8")) #解码后输出
except Exception as e:
    print(e)
finally:
    socCli.close() #关闭Socket连接

2.高级应用(来源:堡塔插件：堡塔应用管理器 使用说明 - Linux面板 - 宝塔面板论坛 (bt.cn)https://www.bt.cn/bbs/thread-54473-1-1.html

服务端代码(非堵塞服务器)：

#!/usr/bin/python
# coding: utf-8
from __future__ import print_function, absolute_import
import socket
import queue
import logging
from select import select
import sys

SERVER_IP = ('', port ) #设定监听的端口

# 保存客户端发送过来的消息,将消息放入队列中
message_queue = {}
input_list = []
output_list = []

if __name__ == "__main__":
    server = socket.socket()
    server.bind(SERVER_IP)
    server.listen(10)
    # 设置为非阻塞
    server.setblocking(False)

    # 初始化将服务端加入监听列表
    input_list.append(server)
    print("sys stdout:")
    print(sys.stdout.fileno())
    print(sys.stdout)

    while True:
        # 开始 select 监听,对input_list中的服务端server进行监听
        stdinput, stdoutput, stderr = select(input_list, output_list,
                                             input_list)

        # 循环判断是否有客户端连接进来,当有客户端连接进来时select将触发
        for obj in stdinput:
            # 判断当前触发的是不是服务端对象, 当触发的对象是服务端对象时,说明有新客户端连接进来了
            if obj == server:
                # 接收客户端的连接, 获取客户端对象和客户端地址信息
                conn, addr = server.accept()
                message = "Client {0} connected! ".format(addr)
                print(message)
                logging.warning(message)
                # 将客户端对象也加入到监听的列表中, 当客户端发送消息时 select 将触发
                input_list.append(conn)
                # 为连接的客户端单独创建一个消息队列，用来保存客户端发送的消息
                message_queue[conn] = queue.Queue()

            else:
                # 由于客户端连接进来时服务端接收客户端连接请求，将客户端加入到了监听列表中(input_list)，客户端发送消息将触发
                # 所以判断是否是客户端对象触发
                try:
                    recv_data = obj.recv(1024)
                    # 客户端未断开
                    if recv_data:
                        message = "received {0} from client {1}".format(
                            recv_data.decode(), addr)
                        logging.warning(message)
                        print(message)
                        # 将收到的消息放入到各客户端的消息队列中
                        message_queue[obj].put(recv_data)

                        # 将回复操作放到output列表中，让select监听
                        if obj not in output_list:
                            output_list.append(obj)

                except ConnectionResetError:
                    # 客户端断开连接了，将客户端的监听从input列表中移除
                    input_list.remove(obj)
                    # 移除客户端对象的消息队列
                    del message_queue[obj]
                    error_message = "\n[input] Client  {0} disconnected".format(addr)
                    logging.error(error_message)
                    print(error_message)

                    # 如果现在没有客户端请求,也没有客户端发送消息时，开始对发送消息列表进行处理，是否需要发送消息
        for sendobj in output_list:
            try:
                # 如果消息队列中有消息,从消息队列中获取要发送的消息
                if not message_queue[sendobj].empty():
                    # 从该客户端对象的消息队列中获取要发送的消息
                    send_data = message_queue[sendobj].get()
                    sendobj.sendall(send_data)
                else:
                    # 将监听移除等待下一次客户端发送消息
                    output_list.remove(sendobj)

            except ConnectionResetError:
                # 客户端连接断开了
                del message_queue[sendobj]
                output_list.remove(sendobj)
                print("\n[output] Client  {0} disconnected".format(addr))