#!/usr/bin/env python
import cgi, urllib, sys
#import cgitb
version = '0.9.1'
year = '2004'
vendor = 'WHGDev'
def error(about, title, text):
return """
""" + title + """""" + stylesheet() + """
""" + about + """
""" + text + """
""" + ver() + """
"""
def ver():
"""Returns the version"""
return 'relocate ' + version + ' GPLd ' + year + ' by ' + vendor
def stylesheet():
"""Returns the stylesheet"""
return """"""
def initpage(mime="text/plain", disposition=None):
"""This function prepares the HTTP Connection to
send proper headers - content type and content disposition"""
print 'Content-Type: %s' % mime
if disposition != None:
# we got some content dispositon
print 'Content-Disposition: attachment; filename="%s"' % disposition
print
def getfile(url):
"""Opens a remote file and returns a tuple with
a file-like object and with the content type"""
try:
# open the file
remote = urllib.urlopen(url)
# get the content type
mime = remote.headers.gettype()
return (remote, mime)
except IOError:
# Well, no such file found
raise IOError('No such file or directory')
def main():
#cgitb.enable()
# get the HTTP GET Arguments
form = cgi.FieldStorage()
# get url field
try:
url = form['url'].value
except KeyError:
url = None
# get extension field
try:
ext = form['ext'].value
except KeyError:
ext = None
if not url or not ext:
initpage('text/html')
print error(
"Arguments missing",
"Too less arguments given, Sai",
"You need to specify both url and ext arguments.")
sys.exit(100)
# now get the names
import os.path
fileprefix = os.path.splitext(os.path.basename(url))[0]
ext = '.' + ext
dispositionname = fileprefix + ext
# now dispositionname is the new name of the file with the new suffix
# okay, let's start work!
try:
# we try to get that URL
remotefile = getfile(url)
# well done, now initialise the page with the proper
#+Content Type (determined by the content type of the URL)
#+and the proper conten disposition
initpage(mime=remotefile[1], disposition=dispositionname)
if sys.platform == "win32":
# well, on Windows systems the stdout does not really like binary
#+so we need to switch the stdout to binary mode
import os, msvcrt
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
# all prerequisites done, lets start loading the file
sys.stdout.write(remotefile[0].read())
except IOError:
# no such file found
initpage('text/html')
print error("File not found",
"No such file, Sai",
"File %s was not found, check it exists." % url)
if __name__ == '__main__':
main()