Jake/data_pipeline_for_YNAB
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

working callback, known minor visual with total spend

Browse Source
This commit is contained in:
Jake
2025-04-12 16:19:04 +01:00
parent a79c245511
commit 7668f790aa
4 changed files with 120 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data_check.py
+14 -5
View File
@@ -53,8 +53,17 @@ spend_per_payee = master_transactions.sql('''
'''
)
# print(spend_per_day)
today = date(date.today())
# Convert the dates to datetime objects that are compatible with Polars
start_date = pl.Date(today)
print(start_date)
def update_dates(start_date, end_date):
print("start date", start_date)
print("end date", end_date)
print(master_transactions)
master_data = master_transactions.filter(
pl.col("date").is_between(start_date, end_date)
)
return master_data
today = date.today()
one_year_ago = today - timedelta(days=5)
data = update_dates(start_date=one_year_ago, end_date=today)
print(data)
visuals/components.py
+92 -88
View File
@@ -6,116 +6,120 @@ import sys
import config.exit_codes as ec
# import datetime
try:
class data_components():
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')
except FileNotFoundError:
logging.error('Data warehouse files not found. Run the data pipeline to create them.')
sys.exit(ec.MISSING_DATA_FILES)
try:
# Join transactions with accounts, categories, and payees to create a master DataFrame
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')
except Exception as e:
logging.error(f'Error joining DataFrames: {e}')
sys.exit(ec.BAD_JOIN)
def __init__(self):
logging.info("Initializing data components")
pass
def update_dates(start_date, end_date):
master_data = master_transactions.filter(
pl.col("date").is_between(start_date, end_date)
)
return master_data
def update_dates(start_date, end_date):
logging.info("Updating dates")
logging.debug(f"start_date: {start_date}, end_date: {end_date}")
logging.debug(data_components.master_transactions.columns)
try:
master_data = data_components.master_transactions.filter(
pl.col("date").is_between(start_date, end_date)
)
except Exception as e:
logging.error(f"Error updating dates: {e}")
raise e
return master_data
def update_data(master_data):
# Create aggregations
spend_per_day = master_data.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
'''
)
def update_data(master_data,callback=0):
# Create aggregations
spend_per_day = master_data.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_data.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_category = master_data.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_data.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
'''
)
total_spend = master_data.sql('''
SELECT ABS(SUM(transaction_amount)) AS total
FROM self
WHERE payee_name != 'Starting Balance'
AND transaction_amount < 0
''').item()
spend_per_payee = master_data.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
'''
)
total_spend = master_data.sql('''
SELECT ABS(SUM(transaction_amount)) AS total
FROM self
WHERE payee_name != 'Starting Balance'
AND transaction_amount < 0
''').item()
# Convert DataFrame to list of dictionaries
spend_per_day_data = spend_per_day.to_dicts()
spend_per_category_data = spend_per_category.to_dicts()
spend_per_payee_data = spend_per_payee.to_dicts()
# Convert DataFrame to list of dictionaries
spend_per_day_data = spend_per_day.to_dicts()
spend_per_category_data = spend_per_category.to_dicts()
spend_per_payee_data = spend_per_payee.to_dicts()
# Convert list of dictionaries to Pandas DataFrame
spend_per_day_df = pd.DataFrame(spend_per_day_data)
spend_per_category_df = pd.DataFrame(spend_per_category_data)
spend_per_payee_df = pd.DataFrame(spend_per_payee_data)
# Convert list of dictionaries to Pandas DataFrame
spend_per_day_df = pd.DataFrame(spend_per_day_data)
spend_per_category_df = pd.DataFrame(spend_per_category_data)
spend_per_payee_df = pd.DataFrame(spend_per_payee_data)
spend_per_day_line = px.line(spend_per_day_df, x="date", y="total")
spend_per_day_line.update_layout(
plot_bgcolor='black',
paper_bgcolor='black',
font_color='white'
)
spend_per_day_line = px.line(spend_per_day_df, x="date", y="total")
spend_per_day_line.update_layout(
plot_bgcolor='black',
paper_bgcolor='black',
font_color='white'
)
spend_per_category_bar = px.bar(spend_per_category_df, x="category_name", y="total")
spend_per_category_bar.update_layout(
plot_bgcolor='black',
paper_bgcolor='black',
font_color='white'
)
spend_per_category_bar = px.bar(spend_per_category_df, x="category_name", y="total")
spend_per_category_bar.update_layout(
plot_bgcolor='black',
paper_bgcolor='black',
font_color='white'
)
spend_per_payee_bar = px.bar(spend_per_payee_df, x="payee_name", y="total")
spend_per_payee_bar.update_layout(
plot_bgcolor='black',
paper_bgcolor='black',
font_color='white'
)
spend_per_payee_bar = px.bar(spend_per_payee_df, x="payee_name", y="total")
spend_per_payee_bar.update_layout(
plot_bgcolor='black',
paper_bgcolor='black',
font_color='white'
)
data = {"spend_per_day_line": spend_per_day_line,
"spend_per_category_bar": spend_per_category_bar,
"spend_per_payee_bar": spend_per_payee_bar,
"total_spend": total_spend}
return data
data = {"spend_per_day_line": spend_per_day_line,
"spend_per_category_bar": spend_per_category_bar,
"spend_per_payee_bar": spend_per_payee_bar,
"total_spend": total_spend}
if callback == 0:
return data
else:
return spend_per_day_line,spend_per_category_bar,spend_per_payee_bar,total_spend
visuals/dash_app.py
+11 -8
View File
@@ -4,14 +4,15 @@ from dash import dcc, html
from dash.dependencies import Input, Output, State
import dash_bootstrap_components as dbc
from visuals.layout import create_layout
from visuals.components import update_data, update_dates
from visuals.components import data_components
from datetime import date, timedelta
import datetime
today = date.today()
one_year_ago = today - timedelta(days=365)
master_data = update_dates(start_date=one_year_ago, end_date=today)
data = update_data(master_data)
master_data = data_components.update_dates(start_date=one_year_ago, end_date=today)
data = data_components.update_data(master_data)
app = dash.Dash(__name__, external_stylesheets=[dbc.themes.DARKLY])
@@ -22,12 +23,14 @@ app = dash.Dash(__name__, external_stylesheets=[dbc.themes.DARKLY])
Output("spend_per_payee","figure"),
Output("total_spend","children"),
[Input('date-picker-range', 'start_date'),
Input('date-picker-range', 'end_date')]
Input('date-picker-range', 'start_date'),
Input('date-picker-range', 'end_date')
)
def update_layout(start_date,end_date):
master_data = update_dates(start_date, end_date)
data = update_data(master_data)
return create_layout(data)
actual_start_date = datetime.date.fromisoformat(start_date)
actual_end_date = datetime.date.fromisoformat(end_date)
master_data = data_components.update_dates(actual_start_date,actual_end_date)
# spend_per_day_line,spend_per_category_bar,spend_per_payee_bar,total_spend = data_components.update_data(master_data,callback=1)
return data_components.update_data(master_data,callback=1)
app.layout = create_layout(data)
visuals/layout.py
+2
View File
@@ -21,6 +21,8 @@ def create_topbar():
[
dbc.Col(
dcc.DatePickerRange(
first_day_of_week=1,
display_format="YYYY-MM-DD",
id="date-picker-range",
start_date=one_year_ago,
end_date=today,