diff pyikriam/buildings.py @ 176:3ba3edda6d1e

Extract more information for building townhall.
author Thinker K.F. Li <thinker@branda.to>
date Sun, 02 Nov 2008 11:32:59 +0800
parents 9f248c8460ce
children 6adad3bcca78
line wrap: on
line diff
--- a/pyikriam/buildings.py	Sun Nov 02 09:59:51 2008 +0800
+++ b/pyikriam/buildings.py	Sun Nov 02 11:32:59 2008 +0800
@@ -19,10 +19,28 @@
     pass
 
 class townhall(position):
-    xpath_patterns = {
-        'level': 'div[@class=\'buildingLevel\']/text()'
+    class_patterns = {
+        'level': 'buildingLevel',
+        'occupied': 'value occupied',
+        'rooms': 'value total',
         }
-
+    value_patterns = {
+        'growth': 'growth',
+        'happiness': 'happiness',
+        'interest_base': 'base',
+        'interest_research': 'research1',
+        'interest_capital': 'capital'
+        }
+    count_patterns = {
+        'pop_citizens': 'citizens',
+        'pop_woodworkers': 'woodworkers',
+        'pop_specialworkers': 'specialworkers',
+        'pop_scientists': 'scientists'
+        }
+    appear_patterns = {
+        'is_upgrading': 'isUpgrading'
+        }
+    
     def __init__(self, city_id, idx, baseurl):
         super(townhall, self).__init__('townhall', city_id, idx, baseurl)
         pass
@@ -30,16 +48,56 @@
     def _sync(self, page):
         parser = etree.HTMLParser(encoding='utf8')
         page_dom = etree.parse(StringIO(page), parser)
-        xpath_building = '/html/body/descendant::'
-        for name, ptn in self.xpath_patterns.items():
-            path = xpath_building + ptn
+
+        xpath_building = '/html/body/descendant::*[@class=\'%s\']/text()'
+        for name, clzname in self.class_patterns.items():
+            path = xpath_building % (clzname)
+            value = page_dom.xpath(path)[0]
+            setattr(self, name, value)
+            pass
+
+        xpath_value = '/html/body/descendant::*[starts-with(@class,\'%s\')]/descendant::*[@class=\'value\']/text()'
+        for name, clzname in self.value_patterns.items():
+            path = xpath_value % (clzname)
+            value = page_dom.xpath(path)[0]
+            setattr(self, name, value)
+            pass
+
+        xpath_count = '/html/body/descendant::*[starts-with(@class,\'%s\')]/descendant::*[@class=\'count\']/text()'
+        for name, clzname in self.count_patterns.items():
+            path = xpath_count % (clzname)
             value = page_dom.xpath(path)[0]
             setattr(self, name, value)
             pass
+
+        xpath_appear = '/html/body/descendant::*[starts-with(@class,\'%s\')]'
+        for name, clzname in self.appear_patterns.items():
+            path = xpath_appear % (clzname)
+            cnt = len(page_dom.xpath(path))
+            if cnt != 0:
+                setattr(self, name, True)
+            else:
+                setattr(self, name, False)
+                pass
+            pass
+
+        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
         pass
 
     def sync(self):
         page = self.get_page()
         self._sync(page)
         pass
+
+    def upgrade(self):
+        url = self._baseurl + self.upgrade_uri
+        page = c(url).get().get_content()
+        pass
     pass