wehub-resource-sync
81 lines
1.7 KiB
Python
81 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import print_function, unicode_literals, absolute_import, division
|
|
import uuid
|
|
|
|
from domain_admin.utils import excel_util, csv_util, txt_util
|
|
|
|
# 文件读取配置
|
|
read_config = {
|
|
'xlsx': excel_util.read_excel,
|
|
'csv': csv_util.read_csv,
|
|
'txt': txt_util.read_txt,
|
|
}
|
|
|
|
# 文件写入配置
|
|
write_config = {
|
|
'xlsx': excel_util.write_excel,
|
|
'csv': csv_util.write_csv,
|
|
'txt': txt_util.write_txt,
|
|
}
|
|
|
|
|
|
def get_random_filename(ext):
|
|
"""
|
|
获取一个随机文件名
|
|
:param ext: str
|
|
:return:
|
|
"""
|
|
return str(uuid.uuid4()) + '.' + ext
|
|
|
|
|
|
def get_filename_ext(filename):
|
|
"""
|
|
获取文件名的扩展名
|
|
:param filename:
|
|
:return:
|
|
"""
|
|
return filename.split('.')[-1]
|
|
|
|
|
|
def read_data_from_file(filename):
|
|
file_type = get_filename_ext(filename)
|
|
|
|
if file_type in read_config:
|
|
return read_config[file_type](filename)
|
|
else:
|
|
raise Exception('not support .{}'.format(file_type))
|
|
|
|
|
|
def write_data_to_file(filename, rows):
|
|
file_type = get_filename_ext(filename)
|
|
|
|
if file_type in write_config:
|
|
return write_config[file_type](filename, rows)
|
|
else:
|
|
raise Exception('not support .{}'.format(file_type))
|
|
|
|
|
|
def convert_to_export(rows, field_mapping):
|
|
lst = []
|
|
for row in rows:
|
|
data = {}
|
|
for item in field_mapping:
|
|
data[item['name']] = row.get(item['field'], '')
|
|
|
|
lst.append(data)
|
|
|
|
return lst
|
|
|
|
|
|
def convert_to_import(rows, field_mapping):
|
|
lst = []
|
|
for row in rows:
|
|
data = {
|
|
item['field']: row.get(item['name'], item.get('default_value', ''))
|
|
for item in field_mapping
|
|
}
|
|
|
|
lst.append(data)
|
|
|
|
return lst
|