Slight tiny up of templates_dao

- 1 method per get query
- logic around create/update removed
This commit is contained in:
Martyn Inglis
2016-02-22 09:46:16 +00:00
parent cc720b137f
commit 443691402f
3 changed files with 125 additions and 39 deletions

View File

@@ -1,5 +1,3 @@
import json
from . import DAOException
from app import db
from app.models import Service
from sqlalchemy import asc

View File

@@ -1,9 +1,6 @@
import json
from datetime import datetime
from sqlalchemy.orm import load_only
from . import DAOException
from app import db
from app.models import (Template, Service)
from sqlalchemy import asc
def save_model_template(template, update_dict=None):
@@ -32,3 +29,21 @@ def get_model_templates(template_id=None, service_id=None):
elif service_id:
return Template.query.filter_by(service=Service.query.get(service_id)).all()
return Template.query.all()
def dao_create_template(template):
db.session.add(template)
db.session.commit()
def dao_update_template(template):
db.session.add(template)
db.session.commit()
def dao_get_template_by_id_and_service_id(template_id, service_id):
return Template.query.filter_by(id=template_id, service_id=service_id).first()
def dao_get_all_templates_for_service(service_id):
return Template.query.filter_by(service=Service.query.get(service_id)).order_by(asc(Template.created_at)).all()