import pycurl
import csv
from mydb import cursor
from MySQLdb import escape_string

class Body:
    def __init__(self):
        self.contents = ''

    def gotData(self, buffer):
        self.contents = self.contents + buffer

login_url = "http://localhost/handleLogin.php"
insert_url = "http://localhost/actions/populateAdmin.php"
cookie_file = "/tmp/wl_cookie"

try:
    csvfile = sys.argv[1]
except:
    print "Usage %s CSVfile" % sys.argv[0]
    sys.exit(1)

# Login in to get a valid session
curl = pycurl.Curl()
body = Body()

payload = [("username", "sts"), ("passwd", "msts")]
try:
    curl.setopt(curl.URL, login_url)
    curl.setopt(curl.COOKIEJAR, cookie_file)
    curl.setopt(curl.HTTPPOST, payload)
    curl.setopt(curl.WRITEFUNCTION, body.gotData)
    curl.perform()
    curl.close()
    print body.contents
except:
    print "Local Curl Failed"
    print "URL: ", insert_url
    print "Payload: ", payload
    print "Result: ", body.contents
    sys.exit(2)

reader = csv.reader(open(csvfile, 'r'))

for row in reader:

   try:
        **(location_id, vrm, valid_datetime) =  [x.decode('utf-8-sig') for x in row ]**
    except:
        print "Error with row: " % row
        sys.exit(3)

    **# Hack for Peel
      if (location_id == 295):
        location_id = "295"**

   curl = pycurl.Curl()
    body = Body()

    payload = [("fn", "addWhitelistEntryValue"),
        ("firstname", "WL"),
        ("surname", "USER"),
        ("vrm", vrm.encode('ascii', 'ignore')),
        ("locId", location_id.encode('ascii', 'ignore'))
        ]
    print payload