import polars as pl

accounts = pl.read_parquet('data/warehouse/accounts.parquet')
categories = pl.read_parquet('data/warehouse/categories.parquet')
dates = pl.read_parquet('data/warehouse/dates.parquet')
payees = pl.read_parquet('data/warehouse/payees.parquet')
scheduled_transactions = pl.read_parquet('data/warehouse/scheduled_transactions.parquet')
transactions = pl.read_parquet('data/warehouse/transactions.parquet')


master_transactions = transactions.join(categories, left_on='category_id', right_on='category_id', suffix='_category')\
                    .join(accounts, left_on='account_id', right_on='account_id', suffix='_account')\
                    .join(payees, left_on='payee_id', right_on='payee_id', suffix='_payee')\
                    .join(dates, left_on='transaction_date', right_on='date_id', suffix='_date')


# Create aggregations
spend_per_day = master_transactions.sql('''
    SELECT 
        date,
        year,
        month,
        day,
        ABS(SUM(transaction_amount)) as total
    FROM self
    WHERE category_name != 'Inflow: Ready to Assign'
    GROUP BY date, year, month, day
    ORDER BY date DESC
    '''
)

spend_per_category = master_transactions.sql('''
    SELECT
        category_name,
        ABS(SUM(transaction_amount)) as total
    FROM self
    WHERE category_name != 'Inflow: Ready to Assign'
    GROUP BY category_name
    ORDER BY total DESC
    '''
)

spend_per_payee = master_transactions.sql('''
    SELECT
        payee_name,
        ABS(SUM(transaction_amount)) as total
    FROM self
    WHERE payee_name != 'Starting Balance'
        AND transaction_amount < 0
    GROUP BY payee_name
    ORDER BY total DESC
    '''
)

print(spend_per_payee)