Lunaresk
b705f82ecb
Everything kinda changed. Huge refactoring. Website now has sidebar instead of top navigation. Database got huge changes. More forms!
55 lines
2.7 KiB
Python
55 lines
2.7 KiB
Python
from app import db, LOGGER
|
|
from app.models import AmountChange, Bought, PriceChange
|
|
from datetime import date
|
|
from flask import render_template
|
|
from flask_login import current_user
|
|
from sqlalchemy_utils import create_view
|
|
|
|
def group_results(results: tuple) -> list:
|
|
result_list = []
|
|
LOGGER.debug("Grouping...")
|
|
for result in results:
|
|
try:
|
|
result_user_index = [result[0] == result_item['id'] for result_item in result_list].index(True)
|
|
except ValueError as e:
|
|
result_list.append({"id": result[0], "name": result[1], "sum": 0, "item_infos": []})
|
|
result_user_index = -1
|
|
result_user = result_list[result_user_index]
|
|
try:
|
|
result_date_index = [result[2] == result_list_date['date'] for result_list_date in result_user["item_infos"]].index(True)
|
|
except ValueError as e:
|
|
result_user["item_infos"].append({'date': result[2], 'item_list': []})
|
|
result_date_index = -1
|
|
result_date = result_user['item_infos'][result_date_index]
|
|
result_date['item_list'].append({'id': result[3], 'name': result[4], 'amount': result[5], 'price': result[6]})
|
|
for result_user in result_list:
|
|
for result_date in result_user['item_infos']:
|
|
for result_item in result_date['item_list']:
|
|
result_user['sum'] += result_item['amount'] * result_item['price']
|
|
LOGGER.debug("Grouped.")
|
|
return result_list
|
|
|
|
def selectable_price_per_amount_view():
|
|
p = db.aliased(PriceChange, name="p")
|
|
a = db.aliased(AmountChange, name="a")
|
|
date = db.func.greatest(p.date, a.date).label("date")
|
|
price = db.func.ceil(p.price.cast(db.Float)/db.func.coalesce(a.amount, 1)).label("price")
|
|
select = db.select(p.item.label("item"), date, price)
|
|
select = select.distinct(p.item, date)
|
|
select = select.join(a, p.item==a.item, isouter=True)
|
|
select = select.order_by(p.item, db.desc(db.func.greatest(p.date, a.date)))
|
|
return select
|
|
|
|
price_per_amount = create_view("price_per_amount", selectable_price_per_amount_view(), db.metadata)
|
|
|
|
def selectable_bought_with_prices_view():
|
|
b = db.aliased(Bought, name="b")
|
|
ppa = price_per_amount.alias("ppa")
|
|
select = db.select(b.token.label("token"), b.date.label("date"), b.item.label("item"), b.amount.label("amount"), ppa.c.price.label("price"))
|
|
select = select.distinct(b.token, b.date, b.item)
|
|
select = select.join(ppa, b.item==ppa.c.item)
|
|
select = select.where(ppa.c.date<=b.date)
|
|
select = select.order_by(db.desc(b.token), db.desc(b.date), db.desc(b.item), db.desc(ppa.c.date))
|
|
return select
|
|
|
|
bought_with_prices = create_view("bought_with_prices", selectable_bought_with_prices_view(), db.metadata) |