main.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
import os
import re
import urllib3
import certifi
import codecs

from argparse import ArgumentParser

def parse_args():
    parser = ArgumentParser()
    parser.add_argument('-i', '--input', dest='input', default=os.path.join('data','whitelist.pac'),
        help='path to gfwlist')
    parser.add_argument('-o', '--output', dest='output', default='whitelist.pac',
        help='path to output pac', metavar='PAC')
    parser.add_argument('-p', '--proxy', dest='proxy', default='"SOCKS5 127.0.0.1:1080; SOCKS 127.0.0.1:1080;"',
        help='the proxy parameter in the pac file, for example,\
        "127.0.0.1:1080;"', metavar='SOCKS5')
    return parser.parse_args()

def writefile(input_file, proxy, output_file):

    domains_content = final_list()
    proxy_content = get_file_data(input_file)
    proxy_content = proxy_content.replace('__PROXY__', proxy)
    proxy_content = proxy_content.replace('__DOMAINS__', domains_content)

    with open(output_file, 'w') as file_obj:
        file_obj.write(proxy_content)


def get_list():
    print('Getting domain whitelist...')
    dnsmasq_china_list = 'https://raw.githubusercontent.com/felixonmars/dnsmasq-china-list/master/accelerated-domains.china.conf'
    whitelist = []
    try:
        content = getList(dnsmasq_china_list)
        content = content.decode('utf-8')
        f = codecs.open('whitelistCache', 'w', 'utf-8')
        f.write(content)
        f.close()

    except:
        print('Get list update failed,use cache to update instead.')

    f = codecs.open('whitelistCache', 'r', 'utf-8')
    for line in f.readlines():
        l = re.findall(r'(?<==/).+?(?=/)', line)
        whitelist.append('"'+l[0]+'":1,')
    f.close()

    return whitelist

def getList(listUrl):
    http = urllib3.PoolManager(
        cert_reqs='CERT_REQUIRED',  # Force certificate check.
        ca_certs=certifi.where(),  # Path to the Certifi bundle.
    )

    data = http.request('GET', listUrl, timeout=10).data
    return data


def final_list():

    list_result = get_list()
    content = '\n'.join(list_result)
    content = '{\n' + content + '\n"yourdomainhere.com":1\n}'
    print('All done!')
    return content


def get_file_data(filename):
    content = ''
    with open(filename, 'r') as file_obj:
        content = file_obj.read()
    return content


def main():
    args = parse_args()
    writefile(args.input, '"' + args.proxy.strip('"') + '"', args.output)

if __name__ == '__main__':
    main()