Incorporate some of the feedback

Author: philsc

python/bin/repl_stub.py

@@ -1,3 +1,18 @@
+"""Simulates the REPL that Python spawns when invoking the binary with no arguments.
+
+The code module is responsible for the default shell.
+
+The import and `ocde.interact()` call here his is equivalent to doing:
+
+    $ python3 -m code
+    Python 3.11.2 (main, Mar 13 2023, 12:18:29) [GCC 12.2.0] on linux
+    Type "help", "copyright", "credits" or "license" for more information.
+    (InteractiveConsole)
+    >>>
+
+The logic for PYTHONSTARTUP is handled in python/private/repl_template.py.
+"""
+
 import code
 
 code.interact()

python/private/repl.bzl

@@ -1,17 +1,3 @@
-# Copyright 2025 The Bazel Authors. All rights reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 """Implementation of the rules to expose a REPL."""
 
 load("//python:py_binary.bzl", "py_binary")
@@ -33,19 +19,24 @@ _generate_repl_main = rule(
     attrs = {
         "out": attr.output(
             mandatory = True,
+            doc = "The path to the file to generate.",
         ),
         "stub": attr.label(
             mandatory = True,
             allow_single_file = True,
+            doc = ("The stub responsible for actually invoking the final shell. " +
+                   "See the \"Customizing the REPL\" docs for details."),
         ),
         "_template": attr.label(
             default = "//python/private:repl_template.py",
             allow_single_file = True,
+            doc = "The template to use for generating `out`.",
         ),
     },
 )
 
 def py_repl_binary(name, stub, deps = [], data = [], **kwargs):
+    """A 
     _generate_repl_main(
         name = "%s_py" % name,
         stub = stub,