Move sys.path manipulation to __main__.py

Author: antonio-rojas

pyproject.toml

@@ -103,7 +103,7 @@ content-type = "text/markdown"
 file = "README.md"
 
 [project.scripts]
-sage = "sage.cli:main"
+sage = "sage.cli.__main__:main"
 
 [tool.conda-lock]
 platforms = ['linux-64', 'linux-aarch64', 'osx-64', 'osx-arm64']

src/sage/cli/__main__.py

@@ -1,5 +1,9 @@
+import os
 import sys
 
+# Allow importing modules from the current directory, matching python behavior
+sys.path.append(os.getcwd())
+
 from sage.cli import main
 
 sys.exit(main())

src/sage/cli/eval_cmd.py

@@ -3,8 +3,6 @@
 from sage.cli.options import CliOptions
 from sage.repl.preparse import preparse
 from sage.all import sage_globals
-import os
-import sys
 
 
 class EvalCmd:
@@ -38,9 +36,6 @@ def run(self) -> int:
         r"""
         Execute the given command.
         """
-        # Allow importing modules from the current directory, matching python behavior
-        sys.path.append(os.getcwd())
-
         code = preparse(self.options.command)
         eval(compile(code, "<cmdline>", "exec"), sage_globals())
         return 0

src/sage/cli/run_file_cmd.py

@@ -1,12 +1,11 @@
 import argparse
+import sys
 
 from sage.cli.options import CliOptions
 from sage.repl.preparse import preparse_file_named
 from sage.repl.load import load_cython
 from sage.misc.temporary_file import tmp_filename
 from sage.all import sage_globals
-import os
-import sys
 
 
 class RunFileCmd:
@@ -41,9 +40,6 @@ def run(self) -> int:
         r"""
         Execute the given command.
         """
-        # Allow importing modules from the current directory, matching python behavior
-        sys.path.append(os.getcwd())
-
         input_file = self.options.file[0]
         if input_file.endswith('.sage'):
             input_file = str(preparse_file_named(input_file))