Getfit automation

getfit.py

from selenium.webdriver import Firefox
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.firefox.options import Options

import time

def getfit_submit(user, pwd, minutes, activity, intensity):
    URL = 'https://getfit.mit.edu/user/login?current=team-fitness/home'
    options = Options()
    options.headless = True
    driver = Firefox(options=options)
    driver.get(URL)
    minutes = str(int(minutes))

    def enter(eid, keys):
        elem = driver.find_element_by_id(eid)
        elem.send_keys(keys)
        return elem

    e = enter("edit-name", user)
    e = enter("edit-pass", pwd)
    e.send_keys(Keys.RETURN)

    time.sleep(5)

    e = enter("duration", minutes)
    e = enter("activity-name", activity)
    e.send_keys(Keys.RETURN)
    e = enter("intensity-tid", intensity)
    e = driver.find_element_by_class_name('tf-button')
    e.click()
    time.sleep(1)
    driver.close()

# Credentials
user = 'USER'
pwd = 'GETFIT_PASSWORD'

# Example
getfit_submit(user, pwd, 30, 'Running', 'm')