Pārlūkot izejas kodu

code for delete unneeded end block

vnc-console 8 mēneši atpakaļ
vecāks
revīzija
849e839e9e
1 mainītis faili ar 43 papildinājumiem un 0 dzēšanām
  1. 43 0
      pymt940/CSV2MentorOtp.py

+ 43 - 0
pymt940/CSV2MentorOtp.py

@@ -2,6 +2,9 @@
 
 from datetime import date
 import pandas as pd
+import csv
+import os
+import shutil
 from pymt940.utils import (get_partners, re_check_otp, dig_it, get_last_month)
 
 class PaymentsProcessorOtp:
@@ -78,6 +81,26 @@ class PaymentsProcessorOtp:
         mentor_names = df_dict['Denumire']
         self.fiscal_code_dict = {k: v for k, v in zip(mentor_names, fiscal_codes)}
 
+        empty_row_index = -1
+        with open(self.csv_path, 'r') as csvfile:
+            reader = csv.reader(csvfile)
+            for i, row in enumerate(reader):
+                if all(not cell or cell.isspace() for cell in row):
+                    empty_row_index = i
+                    break
+        # Temporary file to hold the non-empty rows
+        temp_file_path = os.path.join(os.path.expanduser('~'), self.csv_path)  # '/home/deeejas/Documents/temp.csv'
+
+        with open(self.csv_path, 'r') as csvfile, open(temp_file_path, 'w', newline='') as temp_csv:
+            reader = csv.reader(csvfile)
+            writer = csv.writer(temp_csv)
+            for i, row in enumerate(reader):
+                if i >= empty_row_index:
+                    break
+                writer.writerow(row)
+        # Overwrite the original file by renaming the temp file
+        shutil.move(temp_file_path, self.csv_path)
+
         payments_df = pd.read_csv(self.csv_path, keep_default_na=False)
         payments_df = payments_df.iloc[1:]
 
@@ -93,6 +116,26 @@ class PaymentsProcessorOtp:
         mentor_names = df_dict['Denumire']
         self.fiscal_code_dict = {k: v for k, v in zip(mentor_names, fiscal_codes)}
 
+        empty_row_index = -1
+        with open(self.csv_path, 'r') as csvfile:
+            reader = csv.reader(csvfile)
+            for i, row in enumerate(reader):
+                if all(not cell or cell.isspace() for cell in row):
+                    empty_row_index = i
+                    break
+        # Temporary file to hold the non-empty rows
+        temp_file_path = os.path.join(os.path.expanduser('~'), self.csv_path)  # '/home/deeejas/Documents/temp.csv'
+
+        with open(self.csv_path, 'r') as csvfile, open(temp_file_path, 'w', newline='') as temp_csv:
+            reader = csv.reader(csvfile)
+            writer = csv.writer(temp_csv)
+            for i, row in enumerate(reader):
+                if i >= empty_row_index:
+                    break
+                writer.writerow(row)
+        # Overwrite the original file by renaming the temp file
+        shutil.move(temp_file_path, self.csv_path)
+
         income_df = pd.read_csv(self.csv_path, keep_default_na=False)
         income_df = income_df.iloc[1:]
         # income_df['Credit'].replace(' ', '', regex=True, inplace=True)