pre-commit

Author: philsc

python/private/repl_template.py

@@ -14,6 +14,8 @@ def start_repl():
         cprt = 'Type "help", "copyright", "credits" or "license" for more information.'
         sys.stderr.write("Python %s on %s\n%s\n" % (sys.version, sys.platform, cprt))
 
+    # If there's a PYTHONSTARTUP script, we need to capture the new variables
+    # that it defines.
     new_globals = {}
 
     # Simulate Python's behavior when a valid startup script is defined by the
@@ -32,7 +34,11 @@ def start_repl():
             eval(compiled_code, new_globals)
 
     bazel_runfiles = runfiles.Create()
-    runpy.run_path(bazel_runfiles.Rlocation(STUB_PATH), init_globals=new_globals, run_name="__main__")
+    runpy.run_path(
+        bazel_runfiles.Rlocation(STUB_PATH),
+        init_globals=new_globals,
+        run_name="__main__",
+    )
 
 
 if __name__ == "__main__":

tests/repl/repl_test.py

@@ -3,8 +3,8 @@
 import sys
 import tempfile
 import unittest
-from typing import Iterable
 from pathlib import Path
+from typing import Iterable
 
 from python import runfiles
 
@@ -21,6 +21,7 @@
 foo = 1234
 """
 
+
 class ReplTest(unittest.TestCase):
     def setUp(self):
         self.repl = rfiles.Rlocation("rules_python/python/bin/repl")
@@ -87,9 +88,12 @@ def test_pythonstartup_gets_executed(self):
             env = os.environ.copy()
             env["PYTHONSTARTUP"] = str(pythonstartup)
 
-            result = self.run_code_in_repl([
-                "print(f'The value of foo is {foo}')",
-            ], env=env)
+            result = self.run_code_in_repl(
+                [
+                    "print(f'The value of foo is {foo}')",
+                ],
+                env=env,
+            )
 
         self.assertIn("The value of foo is 1234", result)
 
@@ -109,8 +113,9 @@ def test_pythonstartup_doesnt_leak(self):
             for var_name in ("exitmsg", "sys", "code", "bazel_runfiles", "STUB_PATH"):
                 with self.subTest(var_name=var_name):
                     result = self.run_code_in_repl([f"print({var_name})"], env=env)
-                    self.assertIn(f"NameError: name '{var_name}' is not defined", result)
-
+                    self.assertIn(
+                        f"NameError: name '{var_name}' is not defined", result
+                    )
 
 
 if __name__ == "__main__":