json2python.py

#!/usr/bin/env python
#
#  Copyright 2012, 2013 Nick Galbreath
#  nickg@client9.com
#  BSD License -- see COPYING.txt for details
#

"""
Converts a libinjection JSON data file to python dict
"""

def toc(obj):
    """ main routine """

    print("""
import libinjection

def lookup(state, stype, keyword):
    keyword = keyword.upper()
    if stype == libinjection.LOOKUP_FINGERPRINT:
        if keyword in fingerprints and libinjection.sqli_not_whitelist(state):
            return 'F'
        else:
            return chr(0)
    return words.get(keyword, chr(0))

""")

    words = {}
    keywords = obj['keywords']
    for k,v in keywords.items():
        words[str(k)] = str(v)

    print('words = {')
    for k in sorted(words.keys()):
        print("'{0}': '{1}',".format(k, words[k]))
    print('}\n')


    keywords = obj['fingerprints']
    print('fingerprints = set([')
    for k in sorted(keywords):
        print("'{0}',".format(k.upper()))
    print('])')

    return 0

if __name__ == '__main__':
    import sys
    import json
    sys.exit(toc(json.load(sys.stdin)))