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()