Subversion Repositories SmartDukaan

from selenium import webdriver

from selenium.webdriver.common.keys import Keys

import time

#from dtr.utils.utils import get_mongo_connection

import urllib2

import datetime

USER_NAME = "1202479977"

PASSWORD = "5409"

#VOI_USER_NAME = "spiceonlineretailvoi@gmail.com"

#VOI_PASSWORD = "saholicsnapdeal1"

headers = {

"Accept":"*/*",

"Accept-Encoding":"gzip,deflate",

"Accept-Language":"en-US,en;q=0.8",

"Connection":"keep-alive",

"Host":"dial2verify.com",

"Origin":"https://dial2verify.com",

"Referer":"https://dial2verify.com/corp/control-panel/v2/Dial2Verify_UI_Data_export.php",

"User-Agent":"Mozilla/5.0 (X11; Linux i686) AppleWebKit/537.36 (KHTML, like Gecko) Ubuntu Chromium/37.0.2062.120 Chrome/37.0.2062.120 Safari/537.36",

}

def run():

    try:

        driver = webdriver.PhantomJS('/usr/local/bin/phantomjs')

        driver.set_window_size(1124, 850) # set browser size.

        driver.get("https://dial2verify.com/corp/control-panel/v2/login.php")

        driver.maximize_window()

        element = driver.find_element_by_name("username")

        element.send_keys(USER_NAME)

        element = driver.find_element_by_name("password")

        element.send_keys(PASSWORD)

        element = driver.find_element_by_css_selector(".runner-button")

        element.click()

        time.sleep(5)

        driver.get("https://dial2verify.com/corp/control-panel/v2/Dial2Verify_UI_Data_export.php")

        time.sleep(5)

        element = driver.find_element_by_css_selector(".runner-button")

        element.click()

        cookieMap = {}

        cookie = ""

        for x in driver.get_cookies():

            if x.get('name') == "lhc_per":

                cookieMap['lhc_per'] = x.get('value')

            elif x.get('name') == "PHPSESSID":

                cookieMap['PHPSESSID'] = x.get('value')

            else:

                pass

        cookie = "lhc_per="+cookieMap['lhc_per']+";PHPSESSID="+cookieMap['PHPSESSID']

        return cookie

    finally:

        print "Closing driver "

        driver.quit()

def downloadFile(cookie):

    headers['cookie'] = cookie

    url = "https://dial2verify.com/corp/control-panel/v2/Dial2Verify_UI_Data_export.php?type=excel2007&records=all&rndVal=0.1899"

    req = urllib2.Request(url,headers=headers)

    res = urllib2.urlopen(req)

    filePath = "/tmp/dial2verify"+str(datetime.datetime.now())+".xlsx"

    output = open(filePath,'wb')

    output.write(res.read())

    output.close()

    return filePath

def main():

    cookie = run()

    return downloadFile(cookie)

if __name__ == '__main__':

    main()