From a3b900b9f4938cb8de4880d5a99581726499e933 Mon Sep 17 00:00:00 2001
From: philipperemy <premy.enseirb@gmail.com>
Date: Tue, 15 Nov 2022 08:44:44 +0700
Subject: [PATCH] server

---
 api/requirements.txt |  3 ++
 api/server.py        | 90 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 93 insertions(+)
 create mode 100644 api/requirements.txt
 create mode 100644 api/server.py

diff --git a/api/requirements.txt b/api/requirements.txt
new file mode 100644
index 0000000..959e7bd
--- /dev/null
+++ b/api/requirements.txt
@@ -0,0 +1,3 @@
+waitress
+flask
+paste
\ No newline at end of file
diff --git a/api/server.py b/api/server.py
new file mode 100644
index 0000000..01c00f3
--- /dev/null
+++ b/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)