CC statement work

pythonFiles/breaking_up_cc_statements.py

@@ -0,0 +1,93 @@
+import os
+from openpyxl import load_workbook
+import pandas as pd
+
+
+work_dir = 'C:/Users/Lenovo/Desktop/CC_Statements_xlsx'
+target_dir = 'C:/Users/Lenovo/Desktop/Temp_Target'
+target_file = 'Export_4_2023.xlsx'
+card_names = {'מסטרקארד - 3235':'3235', 'מסטרקארד - 3349':'3349', 'גולד - מסטרקארד - 3047 *':'3047'}
+
+def process_directory():
+    dir = os.listdir(work_dir)
+    for file in dir:
+        process_file(file)
+
+def process_file(file:str):
+    wb = load_workbook(f'{work_dir}/{file}')
+    
+    ws = transform_generator(wb.active)
+    
+    for card, card_num in card_names.items():
+        process_card(card, card_num, file, ws)
+        
+def process_card(
+        card:str, 
+        card_num:str, 
+        file:str, 
+        data:list[list[str]]
+):
+    start = None
+    end = None
+    lines_to_skip = []
+    for row_num, row in enumerate(data):
+        if row[0] == card:
+            if data[row_num + 1][0] == 'אין נתונים להצגה':
+                return None
+            start = row_num + 3
+            break
+    for row_num, row in enumerate(data[start + 1:]):
+        if row[0] in card_names:
+            end = row_num - 2
+            break
+    for row_num, row in enumerate(data[start:end]):
+        if row[0] == 'עסקאות בחו˝ל':
+            lines_to_skip = list(range(row_num - 1, row_num + 2))
+            # for chul_line_num, chul_line in enumerate(data[row_num:end]):
+            #     if chul_line[0] == None:
+            #         lines_to_skip.append(chul_line_num)
+    if end == None:
+        end = len(data)
+        
+    if start and end:
+        new_file_name = name_file(og_file=file, card_num=card_num)
+        build_new_file(new_file_name, data[start:end])
+    
+    return True
+
+def name_file(og_file:str, card_num:str):
+    name, ext = og_file.split('.')
+    name += f'_{card_num}'
+    return '.'.join([name, ext])
+
+
+def transform_generator(ws):
+    return [[x.value for x in row] for row in ws.rows]
+
+def confirm_has_values(rows:list):
+    total = 0
+    for row in rows:
+        total += int(row[0])
+    return True if total > 0 else False
+
+def build_new_file(name_of_new_file:str, list_of_rows:list[str]):
+    new_columns_with_indexes = {'Amount':4, 'Vendor':1, 'Confirmation Code':6}
+    results = []
+    for row in list_of_rows:
+        if row[new_columns_with_indexes['Amount']] == 0:
+            continue
+        relevant_values = []
+        for index in new_columns_with_indexes.values():
+            relevant_values.append(row[index])
+        results.append(relevant_values)
+    if confirm_has_values(relevant_values):
+        # Convert the list of lists to a DataFrame
+        df = pd.DataFrame(results, columns=['Amount', 'Vendor', 'Confirmation Code'])
+
+        # Write the DataFrame to an Excel file
+        df.to_excel(f'{target_dir}/{name_of_new_file}', index=False)
+
+if __name__ == '__main__':
+    print(list(range(4,6)))
+    # process_file(target_file)
+