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