Finish work module

rival_regions_wrapper/api_wrapper/work.py

@@ -7,6 +7,14 @@ from bs4 import BeautifulSoup
 from . import MIDDLEWARE
 
 
+RESOURCE_DICT = {
+    'oil': 'oil',
+    'ore': 'ore',
+    'yellow': 'gold',
+    'uranium': 'uranium',
+    'diamond': 'diamond'
+}
+
 class Work(object):
     """Wrapper class for profile"""
 
@@ -16,11 +24,43 @@ class Work(object):
         path = 'work'
         response = MIDDLEWARE.get(path)
         soup = BeautifulSoup(response, 'html.parser')
-        factory_header = soup.select_one('.factory_slide')
+
+        factory = soup.select_one('.work_item:nth-child(9)')
+        factory_slide = factory.select_one('.factory_slide')
+        factory_owner = factory.select_one('.factory_whose')
+        factory_dict = {
+            'id': int(factory_slide['action'].replace('factory/index/', '')),
+            'name': factory_slide.text,
+            'owner_name': factory_owner.text,
+            'owner_id': factory_owner['action'].replace('slide/profile/', ''),
+        }
+        level_str = re.search(r'level\s\d+', factory.select_one('.work_source_1').text)
+        if level_str:
+            factory_dict['level'] = int(re.sub(r'level\s', '', level_str.group(0)))
+
+        string_list = [string.strip() for string in factory.select_one('.button_white').strings]
+        try:
+            wage = string_list[2]
+            if '%' in wage:
+                factory_dict['wage_type'] = 'procentage'
+                factory_dict['wage'] = float(wage.replace(' %', ''))
+        except IndexError:
+            pass
+
         work_page = {
-            'factory_id': int(factory_header['action'].replace('factory/index/', '')),
-            'factory_name': factory_header.text
+            'factory': factory_dict,
+            'work_exp': {},
+            'resources_left': {},
         }
-        factory = soup.select_one('.work_item:nth-child(4)')
-        print('"{}"'.format(factory.text))
+
+        for resource_left in soup.select('span.imp.tip'):
+            classses = resource_left.get('class')
+            for class_name, resource_name in RESOURCE_DICT.items():
+                if class_name in classses:
+                    work_page['resources_left'][resource_name] = float(resource_left.text)
+                    break
+
+        for work_exp in soup.select('.work_exp'):
+            work_exp_amount = int(re.sub(r'exp: | Pt\.', '', work_exp.text.strip()))
+            work_page['work_exp'][work_exp['url']] = work_exp_amount
         return work_page

tests/test_rival_regions_wrapper.py

@@ -185,4 +185,8 @@ def test_war_info():
 def test_work_info():
     """Test work info"""
     response = Work.page()
-    print(response)
+
+    assert isinstance(response, dict), "The response should be a dict"
+    assert isinstance(response['factory'], dict), "Factory should be a dict"
+    assert isinstance(response['resources_left'], dict), "Resources left should be a dict"
+    assert isinstance(response['work_exp'], dict), "Work exp should be a dict"