mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-31 15:15:38 -05:00
Added dao, test framework and tests for dao.
This commit is contained in:
@@ -10,7 +10,6 @@ from utils import logging
|
||||
|
||||
db = SQLAlchemy()
|
||||
|
||||
|
||||
api_user = LocalProxy(lambda: _request_ctx_stack.top.api_user)
|
||||
|
||||
|
||||
|
||||
@@ -6,11 +6,11 @@ from app import db
|
||||
from app.models import Service
|
||||
|
||||
|
||||
def create_new_service(service_name,
|
||||
user,
|
||||
limit=1000,
|
||||
active=False,
|
||||
restricted=True):
|
||||
def create_service(service_name,
|
||||
user,
|
||||
limit=1000,
|
||||
active=False,
|
||||
restricted=True):
|
||||
service = Service(name=service_name,
|
||||
created_at=datetime.now(),
|
||||
limit=limit,
|
||||
@@ -22,7 +22,10 @@ def create_new_service(service_name,
|
||||
return service.id
|
||||
|
||||
|
||||
def get_services(user, service_id=None):
|
||||
def get_services(service_id=None, user_id=None):
|
||||
# TODO need better mapping from function params to sql query.
|
||||
if service_id:
|
||||
return Service.query.filter_by(user=user, service_id=service_id).one()
|
||||
return Service.query.filter_by(user=user).all()
|
||||
return Service.query.filter_by(id=service_id).one()
|
||||
elif user_id:
|
||||
return Service.query.filter(Service.users.any(id=user_id)).all()
|
||||
return Service.query.all()
|
||||
|
||||
@@ -6,7 +6,7 @@ from app import db
|
||||
from app.models import User
|
||||
|
||||
|
||||
def create_new_user(email_address):
|
||||
def create_user(email_address):
|
||||
user = User(email_address=email_address,
|
||||
created_at=datetime.now())
|
||||
db.session.add(user)
|
||||
@@ -16,5 +16,5 @@ def create_new_user(email_address):
|
||||
|
||||
def get_users(user_id=None):
|
||||
if user_id:
|
||||
return User.query.filter_by(user_id=user_id).one()
|
||||
return User.query.filter_by(id=user_id).one()
|
||||
return User.query.filter_by().all()
|
||||
|
||||
Reference in New Issue
Block a user