#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import sys
import csv
'''
该模块的主要功能,是根据已有的csv文件,
通过readDataToDicl函数,把csv中对应的部分,
写入字典中,每个字典当当作一条json数据
'''
class GenExceptData(object):
def __init__(self):
try:
FileList = os.listdir("CSV")
for index, value in enumerate(FileList):
print(value)
dataDic = {} #用来存放json数据的字典
csvHand = open("CSV/%s" % value,"r") #打开csv文件,设置读的权限
if csvHand == None:
continue
readcsv = csv.reader(csvHand) #创建读取csv文件句柄
fieldnames = next(readcsv)
csv_reader = csv.DictReader(csvHand, fieldnames = fieldnames)
for row in csv_reader:
d = {}
for key, info in row.items():
op, name = key.split('$')
if op == "I":
d[name] = int(info)
elif op == "S":
d[name] = str(info)
elif op == "F":
d[name] = float(info)
dataDic.setdefault(int(d['ID']), d)
fileName, prefix = value.split('.')
fileObject = open("PyFile/%s.py" % fileName, "w")
fileObject.write("data = ")
fileObject.write(str(dataDic))
fileObject.close()
except Exception:
print("Read Excel error:")
def test():
GenExceptData()
if __name__=="__main__":
test()