SimpleHTTPServer with custom headers (Python3)

httpd.py

#!/usr/bin/env python3

import sys

from http.server import (
        SimpleHTTPRequestHandler,
        HTTPServer,
        test,
)

"""
Usage:
    python3 httpd.py [port] [additional headers ...]

Example:
    python3 httpd.py 8000 'Pragma: no-cache' 'Cache-Control: no-cache' 'Expires: 0'
"""

class CustomHTTPRequestHandler(SimpleHTTPRequestHandler):
    def end_headers(self):
        self.send_new_headers()

        SimpleHTTPRequestHandler.end_headers(self)

    def send_new_headers(self):
        for i in sys.argv[2:]:
            key, value = i.split(":", 1)
            self.send_header(key, value)

if __name__ == '__main__':
    test(HandlerClass=CustomHTTPRequestHandler, ServerClass=HTTPServer, protocol="HTTP/1.1")