eta_python/ths_wind.py

import hug
from bottle import route, run, NORUN
from WindPy import w
import iFinDPy as THS
import json
import pandas as pd
import time
import os
import logging
import sys

hug.API(__name__).http.output_format = hug.output_format.json

username = ""
password = ""


@hug.get('/hz_server')
def hello():
    return 1
    # return 'wind true'


# 同花顺登录函数
def thslogin():
    # 输入用户的帐号和密码
    thsLogin = THS.THS_iFinDLogin(username, password)
    print(thsLogin)
    if (thsLogin == 0 or thsLogin == -201):
        print('ths 登录成功')
        logger.info('ths login success')
    else:
        print('ths 登录失败')
        logger.info('ths login fail')


@hug.get('/edbInfo/ths')
def GetEdbDataByThs(EdbCode, StartDate, EndDate):
    print("GetEdbDataByThs")
    logger.info('GetEdbDataByThs')
    thsEDBDataQuery = THS.THS_EDBQuery(EdbCode, StartDate, EndDate, True)
    print(thsEDBDataQuery)
    logger.info(thsEDBDataQuery)

    print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + ":start GetEdbDataByThs")
    print("thsEDBDataQuery")
    print(thsEDBDataQuery)
    thsEDBDataQuery = thsEDBDataQuery.decode("unicode_escape")

    result = json.loads(thsEDBDataQuery)
    print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + ":end GetEdbDataByThs")

    if result["errorcode"] == -1010:
        thsLogin = THS.THS_iFinDLogin(username, password)
        print("登录一次")
        print(thsLogin)
        logger.info("ths login")
        if (thsLogin == 0 or thsLogin == -201):
            thsEDBDataQuery = THS.THS_EDBQuery(EdbCode, StartDate, EndDate, True)
            thsEDBDataQuery = thsEDBDataQuery.decode("unicode_escape")
            result = json.loads(thsEDBDataQuery)
            logger.info("login success")
    else:
        print("已登录")

    return result


def has_comma(input_string):
    return isinstance(input_string, str) and ',' in input_string


def process_single_edb_code(StockCode, SingleEdbCode, StartDate, EndDate):
    thsEDBDataQuery = THS.THS_DS(StockCode, SingleEdbCode, '', '', StartDate, EndDate)
    print(thsEDBDataQuery)
    result = json.dumps(ths_data_to_dict(thsEDBDataQuery))
    return result


@hug.get('/edbInfo/ths/ds')
def GetEdbDataByThsDs(StockCode, EdbCode, StartDate, EndDate):
    print("THS_DataSequence start")
    print(EdbCode)
    result_list = []

    if isinstance(EdbCode, list):
        for SingleEdbCode in EdbCode:
            result_list.append(process_single_edb_code(StockCode, SingleEdbCode, StartDate, EndDate))
            print("THS_DS end")
        print(result_list)
        return result_list
    else:
        # 如果不包含逗号，直接处理
        result = process_single_edb_code(StockCode, EdbCode, StartDate, EndDate)
        print(result)
        return result


@hug.get('/edbInfo/ths/future_good')
def GetFutureGoodEdbDataByThs(EdbCode, StartDate, EndDate):
    print("THS_HistoryQuotes start")
    thsLogin = THS.THS_iFinDLogin("hzmd150", "689719")

    filed = 'lastclose,open,high,low,close,avgprice,change,changeper,volume,amount,hsl,lastsettlement,settlement,zdsettlement,zdfsettlement,ccl,ccbd,zf,zjlx,zjcd'
    # filed = 'open,high,low,close,volume,amount,settlement,ccl,zjlx'
    params = 'Interval:D,CPS:1,baseDate:1900-01-01,Currency:YSHB,fill:Previous'
    # params = 'YSHB;Tradedays'
    # print(params)
    thsEDBDataQuery = THS.THS_HistoryQuotes(EdbCode, filed, params, StartDate, EndDate, True)
    # thsEDBDataQuery = THS.THS_HQ(EdbCode, filed, params, StartDate, EndDate)
    # thsEDBDataQuery = THS.THS_EDBQuery(EdbCode, StartDate, EndDate, True)
    print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + ":start GetFutureGoodEdbDataByThs")
    print("THS_HistoryQuotes end")
    print(thsEDBDataQuery)
    thsEDBDataQuery = thsEDBDataQuery.decode("unicode_escape")
    result = json.loads(thsEDBDataQuery)
    print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + ":end GetFutureGoodEdbDataByThs")
    return result


