背景
为了提供一些背景信息,我正在尝试使用 Python 3.5 创建一个将工作表转换为 API 调用的工具。
为了将表格单元格转换为 API 调用所需的模式,我已经开始使用类似于 JavaScript 的语法来处理电子表格中使用的标题。例如:
工作表标题(字符串)
dict.list[0].id
Python字典
{
"dict":
"list": [
{"id": "my cell value"}
]
}
头部模式也有可能包含嵌套的数组/字典:
one.two[0].three[0].four.five[0].six
当我遍历每个标题时,我还需要在创建对象之后追加到该对象。
我尝试过的方法
add_branch
根据https://stackoverflow.com/a/47276490/2903486,使用像one.two.three.four
这样的值,我能够设置嵌套字典,并且我能够在遍历行时追加到现有字典,但是我无法添加数组支持:
def add_branch(tree, vector, value):
key = vector[0]
tree[key] = value \
if len(vector) == 1 \
else add_branch(tree[key] if key in tree else {},
vector[1:],
value)
return tree
file = Worksheet(filePath, sheet).readRow()
rowList = []
for row in file:
rowObj = {}
for colName, rowValue in row.items():
rowObj.update(add_branch(rowObj, colName.split("."), rowValue))
rowList.append(rowObj)
return rowList
我自己的add_branch版本
import re, json
def branch(tree, vector, value):
"""
Used to convert JS style notation (e.g dict.another.array[0].id) to a python object
Originally based on https://stackoverflow.com/a/47276490/2903486
"""
# Convert Boolean
if isinstance(value, str):
value = value.strip()
if value.lower() in ['true', 'false']:
value = True if value.lower() == "true" else False
# Convert JSON
try:
value = json.loads(value)
except:
pass
key = vector[0]
arr = re.search('\[([0-9]+)\]', key)
if arr:
arr = arr.group(0)
key = key.replace(arr, '')
arr = arr.replace('[', '').replace(']', '')
newArray = False
if key not in tree:
tree[key] = []
tree[key].append(value \
if len(vector) == 1 \
else branch({} if key in tree else {},
vector[1:],
value))
else:
isInArray = False
for x in tree[key]:
if x.get(vector[1:][0], False):
isInArray = x[vector[1:][0]]
if isInArray:
tree[key].append(value \
if len(vector) == 1 \
else branch({} if key in tree else {},
vector[1:],
value))
else:
tree[key].append(value \
if len(vector) == 1 \
else branch({} if key in tree else {},
vector[1:],
value))
if len(vector) == 1 and len(tree[key]) == 1:
tree[key] = value.split(",")
else:
tree[key] = value \
if len(vector) == 1 \
else branch(tree[key] if key in tree else {},
vector[1:],
value)
return tree
还需要帮助的地方
在添加一些东西之后,我的分支解决方案实际上效果很好,但我想知道我是否在这里做错了/笨拙,或者是否有更好的方法来处理我正在编辑嵌套数组的位置(我的尝试始于代码中的if IsInArray
部分)
我期望这两个标题可以编辑最后一个数组,但实际上我会在第一个数组上创建一个重复的字典:
file = [{
"one.array[0].dict.arrOne[0]": "1,2,3",
"one.array[0].dict.arrTwo[0]": "4,5,6"
}]
rowList = []
for row in file:
rowObj = {}
for colName, rowValue in row.items():
rowObj.update(add_branch(rowObj, colName.split("."), rowValue))
rowList.append(rowObj)
return rowList
输出:
[
{
"one": {
"array": [
{
"dict": {
"arrOne": [
"1",
"2",
"3"
]
}
},
{
"dict": {
"arrTwo": [
"4",
"5",
"6"
]
}
}
]
}
}
]
替代方法:
[
{
"one": {
"array": [
{
"dict": {
"arrOne": [
"1",
"2",
"3"
],
"arrTwo": [
"4",
"5",
"6"
]
}
}
]
}
}
]
list
看起来像[{'id': 1}, {'id': 2}]
,那么你如何知道你正在引用哪个id
? - benvclist[0].id
或者检查指示器是否为整数,例如list.1.id
-- 希望它能像最后一个那样,但还没有完全弄清楚)。 - Andrew Bowman