Jelajahi Sumber

improved sort with formatted date

marton levente 9 bulan lalu
induk
melakukan
44def405bf
2 mengubah file dengan 5 tambahan dan 4 penghapusan
  1. 1 1
      anafapi/anafspv.py
  2. 4 3
      tests/test_anafspv.py

+ 1 - 1
anafapi/anafspv.py

@@ -82,7 +82,7 @@ class anafSPV(object):
     def get_messages(self, days=10):
         response = self.session.get(self.BASEURL.format(self.MESLIST.format(days)), cookies=self.requests_cookies).json()
         sorted_response = sorted(map(self._convert_and_extract_date, response['mesaje']), key=lambda x: x['created_date'])
-        filtered_list = sorted([item for item in response['mesaje'] if item['tip'] != 'RECIPISA'], key=lambda x: x['created_date'])
+        filtered_list = sorted([item for item in response['mesaje'] if item['tip'] != 'RECIPISA'], key=lambda x: datetime.strptime(x['created_date'], '%d-%b-%Y'))
         response['mesaje'] = filtered_list
         return response
 

+ 4 - 3
tests/test_anafspv.py

@@ -13,8 +13,8 @@ class spvTest(unittest.TestCase):
         requests_cookies = self.spv.get_cookies_from_browser()
         messages = self.spv.get_messages(10)
         pprint.pprint(messages)
-        for message in messages['mesaje'][-1:]:  # -1:
-            self.spv.download_message(message['cif'], id_=message['id'], details=message['detalii'])
+        # for message in messages['mesaje'][-1:]:  # -1:
+        #     self.spv.download_message(message['cif'], id_=message['id'], details=message['detalii'])
         # self.spv.download_message('27100508', id_='556755658', details=message['tip'])
 
     def test_get_fin_sheet(self):
@@ -29,4 +29,5 @@ class spvTest(unittest.TestCase):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    # unittest.main()
+    spvTest().test_get_messages()