O Market Collector foi atualizado novamente - suporta importação de arquivo no formato CSV e fornece fonte de dados personalizada

Recentemente, um usuário precisou usar seu próprio arquivo no formato CSV como fonte de dados para o sistema de backtesting da Plataforma de Negociação Quantitativa Inventor. O sistema de backtesting da Inventor Quantitative Trading Platform tem muitas funções e é simples e eficiente de usar. Contanto que você tenha os dados, você pode executar o backtesting, e você não está mais limitado às bolsas e produtos suportados pelo data center da plataforma .

Ideias de design

A ideia do design é, na verdade, muito simples. Precisamos apenas fazer uma pequena alteração no coletor de mercado anterior. Adicionamos um parâmetro ao coletor de mercado.isOnlySupportCSVUsado para controlar se deve usar apenas arquivos CSV como fontes de dados para o sistema de backtesting e adicionar outro parâmetrofilePathForCSV, usado para definir o caminho para o arquivo de dados CSV no servidor onde o robô coletor de mercado é executado. Finalmente, de acordo comisOnlySupportCSVO parâmetro está definido comoTruePara decidir qual fonte de dados usar (1. coletados por você, 2. dados em arquivo CSV), essa mudança ocorre principalmente emProviderAulado_GETfunção.

O que é um arquivo CSV?

Valores separados por vírgula (CSV, às vezes também chamados de valores separados por caracteres porque os caracteres delimitadores podem ser diferentes de vírgulas) são arquivos que armazenam dados tabulares (números e texto) em texto simples. Texto simples significa que o arquivo é uma sequência de caracteres e não contém dados que devem ser interpretados como números binários. Um arquivo CSV consiste em qualquer número de registros, separados por algum tipo de caractere de quebra de linha; cada registro consiste em campos, e o separador entre os campos são outros caracteres ou strings, os mais comuns dos quais são vírgulas ou tabulações. Normalmente, todos os registros têm exatamente a mesma sequência de campos. Geralmente são arquivos de texto simples. É recomendado usar o WORDPAD ou o Notepad para abri-lo. Outro método é salvá-lo como um novo arquivo e então abri-lo com o EXCEL.

Não há um padrão universal para o formato de arquivo CSV, mas existem certas regras, geralmente uma linha por registro, com a primeira linha sendo o cabeçalho. Os dados em cada linha são separados por vírgulas.

Por exemplo, o arquivo CSV que usamos para testes fica assim quando aberto com o Bloco de Notas:

Observe que a primeira linha do arquivo CSV é o cabeçalho da tabela.

,open,high,low,close,vol

Precisamos analisar e organizar esses dados e, então, construí-los no formato exigido pela fonte de dados personalizada do sistema de backtesting. Isso já foi tratado no código em nosso artigo anterior e requer apenas pequenas modificações.

Código modificado

import _thread
import pymongo
import json
import math
import csv
from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.parse import parse_qs, urlparse

def url2Dict(url):
    query = urlparse(url).query  
    params = parse_qs(query)  
    result = {key: params[key][0] for key in params}  
    return result

