Created
June 1, 2020 11:03
-
-
Save vpetersson/2607bc47154dee7801f76af8dd81acaa to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/env python3 | |
import http.server | |
import socketserver | |
import io | |
import cgi | |
# Source: https://stackoverflow.com/questions/39788591/python-simplehttpserver-to-receive-files | |
# Upload using `curl -F 'file=@<FILENAME>' http://host:44444/` | |
# Change this to serve on a different port | |
PORT = 44444 | |
class CustomHTTPRequestHandler(http.server.SimpleHTTPRequestHandler): | |
def do_POST(self): | |
r, info = self.deal_post_data() | |
print(r, info, "by: ", self.client_address) | |
f = io.BytesIO() | |
if r: | |
f.write(b"Success\n") | |
else: | |
f.write(b"Failed\n") | |
length = f.tell() | |
f.seek(0) | |
self.send_response(200) | |
self.send_header("Content-type", "text/plain") | |
self.send_header("Content-Length", str(length)) | |
self.end_headers() | |
if f: | |
self.copyfile(f, self.wfile) | |
f.close() | |
def deal_post_data(self): | |
ctype, pdict = cgi.parse_header(self.headers['Content-Type']) | |
pdict['boundary'] = bytes(pdict['boundary'], "utf-8") | |
pdict['CONTENT-LENGTH'] = int(self.headers['Content-Length']) | |
if ctype == 'multipart/form-data': | |
form = cgi.FieldStorage( fp=self.rfile, headers=self.headers, environ={'REQUEST_METHOD':'POST', 'CONTENT_TYPE':self.headers['Content-Type'], }) | |
print (type(form)) | |
try: | |
if isinstance(form["file"], list): | |
for record in form["file"]: | |
open("./%s"%record.filename, "wb").write(record.file.read()) | |
else: | |
open("./%s"%form["file"].filename, "wb").write(form["file"].file.read()) | |
except IOError: | |
return (False, "Can't create file to write, do you have permission to write?") | |
return (True, "Files uploaded") | |
Handler = CustomHTTPRequestHandler | |
with socketserver.TCPServer(("", PORT), Handler) as httpd: | |
print("serving at port", PORT) | |
httpd.serve_forever() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment