mirror of
https://github.com/adulau/abstract-key-server.git
synced 2024-12-22 00:36:03 +00:00
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
from flask import Flask, request, abort, Response
|
|
import configparser
|
|
import redis
|
|
|
|
app = Flask(__name__)
|
|
|
|
config = configparser.RawConfigParser()
|
|
config.read('../conf/aks.conf')
|
|
version = config.get('global', 'version')
|
|
ardb_port = int(config.get('global', 'ardb-port'))
|
|
namespace = config.get('global', 'namespace')
|
|
backend = redis.Redis(host='127.0.0.1', port=ardb_port, db=0)
|
|
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return 'Index Page'
|
|
|
|
@app.route('/pks/lookup')
|
|
def pks():
|
|
if (request.headers.get('Content-Type') is None) or (request.headers.get('ContentType') == 'text/plain') or (request.headers.get('ContentType') == 'application/pgp-keys'):
|
|
machinereadable = True
|
|
else:
|
|
machinereadable = False
|
|
op = request.args.get('op').lower()
|
|
if not (op == 'get' or op == 'index'):
|
|
abort(501)
|
|
search = request.args.get('search')
|
|
if not search:
|
|
abort(500)
|
|
if op == 'get' and search.lower().startswith('0x'):
|
|
print('Get for {}'.format(search))
|
|
if backend.exists('k:{}'.format(search.lower()[2:])):
|
|
return Response('{}'.format(backend.get('k:{}'.format(search.lower()[2:]))), mimetype='application/pgp-keys')
|
|
if op == 'index' and search.lower():
|
|
print('Searching for {}'.format(search))
|
|
ret = backend.scan(0, 'ue:*{}'.format(search), count=100)
|
|
if len(ret[1]) > 0:
|
|
return '{}'.format(ret[1])
|
|
else:
|
|
abort(404)
|
|
return '{}\n'.format(op)
|
|
|
|
@app.route('/version')
|
|
def hello():
|
|
return '{}\n'.format(version)
|