#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import sys
from openpyxl.reader.excel import load_workbook
import os
import os.path

def gen_data(filename):
 wb = load_workbook('dataxlsx/' + filename + '.xlsx') # 加載文件所有分頁(yè)

 sheetnames = wb.get_sheet_names() # 獲取所有分頁(yè)的名字列表
 ws = wb.get_sheet_by_name(sheetnames[0]) # 取第一個(gè)分頁(yè)的數(shù)據(jù)
# print 'ws:', ws

# print "Work Sheet Titile:", ws.title # 分頁(yè)名稱(chēng)
# print "Work Sheet Rows:", ws.max_row # 分頁(yè)行數(shù)
# print "Work Sheet Cols:", ws.max_column # 分頁(yè)列數(shù)

 content = [] # 數(shù)據(jù)內(nèi)容
 id_list = [] # ID列表

# ========================start concat need data=================
 content.append('%% this file is auto maked!\n')
 content.append('-module(' + filename + ').\n')
 content.append('-compile(export_all).\n')

 for i in range(4, ws.max_row + 1):  # 從表格第三行開(kāi)始讀取，由于range函數(shù)不包含文件尾，所以為了讀到最后一行需+1
  for j in range(ws.max_column):
   if ws[i][j].value == None:
    content.append(' ,""')
   elif j == 0:
    id_list.append(int(ws[i][j].value))
    content.append('get(' + str(ws[i][j].value).strip() + ') ->\n')
    content.append(' {r_' + filename + ', ' + str(ws[i][j].value).strip())
   else:
    content.append(' ,' + str(ws[i][j].value).strip())
  content.append('};\n')

 content.append('get(_) ->\n')
 content.append(' not_match.\n')

 content.append('length() ->\n')
 content.append(' ' + str(ws.max_row - 1) + '.\n')
 content.append('id_list() ->\n ' + str(id_list) + '.')
# ==============================end===========================
 # 寫(xiě)入數(shù)據(jù)
 f = file('./server/' + filename + '.erl','w+')
 f.writelines(content)
 print 'create new file:', filename + '.erl'
 f.close() # 關(guān)閉通道
 return

def start_gen():
 # 刪除舊的數(shù)據(jù)
 delnames = os.listdir('./server')
 for delname in delnames:
  os.remove('./server/' + delname)
  print 'delete old file:', delname

 for _, _, filenames in os.walk('./dataxlsx'): # 遍歷文件夾
  for filename in filenames: # 遍歷文件
   find = filename.find('.xlsx') # 返回該文件名稱(chēng)長(zhǎng)度
#   print "find is:", find
   if filename[0] == '~' or find == -1: # 文件名以'~'開(kāi)頭或者找不到文件名， 如以'.'開(kāi)頭的文件
    continue
   else:
    split_list = filename.split('.') # 使用'.'分割文件名，獲得[文件名,文件格式]
#    print split_list
    gen_data(split_list[0]) # 用文件名作為參數(shù)調(diào)用gen_data

start_gen()