from http.server import BaseHTTPRequestHandler, HTTPServer
import os
class HelloWorldHandler(BaseHTTPRequestHandler):
def do_GET(self):
api_key = os.environ.get('API_KEY', 'NOT_SET')
my_env_var = os.environ.get('MY_ENV_VAR', 'NOT_SET')
response = f'Hello world\nAPI_KEY: {api_key}\nMY_ENV_VAR: {my_env_var}'
self.send_response(200)
self.send_header('Content-type', 'text/html')
self.end_headers()
self.wfile.write(response.encode())
def run_server():
port = os.environ.get('PORT')
server_address = ('', 8000 if port is None else int(port))
httpd = HTTPServer(server_address, HelloWorldHandler)
port = httpd.server_address[1]
print(f'Server running on http://localhost:{port}')
httpd.serve_forever()
if __name__ == '__main__':
run_server()