phpIPAM-for-vRA8/venv_tango/lib/python3.7/site-packages/setuptools/tests/files.py
2021-01-11 17:14:27 -06:00

38 lines
1 KiB
Python

import os
def build_files(file_defs, prefix=""):
"""
Build a set of files/directories, as described by the
file_defs dictionary.
Each key/value pair in the dictionary is interpreted as
a filename/contents
pair. If the contents value is a dictionary, a directory
is created, and the
dictionary interpreted as the files within it, recursively.
For example:
{"README.txt": "A README file",
"foo": {
"__init__.py": "",
"bar": {
"__init__.py": "",
},
"baz.py": "# Some code",
}
}
"""
for name, contents in file_defs.items():
full_name = os.path.join(prefix, name)
if isinstance(contents, dict):
os.makedirs(full_name, exist_ok=True)
build_files(contents, prefix=full_name)
else:
if isinstance(contents, bytes):
with open(full_name, 'wb') as f:
f.write(contents)
else:
with open(full_name, 'w') as f:
f.write(contents)