def ready(self):
from django.conf import settings
from api.models.util import get_master_user
from api.models.util import get_lock
from api.models.util import parse_key
from api.models import PublicKey
# Try to set up our master user, this fails under makemigrations.
try:
master_key = settings.MASTER_PUBLIC_KEY
# Try to parse the master key (exception if invalid)
parse_key(master_key)
with transaction.atomic(savepoint=True):
# Ensure our master key user exists
user = get_master_user()
# Acquire lock on master key user
lock = get_lock()
# Delete all keys, which aren't the master key
deleted_count, del_dict = PublicKey.objects.filter(
user=user).exclude(key=master_key).delete()
key, created = PublicKey.objects.get_or_create(user=user,
key=master_key)
if deleted_count != 0:
print "Deleted spurious master key(s)"
if created:
print "Created master key"
except Exception as exception:
print str(exception)
# TODO: Log exception
pass
def validate_key(self, value):
"""Ensure that only valid ssh public keys can be uploaded."""
key_string = str(value)
# Check if parsing throws an exception
try:
util.parse_key(key_string)
except ValueError as value_error:
raise ValidationError(str(value_error))
return key_string
def as_key(self):
return util.parse_key(self.key)