首页 > 代码库 > squid重定向(python 代码)

squid重定向(python 代码)

#!/usr/bin/python2.7

# coding: utf-8
import sys,os,stat
import transport
def redirect_url(line,concurrent):
  list = line.split(‘ ‘)
  if concurrent:
    old_url = list[0]
  else:
    old_url = list[1]
 # f = open(‘/etc/squid3/abc.txt‘,‘w‘)
 # f.write(‘old_url:‘+old_url)
 # f.close()
  new_url = ‘\n‘
  if old_url.endswith(‘.avi/‘):
    new_url = ‘http://www.baidu.com/‘ + new_url
   # f = open(‘/etc/squid3/abc.txt‘,‘w‘)
   # f.write(‘old_url:‘+old_url)
   # f.close()
  elif old_url.endswith(‘.exe/‘):
    new_url = ‘http://www.google.com/‘ + new_url
  return new_url
 
def main(concurrent=True):
  line = sys.stdin.readline().strip()
  while line:
       # f = open(‘/etc/squid3/abc.txt‘,‘w‘)
       # f.write(‘line:‘+line)
       # f.close()
    new_url = redirect_url(line,concurrent)
    id=‘‘
    if concurrent:
      id+=line.split(‘ ‘)[0]+‘ ‘
    new_url = id + new_url
   # f = open(‘/etc/squid3/abc.txt‘,‘w‘)
   # f.write(‘new_url:‘+ new_url)
   # f.close()
    sys.stdout.write(new_url)
    sys.stdout.flush()
    line = sys.stdin.readline().strip()
if __name__==‘__main__‘:
  main(len(sys.argv)>1 and sys.argv[1]==‘-c‘)

 

squid重定向(python 代码)