Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/python
- data_folder = 'data'
- import os
- import sys
- from lxml import etree as ET
- if len(sys.argv) < 2 :
- sys.exit("Usage: gproj-update my_project.gproj [data_filename='data']")
- if len(sys.argv) >= 3 :
- data_folder = sys.argv[2]
- filename = sys.argv[1]
- parser = ET.XMLParser(remove_blank_text=True)
- tree = ET.parse(filename, parser)
- root = tree.getroot()
- data = None
- for folder in root.findall('folder') :
- if folder.attrib['name'] == data_folder :
- data = folder
- for child in folder :
- folder.remove(child)
- def walk(root, tag) :
- for item in os.listdir(root) :
- path = os.path.join(root, item)
- print(path)
- if os.path.isdir(path) :
- folder = ET.SubElement(tag, 'folder')
- folder.attrib['name'] = item
- walk(path, folder)
- else :
- file = ET.SubElement(tag, 'file')
- file.attrib['source'] = path
- walk(data_folder, data)
- tree.write(filename, pretty_print=True)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement