import os
import sys
import importlib.util
import types
from collections import defaultdict
from sys import getsizeof


def get_bytecode_size(module_name):
    """Get the size of the cached .pyc bytecode file for a module."""
    spec = importlib.util.find_spec(module_name)
    if spec and spec.origin and spec.origin.endswith('.py'):
        pyc_file = importlib.util.cache_from_source(spec.origin)
        if os.path.exists(pyc_file):
            return os.path.getsize(pyc_file)
    return 0


def get_dso_size(module_name):
    """Get the size of the DSO (Dynamic Shared Object) module file for a module."""
    spec = importlib.util.find_spec(module_name)
    if spec and spec.origin and spec.origin.endswith(('.so', '.dll')):
        return os.path.getsize(spec.origin)
    return 0


def generate_module_report():
    """Generate a report summarizing the module environment."""
    total_modules = len(sys.modules)
    bytecode_size = sum(get_bytecode_size(module_name) for module_name in sys.modules)
    dso_size = sum(get_dso_size(module_name) for module_name in sys.modules)
    python_size = sum(
        getsizeof(sys.modules[module_name]) - getsizeof(module_name)
        for module_name in sys.modules
        if isinstance(sys.modules[module_name], types.ModuleType)
    )
    total_size = bytecode_size + dso_size + python_size

    # Calculate percentage of each type of module
    bytecode_percent = bytecode_size / total_size * 100
    dso_percent = dso_size / total_size * 100
    python_percent = python_size / total_size * 100

    # Generate the report
    report = f"Total modules: {total_modules}\n"
    report += f"Total size: {total_size} bytes\n"
    report += f"Bytecode modules: {bytecode_percent:.2f}% ({bytecode_size} bytes)\n"
    report += f"DSO modules: {dso_percent:.2f}% ({dso_size} bytes)\n"
    report += f"Python code modules: {python_percent:.2f}% ({python_size} bytes)\n"

    # Generate histogram of module directory names
    dir_histogram = defaultdict(int)
    for module_name in sys.modules:
        module_file = getattr(sys.modules[module_name], '__file__', None)
        if module_file:
            module_dir = os.path.dirname(module_file)
            dir_histogram[module_dir] += 1

    report += "\nModule Directory Histogram:\n"
    for module_dir, count in dir_histogram.items():
        report += f"{module_dir}: {'*' * count}\n"

    return report


if __name__ == "__main__":
    __spec__ = importlib.util.spec_from_file_location('__main__', __file__)
report = generate_module_report()
print(report)






    report = generate_module_report()
    print(report)