#coding:utf-8

import sys

import xlrd

import xlwt

import csv

#存储客户名称

list=[]

def readexcel():

    file=sys.argv[1]

    print file

    workbook=xlrd.open_workbook(file)

    sheet1=workbook.sheet_by_index(0)

    col=0

    while col<sheet1.ncols:

        if '客户名称' in sheet1.cell(0,col).value.encode('utf-8'):

            row=1

            while row<sheet1.nrows:

                tmp=sheet1.cell(row,col).value.encode('gbk')

                #去掉重复行

                if tmp not in list:

                    list.append(tmp)

                row+=1

        col+=1

    write_excel(list)

#匹配结果输出客户名称、处理后的地址

def write_excel(u_list):

    file=sys.argv[2]

    filename=sys.argv[1].replace('.xlsx','')+'_new.csv'

    workbook=xlrd.open_workbook(file)

    sheet1=workbook.sheet_by_index(0)

    col=0

    while col<sheet1.ncols:

        if '用户名称' == sheet1.cell(0,col).value.encode('utf-8'):

            row=1

            while row<sheet1.nrows:

                uname=sheet1.cell(row,col).value.encode('gbk')

                for u in u_list:

                    if u == uname:

                        print uname+'\t'+sheet1.cell(row,0).value.encode('gbk')

                        with open(filename,'ab') as fw:

                            csv_file=csv.writer(fw,dialect='excel')

                            new_row=[uname,sheet1.cell(row,0).value.encode('gbk')]

                            csv_file.writerow(new_row)

                row+=1

        col+=1

if __name__=='__main__':

    readexcel()