#!/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.iteritems():
 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)))