# -*- coding:utf-8 -*-

 import os

 import re

 p1=r"([0-9][0-9][AB])\.\w{3}$"

 p2=r"^.+\,(\d{4}).+"

 pattern1=re.compile(p1)

 pattern2=re.compile(p2)

 def get_dir(zz):

     listdir=[]

     for filename in os.listdir('./%s'%zz):

         listdir.append(filename)

         # print(filename)

         # print(type(filename))

     # print(listdir)

     return listdir

 def joint_b0(listdir,zz):

     if not os.path.exists('./new/'):

         os.makedirs('./new/')

     fw=open('./new/%s.csv'%(zz),'a')

     for i in listdir:

         # print(i)

         j=0

         matcher1=re.findall(pattern1,i)

         fr=open('./%s/%s'%(zz,i))

         for line in fr.readlines():

             try:

                 # print(len(line))

                 # if(len(line)==1):#判断，跳过第一行

                 #     continue

                 if(j==0):#跳过第一行

                     # print("xixi")

                     j=j+1

                     continue

                 else:

                     fw.write(matcher1[0]+'_%s'%j+','+line.strip('\n')+'\n')

                     j=j+1

             except:

                 pass

     fr.close()

     fw.close()    

 def joint_b1(listdir,zz):#单独拼接

     if not os.path.exists('./new/'):

         os.makedirs('./new/')

     fw=open('./new/%s.csv'%(zz),'a')

     for i in listdir:

         # print(i)

         j=1

         matcher1=re.findall(pattern1,i)

         fr=open('./%s/%s'%(zz,i))

         for line in fr.readlines():

             # print(len(line))

             if(len(line)==4):

                 continue

             else:

                 fw.write(line)

                 # pass

     fr.close()

     fw.close()

 def sort_joint():

     # dir_list=['b0'],'b3','b4','b2_idsd','b3_idcd'

     dir_list=['b1']#用于编号和拼接，会在new目录下生成编号号码的文件

     for i in range(len(dir_list)):

         listdir=get_dir(dir_list[i])

         # print(listdir)

         joint_b0(listdir,dir_list[i])

 def joint_only():

     dir_list=['stock2']#用于拼接，将需要拼接的放到stock目录下，会在new目录下生成stock文件，然后根据需要修改名称

     for i in range(len(dir_list)):

         listdir=get_dir(dir_list[i])

         # print(listdir)

         joint_b1(listdir,dir_list[i])

 def updata(zz):

     fr=open('./new/%s'%(zz))

     fw=open('./new/new.csv','a')

     for line in fr.readlines():

         matcher2=re.findall(pattern2,line)

         if(matcher2):

             fw.write(matcher2[0]+','+line)

 if __name__=="__main__":

     sort_joint()

     # joint_only()

     # updata('b4.csv')