from i3pystatus import IntervalModule
import requests
from collections import OrderedDict
from bs4 import BeautifulSoup


class WhosOnLocation():
    email = None
    password = None
    session = None

    def __init__(self, email, password):
        self.email = email
        self.password = password
        self.session = requests.Session()

    def login(self):
        login_details = {'email_input': self.email,
                         'password_input': self.password,
                         '_redirect_url': '',
                         'continue_submit': 'Login'}
        r = self.session.post('https://login.whosonlocation.com/login', data=login_details)
        return r.url == 'https://au.whosonlocation.com/home?justloggedin=true'

    def get_status(self):
        r = self.session.get('https://au.whosonlocation.com/home?justloggedin=true')
        html = BeautifulSoup(r.content)
        status = html.body.find("span", {"class": "my-status-name"})
        if status:
            return status.text

    def on_site(self):
        return self.__change_status('onsite')

    def off_site(self):
        return self.__change_status('offsite')

    def __change_status(self, status):
        r = self.session.post('https://au.whosonlocation.com/ajax/changestatus', data={'status': status})
        return r.json()

    # _type can be org or location
    def search(self, keyword, _type='location'):
        payload = {'keyword': keyword, 'type': _type}
        r = self.session.get('https://au.whosonlocation.com/home/search', params=payload)
        return self.__parse_results(BeautifulSoup(r.content))

    @staticmethod
    def __parse_results(page):
        titles = ['Name', 'Title', 'Department', 'Current Location', 'Home Location']
        table = page.body.find_all("tr", {"class": "dataRow"})
        results = []
        for row in table:
            values = [v.string for v in row.findAll('td', {'class': 'truncate'})]
            results.append(OrderedDict(zip(titles, values)))
        return results


[docs]class WOL(IntervalModule):
    """
    Change your whosonlocation.com status.

    Requires the PyPi module `beautifulsoup4`
    """
    location = None
    email = None
    password = None

    settings = (
        ('keyring_backend', 'alternative keyring backend for retrieving credentials'),
        'email',
        'password'
    )
    keyring_backend = None

    color_on_site = '#00FF00'
    color_off_site = '#ff0000'
    format = 'Status: {status}'
    status = None

    on_leftclick = 'change_status'

    def init(self):
        self.location = WhosOnLocation(self.email, self.password)
        if not self.location.login():
            raise Exception("Failed to login")

    def change_status(self):
        if self.status == 'On-Site':
            self.location.off_site()
        elif self.status == 'Off-Site':
            self.location.on_site()

    def run(self):
        self.status = self.location.get_status()
        color = None

        if self.status == 'Off-Site':
            color = self.color_off_site
        elif self.status == 'On-Site':
            color = self.color_on_site

        self.output = {
            "full_text": self.format.format(
                status=self.status
            ),
            "color": color
        }