python-snippets/pseudodb/pseudodb.py

import importlib


class PseudoDB:
    def __init__(self, datasource="default.data"):
        self.datasource = datasource
        self.data = self.load_data()

    def load_data(self):
        module_name = self.datasource.replace('.py', '')
        try:
            module = importlib.import_module(module_name)
            return module
        except ImportError:
            print(f"Data module '{module_name}' not found.")
            return None

    def get_collections(self):
        # Check if default has been successfully loaded
        if self.data is not None:
            attr_names = []
            for attr_name in dir(self.data):
                if not attr_name.startswith('__'):
                    attr_names.append(attr_name)
            return attr_names
        else:
            return []

    def get_keys(self, attr_name):
        if self.data is not None:
            collection = getattr(self.data, attr_name)
            if isinstance(collection, dict):
                return list(collection.keys())
            else:
                return {}

    def get_values(self, attr_name, key):
        if self.data is not None:
            collection = getattr(self.data, attr_name)
            if isinstance(collection, dict):
                return list(collection[key])
            else:
                return collection
        else:
            return None