39 lines
860 B
Python
39 lines
860 B
Python
import importlib
|
|
import json
|
|
import pkgutil
|
|
from types import ModuleType
|
|
|
|
|
|
def import_all_modules_from_package(pkg: ModuleType) -> None:
|
|
for _loader, module_name, _is_pkg in pkgutil.walk_packages(
|
|
pkg.__path__, prefix=f"{pkg.__name__}."
|
|
):
|
|
base_name = module_name.split(".")[-1]
|
|
|
|
# Skip test modules
|
|
if (
|
|
base_name.startswith("test_")
|
|
or base_name.endswith("_test")
|
|
or base_name == "conftest"
|
|
):
|
|
continue
|
|
|
|
importlib.import_module(module_name)
|
|
|
|
|
|
def main() -> None:
|
|
import clan_cli
|
|
import clan_lib
|
|
|
|
import_all_modules_from_package(clan_cli)
|
|
import_all_modules_from_package(clan_lib)
|
|
|
|
from clan_lib.api import API
|
|
|
|
schema = API.to_json_schema()
|
|
print(f"""{json.dumps(schema, indent=2)}""")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|