diff --git a/lib/dxf2lib.py b/lib/dxf2lib.py index b73a5b6..1dd966d 100644 --- a/lib/dxf2lib.py +++ b/lib/dxf2lib.py @@ -3,7 +3,7 @@ import ezdxf from ezdxf.math import Vec2 import argparse import sys -# Hilfsfunktion für Umgebungsvariablen mit Fallback +import shutil def check_environment_var(env_str: str, fallback: str) -> str: out_path = os.environ.get(env_str) @@ -44,7 +44,8 @@ if __name__ == "__main__": parser = argparse.ArgumentParser(description="SVG/XML zu DXF Konverter") parser.add_argument('-i', '--input', type=str, help='Input-Verzeichnis mit SVG/XML-Dateien') parser.add_argument('-n', '--name', required=True, type=str, help='Name der zu erzeugenden Bibliothek') - + parser.add_argument('-k', '--keep', action="store_true", help='Behalte converted_dxf Ordner. Sonst Löschen nach erfolgreicher Bibliothekserstellung') + if len(sys.argv) == 2 and sys.argv[1] in ("-h", "--help"): parser.print_help() sys.exit(0) @@ -116,3 +117,11 @@ if __name__ == "__main__": print(f"Bibliotheks-DXF gespeichert: {OUTPUT_FILE}") doc.saveas(OUTPUT_FILE) + + if not args.keep: + try: + if os.path.exists(INPUT_DIR): + shutil.rmtree(INPUT_DIR) + print(f"Verzeichnis '{INPUT_DIR}' wurde gelöscht.") + except Exception as err: + print(f"[WARN] Konnte '{INPUT_DIR}' nicht löschen: {err}")