Python to rework Amazon URL to contain the user's affiliate code

makeAmazonAffUrl.py

#!/usr/bin/env python

# (c) 2012 Brett Kelly.
# Licensed under the MIT license 
# http://www.opensource.org/licenses/mit-license.php 

import re
import sys
from urlparse import urlparse

## PUT YOUR AFFILIATE CODE HERE
affcode = 'nerdgap-20'

rawurl = '%clipboard'.strip()

## Get the base url without all of the query string malarky
baseurl = rawurl.split('?')[0] 

try:
    parts = urlparse(baseurl)
except Exception:
    sys.stdout.write(rawurl)
    raise SystemExit

## Make sure it's actually an Amazon URL
amazonre = re.compile('[www\.]{0,1}amazon\.')

if not amazonre.search(parts.netloc):
    # Not an amazon URL, return whatever was passed initially
    sys.stdout.write(rawurl)
    raise SystemExit

# Format the simpler URL and append the affiliate code
goodurl = "%s://%s%s?tag=%s" % (parts.scheme,parts.netloc,parts.path,affcode)

# Write formatted URL to STDOUT
sys.stdout.write(goodurl)