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]
