extract-external-links.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*- 

import sys
from zipfile import ZipFile

from extractor import KPP
from bundle import Bundle

def process(filename, rtypes=None):
    if not rtypes:
        rtypes = ['requiredBrushFile']

    def process_kpp(kpp, bundle=None):
        links = kpp.get_links()
        for name, value in links.iteritems():
            if bundle is not None:
                if name not in rtypes:
                    continue
                if name == 'requiredBrushFile':
                    ok = bundle.find_brush(value)
                    if ok:
                        print("required brush file {} found in the same bundle.".format(value))
                        continue
                if not value:
                    continue
            print(u"{}: {}".format(name, value).encode('utf-8'))

    if filename.endswith(".bundle"):
        bundle = Bundle.open(filename)
        presets = bundle.presets_data
    else:
        bundle = None
        presets = [KPP(filename)]

    for preset in presets:
        process_kpp(preset, bundle)


for fname in sys.argv[1:]:
    print(u"Processing: {}".format(fname))
    process(fname)