Omit modules from generated imports

robotpy · Jan 6, 2024 · ec7b111 · ec7b111
1 parent 485ea48
commit ec7b111
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/robotpy_build/tool.py b/robotpy_build/tool.py
@@ -8,6 +8,7 @@
 import pprint
 import subprocess
 import sys
+import types
 import re
 from urllib.request import Request, urlopen
 from urllib.error import HTTPError
@@ -233,6 +234,9 @@ def run(self, args):
         # TODO: could probably generate this from parsed code, but seems hard
         ctx = {}
         exec(f"from {compiled} import *", {}, ctx)
+        for k in list(ctx.keys()):
+            if isinstance(ctx[k], types.ModuleType):
+                del ctx[k]
 
         relimport = self._rel(args.base, compiled)