def test_iskeyvalid(): p = mktemp() sshkeyGen(p, True) assert util.iskeyvalid(open(p+'.pub').read()) sshkeyGen(p, False) assert not util.iskeyvalid(open(p+'.pub').read()) os.unlink(p) # big deal
def addkey(): remote_user = request.environ['REMOTE_USER'] key = request.form.get('key', None) if not key: return flask.Response('Empty public key', status=400) if not iskeyvalid(key): return flask.Response('Invalid public key', status=400) flask.g.store.addkey(remote_user, key) return flask.Response(status=200)