start get_oblig

tests/test_winmnt.py

@@ -9,8 +9,11 @@ from winmentor.mentorutils import Mentorutils
 
 class Test(unittest.TestCase):
     
+    mentor = WinMentor()
+    utils = Mentorutils()
+    
     @unittest.skip('reason')
-    def runTest(self):
+    def test_check_list(self):
         info_anaf.check_list('lista_cf.txt')
     
     def test_utils(self):
@@ -21,6 +24,18 @@ class Test(unittest.TestCase):
         utils = Mentorutils()
         office = mentor.filtered_firmlist('/firme.db', headers=headers)
         utils.gen_turnover('./generated_files/turn_over', 'to', office, m_month='2019_12', n_headers=n_headers)
+    
+    def test_get_bank_accounts(self):
+        for account in self.mentor.get_bank_accounts('WEBS'):
+        # if account[2].startswith(' '):
+            print(account)
+    
+    def test_get_oblig(self):
+        head = ['Denumire', 'CF', 'J', 'Adresa', 'Oras', 'Judet', 'Prescurtat', 'Admin', 'AdminP', 'RCNP', 'Obs']
+        firm_list = list(self.mentor.filtered_firmlist(head))
+        
+        print(self.mentor.get_oblig('CHRYPRI/2023_02'))
+        
 
 if __name__ == "__main__":
 

winmentor/dbread.py

@@ -111,7 +111,7 @@ class WinMentor(object):
         return m_list
 
     def make_sal_list(self, file_1, file_2) -> list:
-        '''returns list with all employees in current month
+        '''returns list with all_ employees in current month
         :param file_1 is npers.db from shortname
         :param file_2 is likisal.db shortname/current_month'''
         file_name1 = self.update_copy(self.winment_path + file_1)
@@ -253,7 +253,7 @@ class WinMentor(object):
 
     def get_bank_accounts(self, short_name, db_file='/nbanca.db'):
         #.......................................................................
-        # TO DO: put an all parameter to yield all accounts or
+        # TO DO: put an all_ parameter to yield all_ accounts or
         # just one,
         # make a named tuple with bank accounts.
         #.......................................................................
@@ -275,6 +275,40 @@ class WinMentor(object):
                     bank_account.append(bank_codes[bank_account[0]])
                     if bank_account[2].startswith(' '):
                         yield bank_account
+    
+    def get_oblig(self, short_name, db_file='/ObligPI.DB'):
+        headers = ['Part', 'TipDoc', 'Doc', 'NrDoc', 'DataDoc', 'Valoare', 'Rest']
+        short_name = self.winment_path + short_name
+        # nbanks = self.update_copy(short_name + '/nbanci.db')
+        oblig_db = self.update_copy(short_name + db_file)
+        bank_codes = {}
+        with Table(oblig_db) as table: # , Table(nbanks) as nbanks
+            all_ = []
+            parts = []
+            tips = []
+            docs = []
+            nrs = []
+            dates = []
+            values = []
+            rests = []
+            for row in table:
+                if row.TipDoc == 1:
+                    parts.append(row.Part)
+                    all_.append(parts)
+                    tips.append(row.TipDoc)
+                    all_.append(tips)
+                    docs.append(row.Doc)
+                    all_.append(docs)
+                    nrs.append(row.NrDoc)
+                    all_.append(nrs)
+                    dates.append(row.DataDoc)
+                    all_.append(dates)
+                    values.append(row.Valoare)
+                    all_.append(values)
+                    rests.append(row.Rest)
+                    all_.append(rests)
+            df_dict = dict(zip(headers, all_))
+            print(df_dict)
         
 
     def corp_list(self, name=None):