class Provider(BaseHTTPRequestHandler):
    def do_GET(self):
        global isOnlySupportCSV, filePathForCSV
        try:
            self.send_response(200)
            self.send_header("Content-type", "application/json")
            self.end_headers()

            dictParam = url2Dict(self.path)
            Log("自定义数据源服务接收到请求，self.path:", self.path, "query 参数：", dictParam)
            
            # 目前回测系统只能从列表中选择交易所名称，在添加自定义数据源时，设置为币安，即：Binance
            exName = exchange.GetName()                                     
            # 注意，period为底层K线周期
            tabName = "%s_%s" % ("records", int(int(dictParam["period"]) / 1000))  
            priceRatio = math.pow(10, int(dictParam["round"]))
            amountRatio = math.pow(10, int(dictParam["vround"]))
            fromTS = int(dictParam["from"]) * int(1000)
            toTS = int(dictParam["to"]) * int(1000)

            # 要求应答的数据
            data = {
                "schema" : ["time", "open", "high", "low", "close", "vol"],
                "data" : []
            }
            
            if isOnlySupportCSV:
                # 处理CSV读取,filePathForCSV路径
                listDataSequence = []
                with open(filePathForCSV, "r") as f:
                    reader = csv.reader(f)
                    # 获取表头
                    header = next(reader)
                    headerIsNoneCount = 0
                    if len(header) != len(data["schema"]):
                        Log("CSV文件格式有误，列数不同，请检查！", "#FF0000")
                        return 
                    for ele in header:
                        for i in range(len(data["schema"])):
                            if data["schema"][i] == ele or ele == "":
                                if ele == "":
                                    headerIsNoneCount += 1
                                if headerIsNoneCount > 1:
                                    Log("CSV文件格式有误，请检查！", "#FF0000")
                                    return 
                                listDataSequence.append(i)
                                break
                    
                    # 读取内容
                    while True:
                        record = next(reader, -1)
                        if record == -1:
                            break
                        index = 0
                        arr = [0, 0, 0, 0, 0, 0]
                        for ele in record:
                            arr[listDataSequence[index]] = int(ele) if listDataSequence[index] == 0 else (int(float(ele) * amountRatio) if listDataSequence[index] == 5 else int(float(ele) * priceRatio))
                            index += 1
                        data["data"].append(arr)
                
                Log("数据：", data, "响应回测系统请求。")
                self.wfile.write(json.dumps(data).encode())
                return 
            
            # 连接数据库
            Log("连接数据库服务，获取数据，数据库：", exName, "表：", tabName)
            myDBClient = pymongo.MongoClient("mongodb://localhost:27017")
            ex_DB = myDBClient[exName]
            exRecords = ex_DB[tabName]
            
            # 构造查询条件：大于某个值{'age': {'$gt': 20}} 小于某个值{'age': {'$lt': 20}}
            dbQuery = {"$and":[{'Time': {'$gt': fromTS}}, {'Time': {'$lt': toTS}}]}
            Log("查询条件：", dbQuery, "查询条数：", exRecords.find(dbQuery).count(), "数据库总条数：", exRecords.find().count())
            
            for x in exRecords.find(dbQuery).sort("Time"):
                # 需要根据请求参数round和vround，处理数据精度
                bar = [x["Time"], int(x["Open"] * priceRatio), int(x["High"] * priceRatio), int(x["Low"] * priceRatio), int(x["Close"] * priceRatio), int(x["Volume"] * amountRatio)]
                data["data"].append(bar)
            
            Log("数据：", data, "响应回测系统请求。")
            # 写入数据应答
            self.wfile.write(json.dumps(data).encode())
        except BaseException as e:
            Log("Provider do_GET error, e:", e)


def createServer(host):
    try:
        server = HTTPServer(host, Provider)
        Log("Starting server, listen at: %s:%s" % host)
        server.serve_forever()
    except BaseException as e:
        Log("createServer error, e:", e)
        raise Exception("stop")

