-
-
Save imranity/5674864ab44f9dfb7599a1a98018fa5a to your computer and use it in GitHub Desktop.
Simple Redirect Server in python to redirect requests to a specified URL
This file contains hidden or 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/bin/env python | |
""" | |
Simple HTTP URL redirector | |
Shreyas Cholia 10/01/2015 | |
usage: redirect.py [-h] [--port PORT] [--ip IP] redirect_url | |
HTTP redirect server | |
positional arguments: | |
redirect_url | |
optional arguments: | |
-h, --help show this help message and exit | |
--port PORT, -p PORT port to listen on | |
--ip IP, -i IP host interface to listen on | |
""" | |
import SimpleHTTPServer | |
import SocketServer | |
import sys | |
import argparse | |
def redirect_handler_factory(url): | |
""" | |
Returns a request handler class that redirects to supplied `url` | |
""" | |
class RedirectHandler(SimpleHTTPServer.SimpleHTTPRequestHandler): | |
def do_GET(self): | |
self.send_response(301) | |
self.send_header('Location', url) | |
self.end_headers() | |
return RedirectHandler | |
def main(): | |
parser = argparse.ArgumentParser(description='HTTP redirect server') | |
parser.add_argument('--port', '-p', action="store", type=int, default=80, help='port to listen on') | |
parser.add_argument('--ip', '-i', action="store", default="", help='host interface to listen on') | |
parser.add_argument('redirect_url', action="store") | |
myargs = parser.parse_args() | |
redirect_url = myargs.redirect_url | |
port = myargs.port | |
host = myargs.ip | |
redirectHandler = redirect_handler_factory(redirect_url) | |
handler = SocketServer.TCPServer((host, port), redirectHandler) | |
print("serving at port %s" % port) | |
handler.serve_forever() | |
if __name__ == "__main__": | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment