from lazy.www import c
from lxml import etree
from StringIO import StringIO
from sync_utils import *

class position(object):
    def __init__(self, build_type, city_id, idx, baseurl):
        self._baseurl = baseurl + '/index.php'
        self.build_type = build_type
        self.city_id = city_id
        self.idx = idx
        self._params = {'view': 'buildingGround',
                        'id': city_id,
                        'position': idx}
        pass

    def get_page(self):
        page = c(self._baseurl).get(self._params).get_content()
        return page

    def sync(self):
        page = self.get_page()
        parser = etree.HTMLParser(encoding='utf8')
        page_dom = etree.parse(StringIO(page), parser)

        self._sync(page_dom)
        pass
    pass

class building(position):
    class_patterns = {
        'level': 'buildingLevel'
        }
    appear_patterns = {
        'is_upgrading': 'isUpgrading'
        }

    def _sync(self, page_dom):
        sync_tagclass(self, building.class_patterns, page_dom)

        sync_tagclass_start_appear(self, building.appear_patterns, page_dom)

        xpath_upgrade = '/descendant::ul[@class=\'actions\']/li[@class=\'upgrade\']/a'
        anodes = page_dom.xpath(xpath_upgrade)
        if len(anodes) == 1:
            anode = anodes[0]
            self.upgrade_uri = anode.get('href')
        else:
            self.upgrade_uri = None
            pass

        if self.is_upgrading:
            xpath_countdown = '/descendant::div[@id=\'upgradeCountDown\']/text()'
            value = page_dom.xpath(xpath_countdown)[0]
            self.upgrade_countdown = value
        else:
            self.upgrade_countdown = None
            pass
        pass

    def upgrade(self):
        url = self._baseurl + self.upgrade_uri
        page = c(url).get().get_content()
        pass
    pass

class townhall(building):
    class_patterns = {
        'occupied': 'value occupied',
        'rooms': 'value total',
        }
    value_patterns = {
        'growth': 'growth',
        'happiness': 'happiness',
        'interest_base': 'base',
        'interest_research': 'research1',
        'interest_capital': 'capital',
        'overpopulation': 'cat overpopulation'
        }
    count_patterns = {
        'pop_citizens': 'citizens',
        'pop_woodworkers': 'woodworkers',
        'pop_specialworkers': 'specialworkers',
        'pop_scientists': 'scientists'
        }
    
    def __init__(self, city_id, idx, baseurl):
        super(townhall, self).__init__('townhall', city_id, idx, baseurl)
        pass

    def _sync(self, page_dom):
        sync_tagclass(self, townhall.class_patterns, page_dom)

        sync_tagvalue(self, townhall.value_patterns, page_dom)

        sync_tagcount(self, townhall.count_patterns, page_dom)

        super(townhall, self)._sync(page_dom)
        pass
    pass

class academy(building):
    def __init__(self, city_id, idx, baseurl):
        super(academy, self).__init__('academy', city_id, idx, baseurl)
        pass

    def _sync(self, page_dom):
        xpath_research_name = '/descendant::*[@class=\'researchName\']/a'
        anodes = page_dom.xpath(xpath_research_name)
        if len(anodes) == 1:
            anode = anodes[0]
            self.researching = anode.get('title')
            xpath_countdown = '/descendant::div[@id=\'researchCountDown\']/text()'
            txtnodes = page_dom.xpath(xpath_countdown)
            self.researching_countdown = txtnodes[0]
        else:
            self.researching = None
            self.researching_countdown = None
            pass

        super(academy, self)._sync(page_dom)
        pass
    pass

class warehouse(building):
    def __init__(self, city_id, idx, baseurl):
        super(warehouse, self).__init__('warehouse', city_id, idx, baseurl)
        pass
    pass

class barracks(building):
    def __init__(self, city_id, idx, baseurl):
        super(barracks, self).__init__('barracks', city_id, idx, baseurl)
        pass
    pass

class branchoffice(building):
    def __init__(self, city_id, idx, baseurl):
        super(branchoffice, self).__init__('branchoffice', city_id, idx, baseurl)
        pass
    pass

class port(building):
    def __init__(self, city_id, idx, baseurl):
        super(port, self).__init__('port', city_id, idx, baseurl)
        pass
    pass

class wall(building):
    def __init__(self, city_id, idx, baseurl):
        super(wall, self).__init__('wall', city_id, idx, baseurl)
        pass
    pass