import datetime as dt
from pytz import timezone
import pytz
import django_wsgi
from django.conf import settings
from django.db import IntegrityError

from account.models import ParselyAPIKey, Publisher,Tier, ElasticSearchSettings

apikey_infos = [
    "thesouthern.com",
    "herald-review.com",
    "trib.com",
    "wiscnews.com",
    "jg-tc.com",
    "globegazette.com",
    "rapidcityjournal.com",
    "magicvalley.com",
    "cumberlink.com",
    "santamariatimes.com",
    "syvnews.com",
    "dailyjournalonline.com",
    "siouxcityjournal.com",
    "wcfcourier.com",
    "auburnpub.com",
    "lompocrecord.com",
    "mtstandard.com",
    "helenair.com",
    "beatricedailysun.com",
    "columbustelegram.com",
    "fremonttribune.com",
    "thetandd.com",
    "theworldlink.com",
    "napavalleyregister.com",
    "muscatinejournal.com",
    "democratherald.com",
    "bismarcktribune.com",
    "gazettetimes.com",
    "journaltimes.com",
    "elkodaily.com",
    "tdn.com",
    "winonadailynews.com",
    "chippewa.com",
    "maysville-online.com",
    "azdailysun.com",
    "lebanon-express.com",
    "hanfordsentinel.com",
    "ravallirepublic.com",
    "missoulian.com",
    "qconline.com",
    "missoulanews.com"
]

def main():
    publisher = Publisher.objects.get(name="Lee Enterprises")
    tier = Tier.objects.get(name="2_analytics_team")
    signup_date = dt.datetime.utcnow()

    for apikey_info in apikey_infos:
        # fix an error in source data
        key = apikey_info.strip("/")
        existing = ParselyAPIKey.objects.filter(key=key)
        if not existing:
            apikey = ParselyAPIKey()
            print("Creating apikey {}".format(key))
        else:
            apikey = existing[0]
            print("Editing existing apikey {}".format(key))
        apikey.key = key
        apikey.publisher = publisher
        apikey.tier = tier
        apikey.is_active = True
        apikey.timezone = timezone('US/Central')
        apikey.signup_date = signup_date
        apikey.track_start_date = signup_date
        apikey.registration_date = signup_date
        apikey.save()
        apikey.es_settings = es_settings
        apikey.save()

if __name__ == "__main__":
    main()