blob: 0ea54eedc661263d6e3ee293cc4c054a77c140db (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
import pkgutil
import sys
from importlib import import_module
from pathlib import Path
from types import ModuleType
from typing import List
def get_path(module: ModuleType) -> str:
if getattr(sys, "frozen", False):
# frozen
if getattr(sys, "_MEIPASS", False):
# PyInstaller
lib_dir = Path(getattr(sys, "_MEIPASS"))
else:
# others
lib_dir = Path(sys.executable).parent / "lib"
path = lib_dir.joinpath(*module.__package__.split(".")) # type: ignore
else:
# unfrozen
if module.__file__ is not None:
path = Path(module.__file__).parent
else:
raise RuntimeError(f"Can't find path from module `{module}.")
return str(path)
def list_module(module: ModuleType) -> List[str]:
path = get_path(module)
if getattr(sys, "_MEIPASS", False):
# PyInstaller
return [file.parent.name for file in Path(path).glob("*/__init__.py")]
else:
return [name for _, name, is_pkg in pkgutil.iter_modules([str(path)]) if is_pkg]
def find_available_locales(providers: List[str]) -> List[str]:
available_locales = set()
for provider_path in providers:
provider_module = import_module(provider_path)
if getattr(provider_module, "localized", False):
langs = list_module(provider_module)
available_locales.update(langs)
return sorted(available_locales)
def find_available_providers(modules: List[ModuleType]) -> List[str]:
available_providers = set()
for providers_mod in modules:
if providers_mod.__package__:
providers = [
".".join([providers_mod.__package__, mod]) for mod in list_module(providers_mod) if mod != "__pycache__"
]
available_providers.update(providers)
return sorted(available_providers)
|
