mirror of
https://github.com/infeeeee/dyn2py
synced 2025-12-16 22:16:18 +01:00
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import pathlib
|
|
import dyn2py
|
|
|
|
INPUT_DIR = "tests/input_files"
|
|
OUTPUT_DIR = "tests/output_files"
|
|
|
|
|
|
def cleanup_output_dir():
|
|
output_dir = pathlib.Path(OUTPUT_DIR)
|
|
if output_dir.exists():
|
|
for f in output_dir.iterdir():
|
|
f.unlink()
|
|
else:
|
|
output_dir.mkdir()
|
|
|
|
|
|
def extract_single_node_dyn(modify_py: bool = False):
|
|
"""Extract python from single_node.dyn
|
|
File will be here: f"{OUTPUT_DIR}/single_node_1c5d99792882409e97e132b3e9f814b0.py"
|
|
Modified file will be here: f"{OUTPUT_DIR}/single_node_mod.py"
|
|
|
|
Args:
|
|
modify_py (bool, optional): Also do some changes on the exported file. Defaults to False.
|
|
|
|
"""
|
|
cleanup_output_dir()
|
|
|
|
# Extract py:
|
|
options = dyn2py.Options(python_folder=OUTPUT_DIR)
|
|
dyn = dyn2py.DynamoFile(f"{INPUT_DIR}/single_node.dyn")
|
|
dyn.extract_python(options)
|
|
|
|
if modify_py:
|
|
# Open the extracted file and replace a string:
|
|
with open(f"{OUTPUT_DIR}/single_node_1c5d99792882409e97e132b3e9f814b0.py") as orig_py, \
|
|
open(f"{OUTPUT_DIR}/single_node_mod.py", "w") as mod_py:
|
|
for line in orig_py:
|
|
if "asd_string" in line:
|
|
line = line.replace("asd_string", "qwe_string")
|
|
mod_py.write(line)
|