forked from STI/Astroport.ONE
115 lines
4.3 KiB
Python
115 lines
4.3 KiB
Python
"""
|
|
certifi.py
|
|
~~~~~~~~~~
|
|
|
|
This module returns the installation location of cacert.pem or its contents.
|
|
"""
|
|
import sys
|
|
import atexit
|
|
|
|
def exit_cacert_ctx() -> None:
|
|
_CACERT_CTX.__exit__(None, None, None) # type: ignore[union-attr]
|
|
|
|
|
|
if sys.version_info >= (3, 11):
|
|
|
|
from importlib.resources import as_file, files
|
|
|
|
_CACERT_CTX = None
|
|
_CACERT_PATH = None
|
|
|
|
def where() -> str:
|
|
# This is slightly terrible, but we want to delay extracting the file
|
|
# in cases where we're inside of a zipimport situation until someone
|
|
# actually calls where(), but we don't want to re-extract the file
|
|
# on every call of where(), so we'll do it once then store it in a
|
|
# global variable.
|
|
global _CACERT_CTX
|
|
global _CACERT_PATH
|
|
if _CACERT_PATH is None:
|
|
# This is slightly janky, the importlib.resources API wants you to
|
|
# manage the cleanup of this file, so it doesn't actually return a
|
|
# path, it returns a context manager that will give you the path
|
|
# when you enter it and will do any cleanup when you leave it. In
|
|
# the common case of not needing a temporary file, it will just
|
|
# return the file system location and the __exit__() is a no-op.
|
|
#
|
|
# We also have to hold onto the actual context manager, because
|
|
# it will do the cleanup whenever it gets garbage collected, so
|
|
# we will also store that at the global level as well.
|
|
_CACERT_CTX = as_file(files("certifi").joinpath("cacert.pem"))
|
|
_CACERT_PATH = str(_CACERT_CTX.__enter__())
|
|
atexit.register(exit_cacert_ctx)
|
|
|
|
return _CACERT_PATH
|
|
|
|
def contents() -> str:
|
|
return files("certifi").joinpath("cacert.pem").read_text(encoding="ascii")
|
|
|
|
elif sys.version_info >= (3, 7):
|
|
|
|
from importlib.resources import path as get_path, read_text
|
|
|
|
_CACERT_CTX = None
|
|
_CACERT_PATH = None
|
|
|
|
def where() -> str:
|
|
# This is slightly terrible, but we want to delay extracting the
|
|
# file in cases where we're inside of a zipimport situation until
|
|
# someone actually calls where(), but we don't want to re-extract
|
|
# the file on every call of where(), so we'll do it once then store
|
|
# it in a global variable.
|
|
global _CACERT_CTX
|
|
global _CACERT_PATH
|
|
if _CACERT_PATH is None:
|
|
# This is slightly janky, the importlib.resources API wants you
|
|
# to manage the cleanup of this file, so it doesn't actually
|
|
# return a path, it returns a context manager that will give
|
|
# you the path when you enter it and will do any cleanup when
|
|
# you leave it. In the common case of not needing a temporary
|
|
# file, it will just return the file system location and the
|
|
# __exit__() is a no-op.
|
|
#
|
|
# We also have to hold onto the actual context manager, because
|
|
# it will do the cleanup whenever it gets garbage collected, so
|
|
# we will also store that at the global level as well.
|
|
_CACERT_CTX = get_path("certifi", "cacert.pem")
|
|
_CACERT_PATH = str(_CACERT_CTX.__enter__())
|
|
atexit.register(exit_cacert_ctx)
|
|
|
|
return _CACERT_PATH
|
|
|
|
def contents() -> str:
|
|
return read_text("certifi", "cacert.pem", encoding="ascii")
|
|
|
|
else:
|
|
import os
|
|
import types
|
|
from typing import Union
|
|
|
|
Package = Union[types.ModuleType, str]
|
|
Resource = Union[str, "os.PathLike"]
|
|
|
|
# This fallback will work for Python versions prior to 3.7 that lack the
|
|
# importlib.resources module but relies on the existing `where` function
|
|
# so won't address issues with environments like PyOxidizer that don't set
|
|
# __file__ on modules.
|
|
def read_text(
|
|
package: Package,
|
|
resource: Resource,
|
|
encoding: str = 'utf-8',
|
|
errors: str = 'strict'
|
|
) -> str:
|
|
with open(where(), encoding=encoding) as data:
|
|
return data.read()
|
|
|
|
# If we don't have importlib.resources, then we will just do the old logic
|
|
# of assuming we're on the filesystem and munge the path directly.
|
|
def where() -> str:
|
|
f = os.path.dirname(__file__)
|
|
|
|
return os.path.join(f, "cacert.pem")
|
|
|
|
def contents() -> str:
|
|
return read_text("certifi", "cacert.pem", encoding="ascii")
|