public_git
/
nginx-ldap-auth
mirror of https://github.com/nginxinc/nginx-ldap-auth.git
1
0
Fork 1
Code Issues Releases Wiki Activity

Add Python3 compatibility to sample app

This commit is contained in:
Robert Szulist 2019-09-08 15:00:31 +02:00
parent 8bd5c3ae21
commit a96bbe6a57
1 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
backend-sample-app.py
View File

@ -8,13 +8,29 @@
# 1) accepts GET requests on /login and responds with a login form # 1) accepts GET requests on /login and responds with a login form
# 2) accepts POST requests on /login, sets a cookie, and responds with redirect # 2) accepts POST requests on /login, sets a cookie, and responds with redirect
import sys, os, signal, base64, Cookie, cgi, urlparse import sys, os, signal, base64, cgi
from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler if sys.version_info.major == 2:
from urlparse import urlparse
from Cookie import BaseCookie
from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler
elif sys.version_info.major == 3:
from urllib.parse import urlparse
from http.cookies import BaseCookie
from http.server import HTTPServer, BaseHTTPRequestHandler
Listen = ('localhost', 9000) Listen = ('localhost', 9000)
import threading import threading
from SocketServer import ThreadingMixIn if sys.version_info.major == 2:
from SocketServer import ThreadingMixIn
elif sys.version_info.major == 3:
from socketserver import ThreadingMixIn
def ensure_bytes(data):
return data if sys.version_info.major == 2 else data.encode("utf-8")
class AuthHTTPServer(ThreadingMixIn, HTTPServer): class AuthHTTPServer(ThreadingMixIn, HTTPServer):
pass pass
@ -22,14 +38,14 @@ class AppHandler(BaseHTTPRequestHandler):
def do_GET(self): def do_GET(self):
url = urlparse.urlparse(self.path) url = urlparse(self.path)
if url.path.startswith("/login"): if url.path.startswith("/login"):
return self.auth_form() return self.auth_form()
self.send_response(200) self.send_response(200)
self.end_headers() self.end_headers()
self.wfile.write('Hello, world! Requested URL: ' + self.path + '\n') self.wfile.write(ensure_bytes('Hello, world! Requested URL: ' + self.path + '\n'))
# send login form html # send login form html
@ -69,7 +85,7 @@ class AppHandler(BaseHTTPRequestHandler):
self.send_response(200) self.send_response(200)
self.end_headers() self.end_headers()
self.wfile.write(html.replace('TARGET', target)) self.wfile.write(ensure_bytes(html.replace('TARGET', target)))
# processes posted form and sets the cookie with login/password # processes posted form and sets the cookie with login/password
@ -102,8 +118,8 @@ class AppHandler(BaseHTTPRequestHandler):
# and share a key with auth daemon that extracts this information # and share a key with auth daemon that extracts this information
# #
# WARNING WARNING WARNING # WARNING WARNING WARNING
enc = base64.b64encode(user + ':' + passwd) enc = base64.b64encode(ensure_bytes(user + ':' + passwd))
self.send_header('Set-Cookie', 'nginxauth=' + enc + '; httponly') self.send_header('Set-Cookie', b'nginxauth=' + enc + b'; httponly')
self.send_header('Location', target) self.send_header('Location', target)
self.end_headers() self.end_headers()