mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-03 09:51:11 -05:00
Slight tiny up of templates_dao
- 1 method per get query - logic around create/update removed
This commit is contained in:
@@ -1,5 +1,3 @@
|
||||
import json
|
||||
from . import DAOException
|
||||
from app import db
|
||||
from app.models import Service
|
||||
from sqlalchemy import asc
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user