bl_pkg/bl_extension_local.py

   1 # SPDX-FileCopyrightText: 2023 Blender Foundation
   2 #
   3 # SPDX-License-Identifier: GPL-2.0-or-later
   4
   5 """
   6 High level API for managing an extension local site-packages and wheels.
   7
   8 NOTE: this is a standalone module.
   9 """
  10
  11 __all__ = (
  12     "sync",
  13 )
  14
  15
  16 import os
  17 import sys
  18
  19 from .wheel_manager import WheelSource
  20
  21 from typing import (
  22     List,
  23 )
  24
  25
  26 def sync(
  27         *,
  28         local_dir: str,
  29         wheel_list: List[WheelSource],
  30 ) -> None:
  31     from . import wheel_manager
  32     local_dir_site_packages = os.path.join(
  33         local_dir,
  34         "lib",
  35         "python{:d}.{:d}".format(sys.version_info.major, sys.version_info.minor),
  36         "site-packages",
  37     )
  38
  39     wheel_manager.apply_action(
  40         local_dir=local_dir,
  41         local_dir_site_packages=local_dir_site_packages,
  42         wheel_list=wheel_list,
  43     )
  44     if os.path.exists(local_dir_site_packages):
  45         if local_dir_site_packages not in sys.path:
  46             sys.path.append(local_dir_site_packages)