浏览代码

reconfigure directory structure

year, month, day are now first
marton levente 7 月之前
父节点
当前提交
803b37ddf3
共有 2 个文件被更改,包括 10 次插入8 次删除
  1. 4 2
      anafapi/anafoauth.py
  2. 6 6
      tests/test_anafoauth.py

+ 4 - 2
anafapi/anafoauth.py

@@ -207,10 +207,12 @@ class Efactoauth(Anafoauth):
                 dir_days = '{:02d}'.format(date_.day)
                 if type_ == 'FACTURA TRIMISA':
                     # dirs = os.path.join(root_path, dir_cifs, dir_types, dir_dates, dir_days, dir_cifs_ben)
-                    dirs = os.path.join(root_path, dir_types, dir_dates, dir_cifs, dir_days, dir_cifs_ben)
+                    # dirs = os.path.join(root_path, dir_types, dir_dates, dir_cifs, dir_days, dir_cifs_ben)
+                    dirs = os.path.join(root_path, dir_types, dir_dates, dir_days, dir_cifs, dir_cifs_ben)
                 else:
                     # dirs = os.path.join(root_path, dir_cifs, dir_types, dir_dates, dir_days)
-                    dirs = os.path.join(root_path, dir_types, dir_dates, dir_cifs, dir_days)
+                    # dirs = os.path.join(root_path, dir_types, dir_dates, dir_cifs, dir_days)
+                    dirs = os.path.join(root_path, dir_types, dir_dates, dir_days, dir_cifs)
                 # print(dirs)
                 os.makedirs(dirs, exist_ok=True)
                 new = self.download_invoice(download_id=inv['id'], path=dirs, env=env)

+ 6 - 6
tests/test_anafoauth.py

@@ -70,7 +70,7 @@ class efactTest(unittest.TestCase):
                     self.efactoauth.transform_to_pdf(inv_file, no_validation=True)
 
     def test_get_all_messages(self):
-        self.efactoauth.get_all_messages('17259191', days=10, root_path='17259191', env='prod', filter_='P', extract=True)
+        self.efactoauth.get_all_messages('17259191', days=5, root_path='17259191', env='prod', filter_='T', extract=True)
 
     def test_pretty_messages(self):
         self.efactoauth.pretty_messages('17259191', env='prod')
@@ -95,7 +95,7 @@ class efactTest(unittest.TestCase):
                                                     exp_plan=691,
                                                     warehouse_code=23,
                                                     generic=True,
-                                                    generic_value='946')
+                                                    generic_value='2209')
                 os.rename(latest, latest.replace('.xml'.lower(), '_booked.xml'))
             # os.remove(latest)
 
@@ -129,9 +129,9 @@ class efactTest(unittest.TestCase):
             # next_subfolder = next(os.walk(dir_path))[1][0]
         for root_f, _dirs, files in os.walk(dir_path):
             for file in files:
-                # print(root_f)
+                # print(root_f, files)
                 if file.endswith('.xml') and file.count('_') < 2 and 'cif error' not in root_f:
-                    # print(os.path.join(root, file))
+                    # print(os.path.join(root_f, file))
                     tree = etree.parse(os.path.join(root_f, file))
                     root = tree.getroot()
                     parties = root.xpath('//cbc:RegistrationName', namespaces={'cbc': 'urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2'})
@@ -189,6 +189,6 @@ if __name__ == "__main__":
     # unittest.main()
     # efactTest().test_add_invoice()
     efactTest().test_get_all_messages()
-    # efactTest().test_rename_all(os.path.join(os.path.expanduser('~'), 'git/anafapi/tests/17259191'), kind='seller')
-    # efactTest().test_rename_all(os.path.join(os.path.expanduser('~'), 'git/anafapi/tests/17259191'), kind='buyer')
+    # efactTest().test_rename_all(os.path.join(os.path.expanduser('~'), 'git/anafapi/tests/17259191/'), kind='seller')
+    # efactTest().test_rename_all(os.path.join(os.path.expanduser('~'), 'git/anafapi/tests/17259191/'), kind='buyer')
     # anoauth.get_token()