def quota_class_update(context, class_name, resource, limit):
with write_session() as session:
quota_class_ref = session.query(models.QuotaClass). \
filter_by(deleted=0). \
filter_by(class_name=class_name). \
filter_by(resource=resource).first()
if not quota_class_ref:
raise exception.QuotaClassNotFound(class_name=class_name)
quota_class_ref.hard_limit = limit
quota_class_ref.save(session)
return quota_class_ref
def quota_class_destroy_all(context, class_name):
with write_session() as session:
quota_classes = session.query(models.QuotaClass). \
filter_by(deleted=0). \
filter_by(class_name=class_name). \
all()
if quota_classes:
for quota_class_ref in quota_classes:
session.delete(quota_class_ref)
else:
raise exception.QuotaClassNotFound()
def _quota_class_get(context, class_name, resource):
result = model_query(context, models.QuotaClass). \
filter_by(deleted=0). \
filter_by(class_name=class_name). \
filter_by(resource=resource). \
first()
if not result:
raise exception.QuotaClassNotFound(class_name=class_name)
return result