InitScript,
]
class SharedStorage(Model):
FIELDS = [
Field('type', data_type=str),
Field('source', data_type=str)
]
class ContainerRequest(Container):
pass
hosts = ModelManager(MODELS_DIR, Host)
users = ModelManager(MODELS_DIR, User)
shared_storage = ModelManager(MODELS_DIR, SharedStorage)
container_requests = ModelManager(MODELS_DIR, ContainerRequest)
def _get_user_by_openid(openid):
for x in users.list():
x = users.get(x)
if x.openid == openid:
return x
return None
users.get_user_by_openid = _get_user_by_openid
def _generate_api_key():
def setUp(self):
super(TestModel, self).setUp()
self.modeldir = tempfile.mkdtemp()
self.addCleanup(shutil.rmtree, self.modeldir)
self.models = ModelManager(self.modeldir, MyModel)
