squid custom rewriter script

custom_rewrite.py

#!/usr/bin/python

import sys
import re

port_regex = re.compile(r':8082')

def rewrite_url(line):
    """ If the input URL contains a port number, strip it out """
    url_list = line.split(' ')
    old_url = url_list[0]
    new_url = '\n'
    if port_regex.search(old_url):
        new_url = re.sub(port_regex, '', old_url, count=1) + new_url
    return new_url

while True:
    line = sys.stdin.readline().strip()
    new_url = rewrite_url(line)
    sys.stdout.write(new_url)
    sys.stdout.flush()