get vector, get fin_sheet

anafapi/anafspv.py

@@ -94,8 +94,13 @@ class anafSPV(object):
 
     def get_vector(self, cui):
         response = self.session.get(self.VECT.format(cui), cookies=self.requests_cookies).json()
-        sorted_response = sorted(map(self._convert_and_extract_date, response['mesaje']), key=lambda x: x['created_date'])
-        return sorted_response
+        # sorted_response = sorted(map(self._convert_and_extract_date, response['mesaje']), key=lambda x: x['created_date'])
+        return response
+
+    def get_fin_sheet(self, cui):
+        response = self.session.get(self.FROL.format(cui), cookies=self.requests_cookies).json()
+        # sorted_response = sorted(map(self._convert_and_extract_date, response['mesaje']), key=lambda x: x['created_date'])
+        return response
 
     def _convert_and_extract_date(self, item):
         item['created_date'] = datetime.strptime(item['data_creare'], '%d.%m.%Y %H:%M:%S')

tests/test_anafspv.py

@@ -13,7 +13,13 @@ class spvTest(unittest.TestCase):
         requests_cookies = self.spv.get_cookies_from_browser()
         messages = self.spv.get_messages(10)
         pprint.pprint(messages)
-        self.spv.download_message(messages['mesaje'][2]['cif'], id_=messages['mesaje'][2]['id'], details=messages['mesaje'][2]['tip'])
+        for message in messages['mesaje'][-1:]:  # -1:
+            self.spv.download_message(message['cif'], id_=message['id'], details=message['tip'])
+
+    def test_get_fin_sheet(self):
+        requests_cookies = self.spv.get_cookies_from_browser()
+        response = self.spv.get_fin_sheet(cui='27100508')
+        pprint.pprint(response)
 
 
 if __name__ == "__main__":