@hug.get('/edbInfo/ths/hf')
def GetEdbDataByThsDs(codes, indicators, function_para, start_time, end_time):
    print("codes: {}; indicators: {}; function_para: {}; start_time: {}; end_time: {}".format(codes, indicators,
                                                                                              function_para, start_time,
                                                                                              end_time))
    data_query = THS.THS_DF(codes, indicators, function_para, start_time, end_time)
    print(data_query)
    result = json.dumps(ths_data_to_dict(data_query))
    return result


@hug.get('/edbInfo/wind')
def GetEdbDataByWind(EdbCode, StartDate, EndDate):
    print("GetEdbDataByWind:", EdbCode)
    isConnected = w.isconnected()
    print("isconnected")
    print(isConnected)
    if isConnected == False:
        print("re isconnected")
        w.start()
        isConnected = w.isconnected()
        if isConnected == False:
            return "{'ErrMsg':'启动Wind接口失败'}"

    print("getdata")
    option = "Fill=Previous"
    data = w.edb(EdbCode, StartDate, EndDate, option)
    print("wind data")
    df = pd.DataFrame()
    if data.ErrorCode == -40521010:  # Internet Timeout 超时退出
        os._exit(0)
        return "a"
    df['DT'] = data.Times
    df['CLOSE'] = data.Data[0]
    df['ErrorCode'] = data.ErrorCode
    df = df[['DT', 'CLOSE', 'ErrorCode']]
    df = df.dropna()
    json_data = df.to_json()
    # w.stop()
    print("wind data end")
    result = json.loads(json_data)
    return result


@hug.get('/edbInfo/wind/future_good')
def GetFutureGoodEdbDataByWind(FutureGoodEdbCode, StartDate, EndDate):
    print("GetFutureGoodEdbDataByWind:", FutureGoodEdbCode)
    isConnected = w.isconnected()
    print("isconnected")
    print(isConnected)
    if isConnected == False:
        print("re isconnected")
        w.start()
        isConnected = w.isconnected()
        if isConnected == False:
            return "{'ErrMsg':'启动Wind接口失败'}"

    print("getdata")
    option = "Fill=Previous"
    data = w.edb(FutureGoodEdbCode, StartDate, EndDate, option)
    data = w.wsd(FutureGoodEdbCode, "trade_code,open,high,low,close,volume,amt,oi,settle", StartDate, EndDate, option)
    print("wind data")
    print(data)
    df = pd.DataFrame()

    if data.ErrorCode == -40521010:  # Internet Timeout 超时退出
        os._exit(0)
        return "a"

    df['DT'] = data.Times
    df['TRADE_CODE'] = data.Data[0]
    df['OPEN'] = data.Data[1]
    df['HIGH'] = data.Data[2]
    df['LOW'] = data.Data[3]
    df['CLOSE'] = data.Data[4]
    df['VOLUME'] = data.Data[5]
    df['AMT'] = data.Data[6]
    df['OI'] = data.Data[7]
    df['SETTLE'] = data.Data[8]
    df['ErrorCode'] = data.ErrorCode
    df = df[['DT', 'TRADE_CODE', 'OPEN', 'HIGH', 'LOW', 'CLOSE', 'VOLUME', 'AMT', 'OI', 'SETTLE', 'ErrorCode']]
    df = df.dropna()
    json_data = df.to_json()
    # w.stop()
    print("wind data end")
    result = json.loads(json_data)
    return result


@hug.get('/edbInfo/wind/wsd')
def GetEdbDataWindWsd(StockCode, EdbCode, StartDate, EndDate):
    print("GetEdbDataByWind:", EdbCode)
    isConnected = w.isconnected()
    print("isconnected")
    print(isConnected)
    if isConnected == False:
        print("re isconnected")
        w.start()
        isConnected = w.isconnected()
        if isConnected == False:
            return "{'ErrMsg':'启动Wind接口失败'}"

    option = "Fill=Previous"
    wsd_data = w.wsd(StockCode, EdbCode, StartDate, EndDate, option)

    if wsd_data.ErrorCode == -40521010:  # Internet Timeout 超时退出
        os._exit(0)
        return "a"

    fm = pd.DataFrame(wsd_data.Data, index=wsd_data.Fields, columns=wsd_data.Times)
    json_data = fm.to_json()
    result = json.loads(json_data)
    return result


if __name__ == "__main__":
    # wind 登录
    wStart = w.start()
    if wStart.ErrorCode != 0:
        print("启动万得API接口失败")
    print(wStart)

    # ths登录函数
    thslogin()

    app = __hug__.http.server()
    run(app=app, reloader=True, port=7000)