setup_db.py

from queries import *
import settings


domains = settings.DOMAINS
nxdomains = settings.NXDOMAINS
google_api_key = settings.GOOGLE_API_KEY
path_elements = ['about','feedback','static','terms','articles','images',\
                 'tags','traffic']
pages = ['index.html','contact.php','post.jsp','submit.aspx']

print '[x] Adding domains'
for d in domains:
    add_canary_domain(domain=d)
    print '\t{domain}'.format(domain=d)

print '[x] Adding NX domains'
for d in nxdomains:
    add_canary_nxdomain(domain=d)
    print '\t{domain}'.format(domain=d)

print '[x] Adding path elements'
for pe in path_elements:
    add_canary_path_element(path_element=pe)
    print '\t{pe}'.format(pe=pe)

print '[x] Adding pages'
for p in pages:
    add_canary_page(page=p)
    print '\t{p}'.format(p=p)

print '[x] Adding google api key'
for k in google_api_key:
    add_canary_google_api_key(key=k)
    print '\t{k}'.format(k=k)