[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')