Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # Utils
- import requests, json, os, sys
- # Flask
- from flask import Flask
- from flask import Response
- from flask import request
- # WSGIServer
- from gevent.pywsgi import WSGIServer
- from itertools import groupby
- from operator import itemgetter
- ################# FLASK APP #################
- app = Flask(__name__)
- app.secret_key = os.urandom(24)
- ################# ROUTES #################
- @app.route('/serving-check', methods=['POST'])
- def get_account_ids():
- if request.headers.get('Authorization') != token or request.headers.get('authorization') != token:
- return Response(json.dumps({'Error': 'Missing authorization token'}), mimetype='application/json', status=400)
- else:
- try:
- request_body = request.get_json()
- if request_body.get('domain') != None:
- result = get_ssp_for_serving(request_body['domain'])
- return Response(json.dumps({'ssp': result}), mimetype='application/json', status=200)
- else:
- return Response(json.dumps({'Error': "Missing domain parameter"}), mimetype='application/json', status=400)
- except Exception as e:
- return Response(json.dumps({'Error': str(e)}), mimetype='application/json', status=500)
- ################# MAIN #################
- if __name__ == '__main__':
- print 'Serving on:', str(HOST) + ':' + str(PORT)
- LISTEN = (HOST,PORT)
- http_server = WSGIServer( LISTEN, app )
- http_server.serve_forever()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement