-
-
Save fxsjy/5465353 to your computer and use it in GitHub Desktop.
import BaseHTTPServer | |
from SimpleHTTPServer import SimpleHTTPRequestHandler | |
import sys | |
import base64 | |
key = "" | |
class AuthHandler(SimpleHTTPRequestHandler): | |
''' Main class to present webpages and authentication. ''' | |
def do_HEAD(self): | |
print "send header" | |
self.send_response(200) | |
self.send_header('Content-type', 'text/html') | |
self.end_headers() | |
def do_AUTHHEAD(self): | |
print "send header" | |
self.send_response(401) | |
self.send_header('WWW-Authenticate', 'Basic realm=\"Test\"') | |
self.send_header('Content-type', 'text/html') | |
self.end_headers() | |
def do_GET(self): | |
global key | |
''' Present frontpage with user authentication. ''' | |
if self.headers.getheader('Authorization') == None: | |
self.do_AUTHHEAD() | |
self.wfile.write('no auth header received') | |
pass | |
elif self.headers.getheader('Authorization') == 'Basic '+key: | |
SimpleHTTPRequestHandler.do_GET(self) | |
pass | |
else: | |
self.do_AUTHHEAD() | |
self.wfile.write(self.headers.getheader('Authorization')) | |
self.wfile.write('not authenticated') | |
pass | |
def test(HandlerClass = AuthHandler, | |
ServerClass = BaseHTTPServer.HTTPServer): | |
BaseHTTPServer.test(HandlerClass, ServerClass) | |
if __name__ == '__main__': | |
if len(sys.argv)<3: | |
print "usage SimpleAuthServer.py [port] [username:password]" | |
sys.exit() | |
key = base64.b64encode(sys.argv[2]) | |
test() |
I updated to python 3 but it is not handled properly,
(Updated using 2to3.)
Traceback (most recent call last):
File "SimpleAuthServer.py", line 41, in
key = base64.b64encode (sys.argv [2])
File "/opt/rh/rh-python36/root/usr/lib64/python3.6/base64.py", line 58, in b64encode
encoded = binascii.b2a_base64 (s, newline = False)
TypeError: a bytes-like object is required, not 'str'
Here's a version that's pip installable https://github.com/tianhuil/SimpleHTTPAuthServer and freely licensable version or just run:
Installation
pip install git+git://github.com/tianhuil/SimpleHTTPAuthServer.git@masterUsage
python -m SimpleHTTPAuthServer -h
how would this work with multiple user names and passwords?
I updated to python 3 but it is not handled properly,
(Updated using 2to3.)Traceback (most recent call last):
File "SimpleAuthServer.py", line 41, in
key = base64.b64encode (sys.argv [2])
File "/opt/rh/rh-python36/root/usr/lib64/python3.6/base64.py", line 58, in b64encode
encoded = binascii.b2a_base64 (s, newline = False)
TypeError: a bytes-like object is required, not 'str'
I did not try the 2to3 converted version, but by modifying the 95 line of SimpleHTTPAuthServer/main.py to
SimpleHTTPAuthHandler.KEY = base64.b64encode(args.key.encode("utf-8"))
can make the program start. View here for more info.
(Although some other errors happened.)
Python3 Version, refactored to behave like python3 -m http.serve helper. Added --username and --password for basic HTTP Auth. (Authorization: Basic)
https://gist.github.com/mauler/593caee043f5fe4623732b4db5145a82
Is any chance to support post request as well, probably implement it by BaseHTTPRequestHandler?
Please let me know the update schedule, from python2 to python3.
When will it be updated?