code: def package_version(version_suffix: str):
      release_version_matcher = re.compile(r"^\d+\.\d+\.\d+$")
      airflow_version = get_current_airflow_version()
  
      update_version = False
      if version_suffix:
          if airflow_version.endswith(f".{version_suffix}"):
              console.print(
                  f"[bright_blue]The {airflow_version} already has suffix {version_suffix}. Not updating it.\n"
              )
          elif not release_version_matcher.match(airflow_version):
              console.print(
                  f"[red]You should only pass version suffix if {airflow_version}"
                  f"does not have suffix in code. The version in code is: {airflow_version}.\n"
              )
              console.print(
                  "[yellow]Make sure that you remove the suffix before using `--version-suffix-for-pypi`!"
              )
              sys.exit(1)
          else:
              update_version = True
      if update_version:
          set_package_version(f"{airflow_version}.{version_suffix}")
      try:
          yield
      finally:
          # Set the version back to the original version
          if update_version:
              set_package_version(airflow_version)
  
file_path: airflow/scripts/in_container/run_prepare_airflow_packages.py
imports: from __future__ import annotations
  import os
  import re
  import subprocess
  import sys
  from contextlib import contextmanager
  from pathlib import Path
  from shutil import rmtree
  import yaml
  from rich.console import Console
label_desc: Use image node_id 6 for calling a global function with example usage: package_version(version_suffix) without return types