fail 发表于 2015-9-3 13:32:05

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
else:
    old_url = list
# 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(' ')+' '
    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=='-c')  
页: [1]
查看完整版本: squid重定向(python 代码)