Skip to content

Instantly share code, notes, and snippets.

@Xatpy
Last active August 29, 2015 14:17
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Xatpy/6820e1aa0a679eefc0a7 to your computer and use it in GitHub Desktop.
Save Xatpy/6820e1aa0a679eefc0a7 to your computer and use it in GitHub Desktop.
Selenium - Import world_borders and check styles
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time
#GLOBAL VARIABLE
DEBUG = 1
def debugger(message):
"""debug function. Prints the message given by parameter"""
if DEBUG:
print(message)
driver = webdriver.Firefox()
debugger('Moving window to my second monitor')
#driver.maximize_window()
driver.set_window_position(-110,-600)
driver.maximize_window()
driver.delete_all_cookies()
driver.get('https://xatpy.cartodb.com/login')
email = driver.find_element_by_id("email")
password = driver.find_element_by_id("password")
password.send_keys("MY_PASSWORD")
password.send_keys(Keys.RETURN)
driver.implicitly_wait(300)
debugger('Click on dropdown datasets-maps')
driver.find_element(By.CSS_SELECTOR, ".Header-navigationBreadcrumbLink.DropdownLink.DropdownLink--white").click()
debugger('Click datasets')
driver.find_element(By.CSS_SELECTOR, ".BreadcrumbsDropdown-optionsItem.has-margin").click()
debugger('Click on CONNECT DATASET')
driver.find_element(By.CSS_SELECTOR, ".Button.Button--positive.js-new_dataset").click()
fileCSV = "/Users/jaimechapinal/workspace/datasets/world_borders.csv"
debugger('Uploading file...: ' + fileCSV)
driver.find_element_by_css_selector('.js-fileInput').send_keys(fileCSV)
debugger('Click on connect')
driver.find_element_by_css_selector('.Button.Button--positive.CreateDialog-footerActionsButton.js-connect').click()
debugger('Waiting for until map is active and I can click on wizard...')
element = WebDriverWait(driver, 500).until(
EC.presence_of_element_located((By.CLASS_NAME,'wizards_mod'))
)
debugger('Click on wizards')
element.click()
driver.save_screenshot('simple.png')
debugger('Click on choropleth and wait')
driver.find_element_by_css_selector('.choropleth.polygon').click()
time.sleep(5)
debugger('Save choropleth screenshot')
driver.save_screenshot('cluster.png')
debugger('Click on category and wait')
driver.find_element_by_css_selector('.category.polygon').click()
time.sleep(5)
debugger('Save category screenshot')
driver.save_screenshot('category.png')
debugger('Click on bubble and wait')
driver.find_element_by_css_selector('.bubble.polygon').click()
time.sleep(5)
debugger('Save bubble screenshot')
driver.save_screenshot('bubble.png')
debugger('THE END.')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment