[python]沪深龙虎榜数据导入通达信的自选板块，并标注于K线图上

将沪深龙虎榜数据导入通达信的自选板块，并标注于K线图上

原理：python读取前一次处理完的计算5日后涨跌幅输出的csv文件

文件名前加"[paint]"

安照通达信的画图文件和板块文件格式，输出文件

用通达信的导入功能，导入画图文件和板块文件即可

事前数据截图：

处理后所得文件：

导入通达信后板块截图：

k线截图：

代码：

 #coding=utf-8

 #读取'[paint]'开头的csv文件

 #copyright @ WangXinsheng

 #http://www.cnblogs.com/wangxinsheng/

 import os

 import struct

 import time

 import datetime

 def getPriceByte(infile,arg,code,ss,day):

     dirc = {"-5":-3,"-3":-2,"-1":-1,"":1,"":2,"":3,"":4,"":5,"":6}

     first = 0

     second = 0

     a = float(arg)

     for i in [-5,-3,-1,1,3,5,7,9,11]:

         ib = i - 2

         if(a>=2**ib and a<2**i):

             #print(str(a)+":["+str(2**ib)+","+str(2**i)+"]"+"i:"+str(i)+" ib:"+str(ib))

             first = ib

             break;

     #print(first)

     chushu = 2**first

     if(arg >= chushu*2):

         second = arg / chushu

         #print(second)

         second = int(second*64)

     else:

         second = arg / chushu -1

         #print(second)

         second = int(second*128)

     if(first<0):

         first = 64 + dirc[str(first)]

     else:

         first = 63 + dirc[str(first)]

     '''

     print(first)

     print(second)

     print(struct.pack('B',first))

     print(struct.pack('B',second))

     '''

     #print(struct.pack('B',first))

     #print(struct.pack('B',second))

     infile.write(struct.pack('B',ss))

     infile.write(code.encode())

     infile.write(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')

     infile.write(b'\x03')

     infile.write('STANDK'.encode())

     infile.write(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')

     infile.write(b'\x04')

     infile.write(b'\x00')

     infile.write(b'\x1a\x10\x00\x00')

     infile.write(struct.pack('i',int(day)))

     infile.write(b'\x00\x00')

     infile.write(struct.pack('B',second))

     infile.write(struct.pack('B',first))

     infile.write(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xdcz3\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')

     print(str(code)+':已在 '+day+"日，价格："+str(arg)+"加入标记")

 #infile = open("tdxline.eld","rw")

 #getPriceByte(5.55)

 file = open("WXSline.eld","wb")

 zxg = ''

 path=r'.'

 files = os.listdir(path)

 files.sort()

 j=0

 for f in files:

     if(f.startswith('[paint]') and f.endswith('.csv')):

         print('读取文件：'+path+'/'+f)

         f=open(path+'/'+f,'rt')

         infos = f.readlines()

         f.close()

         i=0

         for info in infos:

             if(i==0):

                 i=i+1

                 continue

             tmp = info.split(',')

             if(len(tmp)<2):

                 continue

             code = str(tmp[1].replace('"','').replace("'",""))

             if(code.startswith('')):

                 ss = 1

             else:

                 ss = 0

             addzxg = ''

             try:

                 zxg.index(str(ss)+code)

             except:

                 addzxg = str(ss)+code+"\n"

             if(addzxg!=''):

                 zxg = zxg + str(ss)+code+"\n"

             if(len(tmp)<10):

                 continue

             days = tmp[0].split('-')

             if(len(days)<2):

                 days = tmp[0].split('/')

             if(len(days)<2):

                 continue

             if(len(days[1])<2):

                 days[1] = ''+days[1]

             if(len(days[2])<2):

                 days[2] = ''+days[2]

             day = days[0]+days[1]+days[2]

             #print(day)

             priceArr = tmp[9].replace('"','').split('%')

             if(len(priceArr)<2):

                 continue

             a = float(priceArr[1].replace('[','').replace(']',''))*(1-float(priceArr[0].replace('%',''))/100)

             a = round(a,2)

             code = str(tmp[1].replace('"','').replace("'",""))

             if(code.startswith('')):

                 ss = 1

             else:

                 ss = 0

             getPriceByte(file,a,code,ss,day)

             '''

             if(j>3):

                 break

             else:

                 j=j+1

             '''

 fileA = open("WXS条件股.ebk","w")

 fileA.write(zxg)

 fileA.close()

 file.close()

 print('处理完了\n画图文件地址：./WXSline.eld')

 print('\n板块文件地址：./WXS条件股.ebk')