From 79d7e4f88b5ea170386628e3d2d00a1bd9634154 Mon Sep 17 00:00:00 2001
From: Patrick Lehmann <Patrick.Lehmann@plc2.de>
Date: Tue, 22 Jun 2021 18:17:05 +0200
Subject: scripts/pnodespy: cleaner print_file_header

---
 scripts/pnodespy.py | 18 ++++--------------
 1 file changed, 4 insertions(+), 14 deletions(-)

(limited to 'scripts')

diff --git a/scripts/pnodespy.py b/scripts/pnodespy.py
index 7860b2f30..e76869683 100755
--- a/scripts/pnodespy.py
+++ b/scripts/pnodespy.py
@@ -35,21 +35,11 @@ def print_file_header(includeIntEnumUnique=True, includeBindToLibGHDL=True):
             # Auto generated Python source file from Ada sources
             # Call 'make' in 'src/vhdl' to regenerate:
             #
-        """), end=''
-    )
-
-    if includeIntEnumUnique:
-        print("from enum import IntEnum, unique")
-
-    print("from pydecor import export")
-
-    if includeBindToLibGHDL:
-        print(dedent("""\
-
-                from pyGHDL.libghdl._decorator import BindToLibGHDL
-
-            """), end=''
+        """) + "{sysImports}from pydecor import export\n{moduleImports}".format(
+            sysImports = "from enum import IntEnum, unique\n" if includeIntEnumUnique else "",
+            moduleImports = "\nfrom pyGHDL.libghdl._decorator import BindToLibGHDL\n" if includeBindToLibGHDL else "",
         )
+    )
 
 
 def do_class_kinds():
-- 
cgit v1.2.3