From 0fee49e57b21ec45acee145709056d7c89684db4 Mon Sep 17 00:00:00 2001
From: Eugenio Parodi <ceccopierangiolieugenio@googlemail.com>
Date: Sat, 16 Nov 2024 14:32:35 +0000
Subject: [PATCH] improved init import modifier

---
 tools/process.__init__.import.py | 79 +++++++++++++++++++++++++++++---
 1 file changed, 72 insertions(+), 7 deletions(-)
 mode change 100644 => 100755 tools/process.__init__.import.py

diff --git a/tools/process.__init__.import.py b/tools/process.__init__.import.py
old mode 100644
new mode 100755
index ab8db3b7..173ab5ed
--- a/tools/process.__init__.import.py
+++ b/tools/process.__init__.import.py
@@ -22,7 +22,7 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-import os, sys
+import os, sys, re
 import ast
 import _ast
 
@@ -30,6 +30,26 @@ sys.path.append(os.path.join(sys.path[0],'..'))
 
 import TermTk as ttk
 
+def find_init_and_files(folder_path):
+    """
+    Recursively scan a folder to find __init__.py files and list all files in their directories.
+
+    Args:
+        folder_path (str): The folder to scan.
+
+    Returns:
+        dict: A dictionary with __init__.py paths as keys and arrays of files in their directories as values.
+    """
+    init_files_dict = {}
+
+    for root, _, files in os.walk(folder_path):
+        if "__init__.py" in files:
+            init_path = os.path.join(root, "__init__.py")
+            files_in_folder = [os.path.join(root, f) for f in files if f != "__init__.py"]
+            init_files_dict[init_path] = files_in_folder
+
+    return init_files_dict
+
 def find_init_and_files(folder_path):
     """
     Recursively scan a folder to find __init__.py files and list all files in their directories.
@@ -71,12 +91,57 @@ def get_variables2(tree):
 
 ret = find_init_and_files('TermTk')
 for init in ret:
-    print(f"{init}")
-    for file in ret[init]:
-        vars = get_variables(ast.parse(open(file).read()))
-        if '__all__' in vars:
-            print(ttk.TTkString(f"  - {file}:{vars['__all__']}", ttk.TTkColor.GREEN).toAnsi())
+    dirname = os.path.dirname(init)
+    initName = os.path.basename(init)
+    print(f"{dirname} - {initName}")
+    initLines = []
+    with open(init) as fh:
+        initLines = fh.read().split('\n')
+    outLines = []
+    for line in initLines:
+        if not (m := re.match('( *from *\.([^ ]*) *import *)\*', line)):
+            outLines.append(line)
         else:
-            print(ttk.TTkString(f"  - {file}: No __all__ Found", ttk.TTkColor.RED).toAnsi())
+            prefix   = m.group(1)
+            fileName = f"{dirname}/{m.group(2)}.py"
+            print(f"{m.group(0)} | {m.group(1)} | {m.group(2)}")
+            if fileName in ret[init]:
+                vars = get_variables(ast.parse(open(fileName).read()))
+                if '__all__' in vars:
+                    if _va:=vars['__all__']:
+                        _line = f"{prefix}{', '.join(_va)}"
+                    else:
+                        _line = f"# {prefix}*"
+                    print(_line)
+                    print(ttk.TTkString(f"  - {fileName}:{vars['__all__']}", ttk.TTkColor.GREEN).toAnsi())
+                    outLines.append(_line)
+                else:
+                    print(ttk.TTkString(f"  - {fileName}: No __all__ Found", ttk.TTkColor.RED).toAnsi())
+                    outLines.append(line)
+            else:
+                outLines.append(line)
+                print(f"{fileName} not in {ret[init]}")
+    print('\n'.join(outLines))
+    with open(init, 'w') as fp:
+        fp.write('\n'.join(outLines))
+
+
+
+
+    # with open(fileName, 'w') as fp:
+    #         fp.write(textEdit.toPlainText())
+
+    # os.path.basename(init)
+    # for file in ret[init]:
+    #     vars = get_variables(ast.parse(open(file).read()))
+    #     if '__all__' in vars:
+    #         print(ttk.TTkString(f"  - {file}:{vars['__all__']}", ttk.TTkColor.GREEN).toAnsi())
+    #     else:
+    #         print(ttk.TTkString(f"  - {file}: No __all__ Found", ttk.TTkColor.RED).toAnsi())
 
+# for root, _, files in os.walk('TermTk'):
+#     for file in files:
+#         initLines = []
+#         with open(f"{root}/{file}") as fh:
+#             initLines = fh.read().split('\n')