113 lines
2.5 KiB
Python
113 lines
2.5 KiB
Python
import yaml
|
||
from openpyxl import Workbook
|
||
from openpyxl import load_workbook
|
||
from openpyxl.worksheet.datavalidation import DataValidation
|
||
from appPublic.dictObject import DictObject
|
||
from appPublic.uniqueID import getID
|
||
from ahserver.filestorage import FileStorage
|
||
|
||
def create_colnames():
|
||
r = {}
|
||
fls = ['', 'A', 'B']
|
||
i = 1
|
||
for fl in fls:
|
||
for c in "ABCDEFGHIJKLMNOPQRSTUVWXYZ":
|
||
r[f'{fl}{c}'] = i
|
||
i = i + 1
|
||
return r
|
||
|
||
colnames = create_colnames()
|
||
reverse_colnames = {v:k for k,v in colnames.items()}
|
||
|
||
def write_pattern_xlsx(fields: dict) -> str:
|
||
wb = Workbook()
|
||
ws_data = wb.active
|
||
hidden = wb.create_sheet('dict')
|
||
hidden.sheet_status = 'hidden'
|
||
# 3. 写入表头
|
||
headers = []
|
||
c = 1
|
||
opt_id = 0
|
||
for f in fields.values():
|
||
ws_data.cell(row=1, column=c, value=f.label or f.name)
|
||
if f.options:
|
||
create_options(ws_data, c, hidden, opt_id, f.options)
|
||
opt_id = opt_id + 1
|
||
c += 1
|
||
fs = FileStorage()
|
||
fp = fs._name2path(f'pricing_{getID}.xlsx')
|
||
wb.save(fp)
|
||
webpath = fs.webpath(fp)
|
||
return webpath
|
||
|
||
def create_options(ws, dcol, sheet, opt_id, options):
|
||
r = 2
|
||
vc = opt_id + 1
|
||
for v in options:
|
||
sheet.cell(row=r, column=vc, value=v)
|
||
r = r + 1
|
||
colname = reverse_colnames[vc]
|
||
dv = DataValidation(type="list",
|
||
formula1=f'dict!${colname}$1:${colname}$4',
|
||
allow_blank=True)
|
||
dv.error = '请选择下拉菜单中的有效职位。'
|
||
dv.errorTitle = '输入无效'
|
||
ws.add_data_validation(dv)
|
||
cn = reverse_colnames[dcol]
|
||
dv.add(f'{cn}2:{cn}1048576')
|
||
|
||
def load_xlsx_pricing(xlsxpath):
|
||
wb = load_workbook(xlsxpath)
|
||
ws = wb.active
|
||
d = []
|
||
for row in ws.iter_rows(min_row=2, values_only=True):
|
||
r = {}
|
||
for c, cell in enumerate(row):
|
||
r.update({ws.cell(row=1, column=c+1).value:cell})
|
||
d.append(r)
|
||
print(d)
|
||
return d
|
||
|
||
if __name__ == '__main__':
|
||
fields_str = """model:
|
||
type: str
|
||
label: "模型"
|
||
options:
|
||
- "viduq3-pro"
|
||
- "viduq3-turbo"
|
||
|
||
resolution:
|
||
type: str
|
||
label: "分辨率"
|
||
options:
|
||
- "1024p"
|
||
- "720p"
|
||
- "540p"
|
||
|
||
duration:
|
||
type: int # 建议改为 int 或 str，times 不是标准类型
|
||
label: "时长"
|
||
# 如果时长也有选项，需要补全，例如:
|
||
# options:
|
||
# - 5
|
||
# - 10
|
||
|
||
off_peak:
|
||
type: int # 建议改为 int 或 str，因为值是 0 和 1
|
||
label: "错峰执行"
|
||
options:
|
||
- off_peak # 正常时段
|
||
- normal # 错峰
|
||
price:
|
||
type: float
|
||
label: 单价
|
||
|
||
"""
|
||
"""
|
||
fields = DictObject(**yaml.safe_load(fields_str))
|
||
print(f'{fields=}, {type(fields)=}')
|
||
write_pattern_xlsx(fields)
|
||
"""
|
||
load_xlsx_pricing('./test.xlsx')
|
||
|