import os import logging from data.registry_model.registry_pre_oci_model import pre_oci_model from data.registry_model.registry_oci_model import oci_model from data.registry_model.modelsplitter import SplitModel logger = logging.getLogger(__name__) class RegistryModelProxy(object): def __init__(self): self._model = oci_model if os.getenv('OCI_DATA_MODEL') == 'true' else pre_oci_model def setup_split(self, v22_whitelist): if os.getenv('OCI_DATA_MODEL') == 'true': return logger.info('===============================') logger.info('Enabling split registry model with namespace whitelist `%s`', v22_whitelist) logger.info('===============================') self._model = SplitModel(v22_whitelist) def set_for_testing(self, use_oci_model): self._model = oci_model if use_oci_model else pre_oci_model logger.debug('Changed registry model to `%s` for testing', self._model) def __getattr__(self, attr): return getattr(self._model, attr) registry_model = RegistryModelProxy() logger.info('===============================') logger.info('Using registry model `%s`', registry_model._model) logger.info('===============================')