web_server.py

from http.server import BaseHTTPRequestHandler, HTTPServer
import json
from parqet_scraper import get_json


class Server(BaseHTTPRequestHandler):
    def _set_headers(self):
        self.send_response(200)
        self.send_header('Content-type', 'application/json')
        self.end_headers()

    def do_HEAD(self):
        self._set_headers()

    def do_GET(self):
        self._set_headers()
        self.wfile.write(json.dumps(get_json()).encode())


def run(server_class=HTTPServer, handler_class=Server, port=8008):
    server_address = ("0.0.0.0", port)
    httpd = server_class(server_address, handler_class)

    print('Starting httpd on port %d...' % port)
    httpd.serve_forever()


if __name__ == "__main__":
    from sys import argv

    if len(argv) == 2:
        run(port=int(argv[1]))
    else:
        run()