mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 07:21:13 -05:00
Created endpoints for create and delete token.
This commit is contained in:
@@ -1,17 +0,0 @@
|
||||
from app import db
|
||||
from app.models import ApiToken
|
||||
|
||||
|
||||
def save_token_model(token, update_dict={}):
|
||||
if update_dict:
|
||||
del update_dict['id']
|
||||
db.session.query(ApiToken).filter_by(id=token.id).update(update_dict)
|
||||
else:
|
||||
db.session.add(token)
|
||||
db.session.commit()
|
||||
|
||||
|
||||
def get_model_api_tokens(token=None):
|
||||
if token:
|
||||
return ApiToken.query.filter_by(token=token).one()
|
||||
return ApiToken.query.filter_by().all()
|
||||
22
app/dao/tokens_dao.py
Normal file
22
app/dao/tokens_dao.py
Normal file
@@ -0,0 +1,22 @@
|
||||
from app import db
|
||||
from app.models import Token
|
||||
|
||||
|
||||
def save_token_model(token, update_dict={}):
|
||||
if update_dict:
|
||||
del update_dict['id']
|
||||
db.session.query(Token).filter_by(id=token.id).update(update_dict)
|
||||
else:
|
||||
db.session.add(token)
|
||||
db.session.commit()
|
||||
|
||||
|
||||
def get_model_tokens(service_id=None):
|
||||
if service_id:
|
||||
return Token.query.filter_by(service_id=service_id).one()
|
||||
return Token.query.filter_by().all()
|
||||
|
||||
|
||||
def delete_model_token(token):
|
||||
db.session.delete(token)
|
||||
db.session.commit()
|
||||
Reference in New Issue
Block a user