fix module not importing when in interpreted session

Bubobubobubobubo · Nov 7, 2022 · dd6bb7c · dd6bb7c
1 parent d1752cf
commit dd6bb7c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/fishery/__main__.py b/fishery/__main__.py
@@ -98,6 +98,8 @@ def callback():
             else:
                 self.showtraceback()
 
+from sys import argv
+print(f"File name: {argv[0]}")
 
 class REPLThread(threading.Thread):
     def run(self):

diff --git a/sardine/__init__.py b/sardine/__init__.py
@@ -326,6 +326,11 @@ def __exit__(self, exc_type, exc_val, exc_tb):
             sleep(self.duration)
 
 
+from sys import argv
+hook_path = argv[0]
+if "__main__.py" in hook_path:
+    os.environ['SARDINE_INIT_SESSION'] = 'YES'
+
 if (
     os.getenv("SARDINE_INIT_SESSION") is not None
     and os.getenv("SARDINE_INIT_SESSION") == "YES"