You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
2.2 KiB
65 lines
2.2 KiB
2 months ago
|
import csv
|
||
|
|
import io
|
||
|
|
|
||
|
|
# Read file
|
||
|
|
with open(r'D:\EnforcementCodeProject\CodeV2\EnforcementCode-2.0\doc\数据权限控制\原始数据.csv', 'r', encoding='utf-8') as f:
|
||
|
|
content = f.read()
|
||
|
|
|
||
|
|
# Parse paths from CSV, store vir_id and vir_zone for each path endpoint
|
||
|
|
path_info = {} # path -> (vir_id, vir_zone)
|
||
|
|
reader = csv.reader(io.StringIO(content))
|
||
|
|
for row in reader:
|
||
|
|
if len(row) >= 8:
|
||
|
|
path = row[6].strip()
|
||
|
|
vir_id = row[0].strip()
|
||
|
|
vir_zone = row[4].strip()
|
||
|
|
if path and path.startswith('深圳市'):
|
||
|
|
path_info[path] = (vir_id, vir_zone)
|
||
|
|
|
||
|
|
# Build tree - store children as ordered dict
|
||
|
|
tree = {}
|
||
|
|
for path in sorted(path_info.keys()):
|
||
|
|
parts = path.split('/')
|
||
|
|
node = tree
|
||
|
|
for part in parts:
|
||
|
|
if part not in node:
|
||
|
|
node[part] = {}
|
||
|
|
node = node[part]
|
||
|
|
|
||
|
|
# Build a lookup: full_path -> (vir_id, vir_zone) for each node
|
||
|
|
# We need to map each node in the tree to its path to find info
|
||
|
|
node_info = {} # tuple of path parts -> (vir_id, vir_zone)
|
||
|
|
for path, info in path_info.items():
|
||
|
|
parts = tuple(path.split('/'))
|
||
|
|
node_info[parts] = info
|
||
|
|
|
||
|
|
# Generate ASCII tree
|
||
|
|
lines = []
|
||
|
|
|
||
|
|
def render_tree(node, prefix='', is_root=True, path_parts=()):
|
||
|
|
keys = sorted(node.keys())
|
||
|
|
for i, key in enumerate(keys):
|
||
|
|
is_last_item = (i == len(keys) - 1)
|
||
|
|
current_path = path_parts + (key,)
|
||
|
|
# Get info suffix
|
||
|
|
info = node_info.get(current_path)
|
||
|
|
suffix = f'({info[0]})({info[1]})' if info else ''
|
||
|
|
|
||
|
|
if is_root:
|
||
|
|
lines.append(key + suffix)
|
||
|
|
render_tree(node[key], '', False, current_path)
|
||
|
|
else:
|
||
|
|
connector = '└── ' if is_last_item else '├── '
|
||
|
|
lines.append(prefix + connector + key + suffix)
|
||
|
|
extension = ' ' if is_last_item else '│ '
|
||
|
|
render_tree(node[key], prefix + extension, False, current_path)
|
||
|
|
|
||
|
|
render_tree(tree)
|
||
|
|
|
||
|
|
# Write output
|
||
|
|
output = '# 树型执法单位\n\n```\n' + '\n'.join(lines) + '\n```\n'
|
||
|
|
with open(r'D:\EnforcementCodeProject\CodeV2\EnforcementCode-2.0\doc\数据权限控制\树型执法单位.md', 'w', encoding='utf-8') as f:
|
||
|
|
f.write(output)
|
||
|
|
|
||
|
|
print(f'Done! Total paths: {len(path_info)}, Total lines: {len(lines)}')
|