server

api/requirements.txt

@@ -0,0 +1,3 @@
+waitress
+flask
+paste

api/server.py

@@ -0,0 +1,90 @@
+import json
+import logging
+import sys
+from typing import Union
+
+from flask import Flask, request
+from paste.translogger import TransLogger
+from waitress import serve
+
+from names_dataset import NameDataset, NameWrapper
+
+logger = logging.getLogger(__name__)
+logging.basicConfig(
+ level=logging.DEBUG,
+ format='%(asctime)s - %(levelname)s - %(threadName)s - %(message)s',
+ stream=sys.stdout
+)
+
+app = Flask(__name__)
+nd = NameDataset()
+
+
+def generate_output(d: Union[str, dict], status: bool) -> str:
+ return json.dumps({'status': status, 'message': d}, ensure_ascii=False, default=str)
+
+
+@app.errorhandler(404)
+def invalid_route(e):
+ return generate_output('invalid endpoint', status=False)
+
+
+@app.route('/')
+def main():
+ return generate_output('Welcome user! Name dataset api. query /search to perform a search.', status=True)
+
+
+def str2bool(s: Union[bool, str]) -> bool:
+ if isinstance(s, bool):
+ return s
+ # noinspection PyBroadException
+ try:
+ return bool(eval(s))
+ except Exception:
+ if s.lower() in ['1', '0', 'true', 'y']:
+ return True
+ return False
+
+
+@app.route('/country_codes', methods=['GET'])
+def country_codes():
+ try:
+ req = request
+ alpha_2 = str2bool(req.args.get('alpha_2', False))
+ result = nd.get_country_codes(alpha_2=alpha_2)
+ return generate_output({'result': result}, status=True)
+ except Exception as e:
+ return generate_output({'error': str(e)}, status=True)
+
+
+@app.route('/top', methods=['GET'])
+def top():
+ try:
+ req = request
+ n = int(req.args.get('n', 100))
+ use_first_names = str2bool(req.args.get('use_first_names', True))
+ country_alpha2 = req.args.get('country_alpha2', None)
+ gender = req.args.get('gender', None)
+ result = nd.get_top_names(n, use_first_names, country_alpha2, gender)
+ return generate_output({'result': result}, status=True)
+ except Exception as e:
+ return generate_output({'error': str(e)}, status=True)
+
+
+@app.route('/search', methods=['GET'])
+def search():
+ try:
+ req = request
+ q = req.args.get('q')
+ if q is None:
+ return generate_output('provide a parameter q, for example q=Mike', status=False)
+ else:
+ result = nd.search(q)
+ result['describe'] = NameWrapper(result).describe
+ return generate_output({'result': result}, status=True)
+ except Exception as e:
+ return generate_output({'error': str(e)}, status=True)
+
+
+if __name__ == '__main__':
+ serve(TransLogger(app, setup_console_handler=False), port=8888, threads=4)