gowthaman/json_to_csv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

format file

Browse Source
This commit is contained in:
gowthaman 2023-06-20 09:01:16 +05:30
parent 43d6635be0
commit b7668d19b9
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -99,7 +99,7 @@ class DBConn:
clients = pd.read_sql(f"SELECT * FROM \"{tbl}\"", self.con)
clients.to_csv(args.output / f"{prefix}{tbl}.csv", index=False, sep=args.delimiter)
def extract_child(self, items_list, current_level, prev_step, existing, step):
def extract_child(self, items_list, current_level, prev_step, existing):
attributes = existing.get("attributes", [args.join_column])
flat_attributes = existing.get("flat_attributes", [])
@ -124,7 +124,7 @@ class DBConn:
existing_next_step = children.get(subKey, {})
children[subKey] = self.extract_child(subValue, subKey,
f"{prev_step}{current_level}_",
existing_next_step, step + 1)
existing_next_step)
else:
child_header = f"{current_level}_{subKey}"
if child_header not in attributes:
@ -190,9 +190,10 @@ class DBConn:
child_attribute = f"{key}_{subKey}"
if child_attribute not in attributes:
attributes.append(child_attribute)
elif value_is_list:
existing = children.get(key, {})
children[key] = self.extract_child(value, key, f"{top_level}_", existing, 1)
children[key] = self.extract_child(value, key, f"{top_level}_", existing)
else:
if key not in attributes:
attributes.append(key)