def model(self):
"""
Return a model. Called by the server.
"""
model_class = self.config['server']['class']
model = None
if model_class == 'MySQLModel':
from codalab.model.mysql_model import MySQLModel
model = MySQLModel(
engine_url=self.config['server']['engine_url'],
default_user_info=self.default_user_info(),
root_user_id=self.root_user_id(),
system_user_id=self.system_user_id(),
)
elif model_class == 'SQLiteModel':
from codalab.model.sqlite_model import SQLiteModel
model = SQLiteModel(
default_user_info=self.default_user_info(),
root_user_id=self.root_user_id(),
system_user_id=self.system_user_id(),
)
else:
raise UsageError(
'Unexpected model class: %s, expected MySQLModel' %
(model_class, ))
return model
def model(self):
"""
Return a model. Called by the server.
"""
model_class = self.config['server']['class']
model = None
if model_class == 'MySQLModel':
from codalab.model.mysql_model import MySQLModel
model = MySQLModel(
engine_url=self.config['server']['engine_url'],
default_user_info=self.default_user_info(),
)
elif model_class == 'SQLiteModel':
from codalab.model.sqlite_model import SQLiteModel
# Patch for backwards-compatibility until we have a cleaner abstraction around config
# that can update configs to newer "versions"
engine_url = self.config['server'].get(
'engine_url', "sqlite:///{}".format(
os.path.join(self.codalab_home, 'bundle.db')))
model = SQLiteModel(engine_url=engine_url,
default_user_info=self.default_user_info())
else:
raise UsageError(
'Unexpected model class: %s, expected MySQLModel or SQLiteModel'
% (model_class, ))
model.root_user_id = self.root_user_id()
model.system_user_id = self.system_user_id()
return model
def setUpClass(cls):
cls.test_root = path_util.normalize("~/.codalab_tests")
path_util.make_directory(cls.test_root)
cls.bundle_store = BundleStore(cls.test_root, [])
cls.model = SQLiteModel(cls.test_root)
cls.model.root_user_id = '0'
users = [
User('root', '0'),
User('user1', '1'),
User('user2', '2'),
User('user4', '4')
]
cls.auth_handler = MockAuthHandler(users)
cls.client = LocalBundleClient('local',
cls.bundle_store,
cls.model,
cls.auth_handler,
verbose=1)
def model(self):
'''
Return a model. Called by the server.
'''
model_class = self.config['server']['class']
model = None
if model_class == 'MySQLModel':
from codalab.model.mysql_model import MySQLModel
model = MySQLModel(engine_url=self.config['server']['engine_url'])
elif model_class == 'SQLiteModel':
codalab_home = self.codalab_home()
from codalab.model.sqlite_model import SQLiteModel
model = SQLiteModel(codalab_home)
else:
raise UsageError(
'Unexpected model class: %s, expected MySQLModel or SQLiteModel'
% (model_class, ))
model.root_user_id = self.root_user_id()
return model
def setUpClass(cls):
cls.test_root = path_util.normalize("~/.codalab_tests")
path_util.make_directory(cls.test_root)
cls.bundle_store = MultiDiskBundleStore(cls.test_root)
cls.model = SQLiteModel(
"sqlite:///{}".format(os.path.join(cls.test_root, 'bundle.db')),
{})
cls.model.root_user_id = '0'
users = [
User('root', '0'),
User('user1', '1'),
User('user2', '2'),
User('user4', '4')
]
cls.auth_handler = MockAuthHandler(users)
cls.client = LocalBundleClient('local',
cls.bundle_store,
cls.model,
None,
cls.auth_handler,
verbose=1)
