#!/usr/bin/env python

import sys
sys.path.append("/var/www/vhosts/realintelligence.com/cgi-bin/xmlcreator/")

import xmleventlist
# import tsp
import cgi
import cPickle as pickle
import cgi_util

query_dict = cgi.FieldStorage()
orgId = str(query_dict['orgId'].value)

try:
    with open("./data.pkl", "rb") as pkl_file:
        pkl_data = pickle.load(pkl_file)
        org_creds = __import__(orgId)
        if pkl_data[orgId] and pkl_data[orgId]['username']:
            sandbox = cgi_util.get_param_bool("sandbox", False)
            xmleventlist.get_events(pkl_data[orgId]['username'], pkl_data[orgId]['password'], sandbox)
        else:
            raise KeyError 
except IOError:
    with open("./data.pkl", "w+b") as pkl_file:
        data = {orgId: {'username': '', 'password': ''}}
        pickle.dump(data, pkl_file)
        print "Content-Type: text/html"
        print 
        print "Pickle file doesn't exist. Creating. Reload"  

    with open("./data.pkl", "rb") as new_pkl_file:
        pkl_data = pickle.load(new_pkl_file)
        sandbox = cgi_util.get_param_bool("sandbox", False)
        xmleventlist.get_events(pkl_data[orgId]['username'], pkl_data[orgId]['password'], sandbox)
except KeyError:
    with open("./data.pkl", "w+b") as pkl_file:
        pkl_data[orgId] = {"username": org_creds.username, "password": org_creds.password}
        pickle.dump(pkl_data, pkl_file)
    with open("./data.pkl", "rb") as new_pkl_file:
        pkl_data = pickle.load(new_pkl_file)
        sandbox = cgi_util.get_param_bool("sandbox", False)
        xmleventlist.get_events(pkl_data[orgId]['username'], pkl_data[orgId]['password'], sandbox)
except ImportError:
    with open("./data.pkl", "w+b") as pkl_file:
        pkl_data[orgId] = {"username": '', "password": ''}
        pickle.dump(pkl_data, pkl_file)
        print "Content-Type: text/html"
        print 
        print "Written to orgId pickle" + orgId