def setUp(self): self.logger = logging.getLogger(get_logger_for_name(__name__)) # everything after this line is embarrassing add_handler = True for handler in self.logger.handlers: if isinstance(handler, logging.StreamHandler): add_handler = False if add_handler: new_handler = logging.StreamHandler() new_handler.setLevel(logging.DEBUG) self.logger.addHandler(new_handler)
import json import logging from connection.connection import get_connection from models.model_utils import get_orient_valid_class_name from utils import get_logger_for_name LOGGER = logging.getLogger(get_logger_for_name(__name__)) def create_class(klass, client=None): from models.orient_sql_utils import is_model from models.orient_sql_utils import get_model_extensions if not is_model(klass): raise ValueError("Only OrientDB models can be used to create Orient classes") if client is None: client = get_connection() class_name = get_orient_valid_class_name(klass) create_str = 'CREATE CLASS %s' % class_name parents = get_model_extensions(klass) if len(parents) > 0: if len(parents) > 1: extenders = ",".join(parents) else: extenders = parents[0]