def main():
    LogReset(1)
    if (isOnlySupportCSV):
        try:
        # _thread.start_new_thread(createServer, (("localhost", 9090), ))         # 本机测试
            _thread.start_new_thread(createServer, (("0.0.0.0", 9090), ))         # VPS服务器上测试
            Log("开启自定义数据源服务线程，数据由CSV文件提供。", "#FF0000")
        except BaseException as e:
            Log("启动自定义数据源服务失败！")
            Log("错误信息：", e)
            raise Exception("stop")
        while True:
            LogStatus(_D(), "只启动自定义数据源服务，不收集数据！")
            Sleep(2000)
    
    exName = exchange.GetName()
    period = exchange.GetPeriod()
    Log("收集", exName, "交易所的K线数据，", "K线周期:", period, "秒")
    
    # 连接数据库服务，服务地址 mongodb://127.0.0.1:27017 具体看服务器上安装的mongodb设置
    Log("连接托管者所在设备mongodb服务，mongodb://localhost:27017")
    myDBClient = pymongo.MongoClient("mongodb://localhost:27017")   
    # 创建数据库
    ex_DB = myDBClient[exName]
    
    # 打印目前数据库表
    collist = ex_DB.list_collection_names()
    Log("mongodb ", exName, " collist:", collist)
    
    # 检测是否删除表
    arrDropNames = json.loads(dropNames)
    if isinstance(arrDropNames, list):
        for i in range(len(arrDropNames)):
            dropName = arrDropNames[i]
            if isinstance(dropName, str):
                if not dropName in collist:
                    continue
                tab = ex_DB[dropName]
                Log("dropName:", dropName, "删除：", dropName)
                ret = tab.drop()
                collist = ex_DB.list_collection_names()
                if dropName in collist:
                    Log(dropName, "删除失败")
                else :
                    Log(dropName, "删除成功")
    
    # 开启一个线程，提供自定义数据源服务
    try:
        # _thread.start_new_thread(createServer, (("localhost", 9090), ))     # 本机测试
        _thread.start_new_thread(createServer, (("0.0.0.0", 9090), ))         # VPS服务器上测试
        Log("开启自定义数据源服务线程", "#FF0000")
    except BaseException as e:
        Log("启动自定义数据源服务失败！")
        Log("错误信息：", e)
        raise Exception("stop")
    
    # 创建records表
    ex_DB_Records = ex_DB["%s_%d" % ("records", period)]
    Log("开始收集", exName, "K线数据", "周期：", period, "打开（创建）数据库表：", "%s_%d" % ("records", period), "#FF0000")
    preBarTime = 0
    index = 1
    while True:
        r = _C(exchange.GetRecords)
        if len(r) < 2:
            Sleep(1000)
            continue
        if preBarTime == 0:
            # 首次写入所有BAR数据
            for i in range(len(r) - 1):
                bar = r[i]
                # 逐根写入，需要判断当前数据库表中是否已经有该条数据，基于时间戳检测，如果有该条数据，则跳过，没有则写入
                retQuery = ex_DB_Records.find({"Time": bar["Time"]})
                if retQuery.count() > 0:
                    continue
                
                # 写入bar到数据库表
                ex_DB_Records.insert_one({"High": bar["High"], "Low": bar["Low"], "Open": bar["Open"], "Close": bar["Close"], "Time": bar["Time"], "Volume": bar["Volume"]})                
                index += 1
            preBarTime = r[-1]["Time"]
        elif preBarTime != r[-1]["Time"]:
            bar = r[-2]
            # 写入数据前检测，数据是否已经存在，基于时间戳检测
            retQuery = ex_DB_Records.find({"Time": bar["Time"]})
            if retQuery.count() > 0:
                continue
            
            ex_DB_Records.insert_one({"High": bar["High"], "Low": bar["Low"], "Open": bar["Open"], "Close": bar["Close"], "Time": bar["Time"], "Volume": bar["Volume"]})
            index += 1
            preBarTime = r[-1]["Time"]
        LogStatus(_D(), "preBarTime:", preBarTime, "_D(preBarTime):", _D(preBarTime/1000), "index:", index)
        # 增加画图展示
        ext.PlotRecords(r, "%s_%d" % ("records", period))
        Sleep(10000)
        

Executando testes

Primeiro, iniciamos o robô coletor de mercado, adicionamos uma bolsa ao robô e deixamos o robô rodar. Configuração de parâmetros:

Então criamos uma estratégia de teste:

function main() {
    Log(exchange.GetRecords())
    Log(exchange.GetRecords())
    Log(exchange.GetRecords())
}

A estratégia é muito simples, bastando obter e imprimir três dados da linha K.

Na página de backtesting, defina a fonte de dados do sistema de backtesting como uma fonte de dados personalizada e preencha o endereço do servidor onde o robô coletor de mercado é executado. Já que os dados em nosso arquivo CSV são K-line de 1 minuto. Portanto, ao fazer o backtest, definimos o período da linha K como 1 minuto.

Clique em Iniciar Backtesting e o robô coletor de mercado receberá a solicitação de dados:

Após o sistema de backtesting concluir a execução da estratégia, ele gera um gráfico de K-line com base nos dados de K-line na fonte de dados.

Compare os dados no arquivo:

RecordsCollecter (atualizado para fornecer função de fonte de dados personalizada, suporte a arquivo de dados CSV para fornecer fonte de dados)

Este é apenas um ponto de partida, fique à vontade para deixar uma mensagem.