key server: misc fixes to make jwtproxy work

This commit is contained in:
Jimmy Zelinskie 2016-04-01 15:48:31 -04:00 committed by Jimmy Zelinskie
parent 5cdc7812dc
commit 885a41e6f5
3 changed files with 8 additions and 7 deletions

View file

@ -1,6 +1,8 @@
from calendar import timegm
from datetime import datetime, timedelta
from peewee import JOIN_LEFT_OUTER
from app import app
from data.database import db_for_update, User, ServiceKey, ServiceKeyApproval
from data.model import ServiceKeyDoesNotExist, ServiceKeyAlreadyApproved, db_transaction
@ -39,8 +41,6 @@ def _notify_superusers(key):
superusers = User.select().where(User.username << app.config['SUPER_USERS'])
for superuser in superusers:
# TODO(jzelinskie): create notification type in the database migration
# I already put it in initdb
create_notification('service_key_submitted', superuser, metadata=notification_metadata,
lookup_path='/service_key_approval/{0}'.format(key.kid))
@ -94,14 +94,15 @@ def delete_service_key(service, kid):
_gc_expired(service)
def approve_service_key(kid, approver, approval_type):
def approve_service_key(kid, approver, approval_type, notes=''):
try:
with db_transaction():
key = db_for_update(ServiceKey.select().where(ServiceKey.kid == kid)).get()
if key.approval is not None:
raise ServiceKeyAlreadyApproved
approval = ServiceKeyApproval.create(approver=approver, approval_type=approval_type)
approval = ServiceKeyApproval.create(approver=approver, approval_type=approval_type,
notes=notes)
key.approval = approval
key.save()
except ServiceKey.DoesNotExist:
@ -112,7 +113,7 @@ def approve_service_key(kid, approver, approval_type):
def _list_service_keys_query(kid=None, service=None, approved_only=False):
query = ServiceKey.select().join(ServiceKeyApproval)
query = ServiceKey.select().join(ServiceKeyApproval, JOIN_LEFT_OUTER)
if approved_only:
query = query.where(~(ServiceKey.approval >> None))

View file

@ -637,7 +637,6 @@ class SuperUserServiceKeyApproval(ApiResource):
@verify_not_prod
@nickname('approveServiceKey')
@require_scope(scopes.SUPERUSER)
@validate_json_request('ApproveServiceKey')
def put(self, kid):
if SuperUserPermission().can():
approver = get_authenticated_user()
@ -648,6 +647,6 @@ class SuperUserServiceKeyApproval(ApiResource):
except model.ServiceKeyAlreadyApproved:
pass
make_response('', 200)
return make_response('', 200)
abort(403)

View file

@ -83,6 +83,7 @@ def list_service_keys(service):
@key_server.route('/services/<service>/keys/<kid>', methods=['GET'])
def get_service_key(service, kid):
logger.debug(kid)
try:
key = data.model.service_keys.get_service_key(kid)
except data.model.ServiceKeyDoesNotExist: