Miscellaneous Notes

#!/usr/bin/env python3
from http.server import SimpleHTTPRequestHandler, HTTPServer
import os

class FileUploadHTTPRequestHandler(SimpleHTTPRequestHandler):
    def do_GET(self):
        self.send_response(200)
        self.send_header("Content-Type", "text/html")
        self.end_headers()
        self.wfile.write(b"""
<form method="POST" enctype="multipart/form-data">
<input type="file" name="file">
<input type="submit">
</form>
""")

    def do_POST(self):
        ctype = self.headers.get("Content-Type", "")
        length = int(self.headers.get("Content-Length", 0))
        data = self.rfile.read(length)

        if ctype.startswith("multipart/form-data"):
            boundary = ctype.split("boundary=")[1].encode()
            parts = data.split(b"--" + boundary)
            for p in parts:
                if b"Content-Disposition" in p:
                    head, body = p.split(b"\r\n\r\n", 1)
                    name = head.split(b'filename="')[1].split(b'"')[0].decode()
                    with open(name, "wb") as f:
                        f.write(body.rstrip(b"\r\n--"))
                    break
        else:
            name = self.headers.get("filename", "upload.bin")
            with open(name, "wb") as f:
                f.write(data)

        self.send_response(200)
        self.end_headers()
        self.wfile.write(b"OK")

if __name__ == "__main__":
    httpd = HTTPServer(("0.0.0.0", 8080), FileUploadHTTPRequestHandler)
    httpd.serve_forever()