import pandas as pd
from openpyxl import load_workbook
from openpyxl.styles import PatternFill

# Load MomentsPay_Bank_new.xlsx
moments_pay_bank_df = pd.read_excel('Payuupi20-05.xlsx')

# Load Transaction details as on 14.12.2023
transaction_details_df = pd.read_excel('Payutransaction20-05.xlsx')

# Convert relevant columns to a common data type (string)
transaction_details_df['PayU ID'] = transaction_details_df['id'].astype(str).replace(r'\.0$', '', regex=True)
transaction_details_df['Amount'] = transaction_details_df['amount'].astype(str).replace(r'\.0$', '', regex=True)

moments_pay_bank_df['Order ID'] = moments_pay_bank_df['Order ID'].astype(str).replace(r'\.0$', '', regex=True).replace(r"'",'',regex=True)
moments_pay_bank_df['Transaction Amount'] = moments_pay_bank_df['Transaction Amount'].astype(str).replace(r'\.0$', '', regex=True)

# Specify the mapping between fields excluding 'processing_id' and 'transaction_id'
field_mapping = {'PayU ID': 'Order ID','Amount':'Transaction Amount'}

# Filter only card transactions
transaction_details_df = transaction_details_df[
    (transaction_details_df['mode'] == 'UPI') | 
    (transaction_details_df['mode'] == 'NB')]

# Remove duplicates from moments_pay_bank_df based on 'approval_code' and 'total_amount'
moments_pay_bank_df_dedup = moments_pay_bank_df.drop_duplicates(subset=['Order ID'], keep='first')

# Merge dataframes on the specified fields
merged_df = pd.merge(transaction_details_df, moments_pay_bank_df_dedup[list(field_mapping.values())],
                     left_on=list(field_mapping.keys()),
                     right_on=list(field_mapping.values()),
                     how='left')
unmerged_df = pd.merge(transaction_details_df, moments_pay_bank_df_dedup[list(field_mapping.values())],
                       left_on=list(field_mapping.keys()),
                       right_on=list(field_mapping.values()),
                       how='outer')

equal_records = merged_df[~merged_df['Order ID'].isnull()]
equal_records.insert(0, 'S.No.', range(1, len(equal_records) + 1))
equal_records['momentpay matched'] = 'YES'
print(equal_records)

# Filter unmatched card transactions
unequal_records = unmerged_df[unmerged_df['Order ID'].isnull() & (unmerged_df['mode'].isin(['UPI', 'NB']))]
unequal_records.insert(0, 'S.No.', range(1, len(unequal_records) + 1))
unequal_records['momentpay matched'] = 'NO'
print(unequal_records)

# Function to create or overwrite an Excel sheet
def create_or_overwrite_sheet(sheet_name, data):
    with pd.ExcelWriter('Rela-HISBANK09-04.xlsx', mode='a', engine='openpyxl', if_sheet_exists='replace') as writer:
        data.to_excel(writer, sheet_name=sheet_name, index=False)

# Generate sheet names
sheet_name_matched = 'PayuBankupi-Matched'
sheet_name_unmatched = 'PayuBankupi-Unmatched'
create_or_overwrite_sheet(sheet_name_matched, equal_records)
create_or_overwrite_sheet(sheet_name_unmatched, unequal_records)

wb = load_workbook('Rela-HISBANK09-04.xlsx')
Summary_Sheet = wb['Summary']

settled_date = moments_pay_bank_df['Settlement Date'].mode()[0]
Summary_Sheet['A18'] = settled_date
print(settled_date)

# Update summary sheet with total card transaction details
num_rows = len(transaction_details_df)
Summary_Sheet['E18'] = num_rows

total_amount = transaction_details_df['Amount'].astype(float).sum()
Summary_Sheet['F18'] = f"₹{total_amount:,.2f}"

# Update summary sheet with matched card transaction details
HIS_Matched_df = pd.read_excel('Rela-HISBANK09-04.xlsx', sheet_name='PayuBankupi-Matched')
num_rows = len(HIS_Matched_df)
Summary_Sheet['G18'] = num_rows

total_amount = HIS_Matched_df['Amount'].astype(float).sum()
Summary_Sheet['H18'] = f"₹{total_amount:,.2f}"

# Update summary sheet with unmatched card transaction details
HIS_UnMatched_df = pd.read_excel('Rela-HISBANK09-04.xlsx', sheet_name='PayuBankupi-Unmatched')
num_rows = len(HIS_UnMatched_df)
Summary_Sheet['I18'] = num_rows

total_amount = HIS_UnMatched_df['Amount'].astype(float).sum()
Summary_Sheet['J18'] = f"₹{total_amount:,.2f}"

# Define the color for the header
header_fill = PatternFill(fgColor='1274bd', fill_type='solid')

for sheet_name in ['PayuBankupi-Matched', 'PayuBankupi-Unmatched']:
    worksheet = wb[sheet_name]

    # Apply style to the header row
    for row in worksheet.iter_rows(min_row=1, max_row=1):
        for cell in row:
            cell.fill = header_fill

# Save the workbook
wb.save('Rela-HISBANK09-04.xlsx')

# Save matched and unmatched records to separate files
#def save_to_new_excel_file(file_name, sheet_name, data):
#    with pd.ExcelWriter(file_name, engine='openpyxl') as writer:
#        data.to_excel(writer, sheet_name=sheet_name, index=False)

#save_to_new_excel_file('PayuBankupi_matched.xlsx', 'Matched', equal_records)
#save_to_new_excel_file('PayuBankupi_unmatched.xlsx', 'Unmatched', unequal_records)