Python socket编程之四：模拟分时图

Posted 2020-07-16 佩雷尔曼的信徒

建立 socket，先运行服务器，再运行客户端，建立连接后服务器从本地数据库调数据一截一截地发送给客户端，客户端接受数据绘图模拟分时图

• 1.socket

# -*- coding: utf-8 -*-
"""socket 模块"""
import socket
import struct
import sqlalchemy
import pandas
########################################################################
class sckt:
    #----------------------------------------------------------------------
    def __init__(self, host = \'192.168.1.3\', port = 12345):
        self.host = host
        self.port = port
    #----------------------------------------------------------------------
    def run_server(self):
        sckt = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sckt.bind((self.host, self.port))
        sckt.listen(5)
        """连接库"""
        engine = sqlalchemy.create_engine(\'mssql+pyodbc://sa:123456@XiTongDSN\')
        """取开盘价"""
        Open = (pandas.read_sql(\'sh\', engine))[\'open\']
        i = 0
        while True:            
            connection, address = sckt.accept()
            if connection.recv(1024) == b\'Link\' and i < (len(Open) - 1):               
                """数据打包"""                
                connection.send(struct.pack(\'f\', Open[i]))
                i += 1
            else:
                connection.send(b\'Cut off the connection!\')
                connection.close()
    #----------------------------------------------------------------------
    def run_client(self, message = b\'Link\'):
        sckt = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sckt.connect((self.host, self.port))
        sckt.send(message)
        """数据解包，注意 unpack 要求四字节，其用法为后接 [:4]"""
        Open = struct.unpack(\'f\', sckt.recv(1024)[:4])
        sckt.close()
        """返回开盘价，去格式的用法后接 [0]"""
        return Open[0]

• 2.server

# -*- coding: utf-8 -*-
"""服务器端单独一个文件"""
import sys
sys.path.append(\'C:\\WinPython-32bit-3.5.1.3\\myWorkSpace1\')
import f1
F1 = f1.sckt()
F1.run_server()

• 3.client

# -*- coding: utf-8 -*-
"""客户端单独一个文件"""
import f1
F1 = f1.sckt()
import matplotlib.pylab as plt
opn = []
for i in range(0, 1000):
    opn.append(F1.run_client())
    plt.plot(opn)
    plt.grid(True, axis = \'both\')    
    """必须有 pause，不然会卡住"""
    plt.pause(0.1)

"""先运行服务器，在运行客户端"""

• 4.runtimeplot