from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time


class Triburile_Login():
    def __init__(self, username, password):
        self.browser = webdriver.Firefox()
        self.username = username
        self.password = password

    def logIn(self):
        self.browser.get("https://www.triburile.ro/")
        time.sleep(1)
        usernameToLog = self.browser.find_element_by_id('user')
        usernameToLog.send_keys(self.username)
        passwordToLog = self.browser.find_element_by_name('password')
        passwordToLog.send_keys(self.password)
        self.browser.find_element_by_class_name('btn-login').click()
        time.sleep(1)
        self.browser.find_element_by_class_name('world_button_active').click()

class Upgrade_Village(Triburile_Login):
    def getResourceInfo(self):
        self.wood = self.browser.find_element_by_id('wood').getText()
        self.stone = self.browser.find_element_by_id('stone').getText()
        self.iron = self.browser.find_element_by_id('iron').getText()
        print (self.wood)

loginBot = Triburile_Login('viclenie','fantomas')
loginBot.logIn()

upgradeBot = Upgrade_Village('viclenie','fantomas')
upgradeBot.getResourceInfo()