old_contents: #
  # Licensed to the Apache Software Foundation (ASF) under one or more
  # contributor license agreements.  See the NOTICE file distributed with
  # this work for additional information regarding copyright ownership.
  # The ASF licenses this file to You 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.
  #
  
  """
  An interactive shell.
  
  This file is designed to be launched as a PYTHONSTARTUP script.
  """
  
  import atexit
  import os
  import platform
  import warnings
  
  from pyspark.context import SparkContext
  from pyspark.sql import SparkSession
  
  if os.environ.get("SPARK_EXECUTOR_URI"):
      SparkContext.setSystemProperty("spark.executor.uri", os.environ["SPARK_EXECUTOR_URI"])
  
  SparkContext._ensure_initialized()  # type: ignore
  
  try:
      spark = SparkSession._create_shell_session()  # type: ignore
  except Exception:
      import sys
      import traceback
  
      warnings.warn("Failed to initialize Spark session.")
      traceback.print_exc(file=sys.stderr)
      sys.exit(1)
  
  sc = spark.sparkContext
  sql = spark.sql
  atexit.register(lambda: sc.stop())
  
  # for compatibility
  sqlContext = spark._wrapped
  sqlCtx = sqlContext
  
  print(
      r"""Welcome to
        ____              __
       / __/__  ___ _____/ /__
      _\ \/ _ \/ _ `/ __/  '_/
     /__ / .__/\_,_/_/ /_/\_\   version %s
        /_/
  """
      % sc.version
  )
  print(
      "Using Python version %s (%s, %s)"
      % (platform.python_version(), platform.python_build()[0], platform.python_build()[1])
  )
  print("Spark context Web UI available at %s" % (sc.uiWebUrl))
  print("Spark context available as 'sc' (master = %s, app id = %s)." % (sc.master, sc.applicationId))
  print("SparkSession available as 'spark'.")
  
  # The ./bin/pyspark script stores the old PYTHONSTARTUP value in OLD_PYTHONSTARTUP,
  # which allows us to execute the user's PYTHONSTARTUP file:
  _pythonstartup = os.environ.get("OLD_PYTHONSTARTUP")
  if _pythonstartup and os.path.isfile(_pythonstartup):
      with open(_pythonstartup) as f:
          code = compile(f.read(), _pythonstartup, "exec")
          exec(code)
  
new_contents: #
  # Licensed to the Apache Software Foundation (ASF) under one or more
  # contributor license agreements.  See the NOTICE file distributed with
  # this work for additional information regarding copyright ownership.
  # The ASF licenses this file to You 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.
  #
  
  """
  An interactive shell.
  
  This file is designed to be launched as a PYTHONSTARTUP script.
  """
  
  import atexit
  import os
  import platform
  import warnings
  
  from pyspark.context import SparkContext
  from pyspark.sql import SparkSession
  
  if os.environ.get("SPARK_EXECUTOR_URI"):
      SparkContext.setSystemProperty("spark.executor.uri", os.environ["SPARK_EXECUTOR_URI"])
  
  SparkContext._ensure_initialized()  # type: ignore
  
  try:
      spark = SparkSession._create_shell_session()  # type: ignore
  except Exception:
      import sys
      import traceback
  
      warnings.warn("Failed to initialize Spark session.")
      traceback.print_exc(file=sys.stderr)
      sys.exit(1)
  
  sc = spark.sparkContext
  sql = spark.sql
  atexit.register((lambda sc: lambda: sc.stop())(sc))
  
  # for compatibility
  sqlContext = spark._wrapped
  sqlCtx = sqlContext
  
  print(
      r"""Welcome to
        ____              __
       / __/__  ___ _____/ /__
      _\ \/ _ \/ _ `/ __/  '_/
     /__ / .__/\_,_/_/ /_/\_\   version %s
        /_/
  """
      % sc.version
  )
  print(
      "Using Python version %s (%s, %s)"
      % (platform.python_version(), platform.python_build()[0], platform.python_build()[1])
  )
  print("Spark context Web UI available at %s" % (sc.uiWebUrl))
  print("Spark context available as 'sc' (master = %s, app id = %s)." % (sc.master, sc.applicationId))
  print("SparkSession available as 'spark'.")
  
  # The ./bin/pyspark script stores the old PYTHONSTARTUP value in OLD_PYTHONSTARTUP,
  # which allows us to execute the user's PYTHONSTARTUP file:
  _pythonstartup = os.environ.get("OLD_PYTHONSTARTUP")
  if _pythonstartup and os.path.isfile(_pythonstartup):
      with open(_pythonstartup) as f:
          code = compile(f.read(), _pythonstartup, "exec")
          exec(code)
  
subject: Update atexit function to avoid issues with late binding