241 lines
6.6 KiB
Python
241 lines
6.6 KiB
Python
"""Module containing operation related models."""
|
|
|
|
# vim: set tw=80 ts=4 sw=4 sts=4:
|
|
|
|
from datetime import date
|
|
|
|
import arrow
|
|
|
|
from . import db
|
|
|
|
|
|
# pylint: disable=no-member,invalid-name,too-many-arguments
|
|
# pylint: disable=too-few-public-methods,too-many-instance-attributes
|
|
class Operation(db.Model):
|
|
"""Class used to handle operations."""
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
|
|
operation_date = db.Column(
|
|
db.Date,
|
|
nullable=False,
|
|
default=date.today,
|
|
server_default=db.func.current_date(),
|
|
index=True
|
|
)
|
|
label = db.Column(db.String(500), nullable=False)
|
|
value = db.Column(db.Numeric(15, 2), nullable=False)
|
|
|
|
scheduled_operation_id = db.Column(
|
|
db.Integer,
|
|
db.ForeignKey('scheduled_operation.id'),
|
|
index=True
|
|
)
|
|
|
|
scheduled_operation = db.relationship(
|
|
"ScheduledOperation", backref=db.backref('operations', lazy="dynamic")
|
|
)
|
|
|
|
account_id = db.Column(db.Integer, db.ForeignKey('account.id'), index=True)
|
|
|
|
account = db.relationship(
|
|
'Account', backref=db.backref('operations', lazy="dynamic")
|
|
)
|
|
|
|
category = db.Column(db.String(100), nullable=True, index=True)
|
|
|
|
pointed = db.Column(
|
|
db.Boolean,
|
|
nullable=False,
|
|
default=False,
|
|
server_default=db.false()
|
|
)
|
|
|
|
confirmed = db.Column(
|
|
db.Boolean,
|
|
nullable=False,
|
|
default=True,
|
|
server_default=db.true()
|
|
)
|
|
|
|
canceled = db.Column(
|
|
db.Boolean,
|
|
nullable=False,
|
|
default=False,
|
|
server_default=db.false()
|
|
)
|
|
|
|
sold = db.column_property(
|
|
db.func.sum(
|
|
db.case(
|
|
whens={canceled: db.text("0")},
|
|
else_=value
|
|
)
|
|
).over(
|
|
partition_by=[account_id],
|
|
order_by=[
|
|
"operation_date",
|
|
db.desc("value"),
|
|
db.desc("label"),
|
|
id
|
|
]
|
|
).label("sold")
|
|
)
|
|
|
|
def __init__(self, label, value, account_id, operation_date=None,
|
|
category=None, pointed=False, confirmed=True, canceled=False,
|
|
scheduled_operation_id=None):
|
|
self.operation_date = operation_date
|
|
self.label = label
|
|
self.value = value
|
|
self.category = category
|
|
self.account_id = account_id
|
|
self.scheduled_operation_id = scheduled_operation_id
|
|
self.pointed = pointed
|
|
self.confirmed = confirmed
|
|
self.canceled = canceled
|
|
|
|
@classmethod
|
|
def query(cls, begin=None, end=None):
|
|
"""Return query for this class."""
|
|
# We have to use a join because the sold is not computed from the
|
|
# begining.
|
|
base_query = db.session.query(
|
|
cls.id,
|
|
cls.sold
|
|
).subquery()
|
|
|
|
query = db.session.query(
|
|
cls.id,
|
|
cls.operation_date,
|
|
cls.label,
|
|
cls.value,
|
|
base_query.c.sold,
|
|
cls.category,
|
|
cls.scheduled_operation_id,
|
|
cls.account_id,
|
|
cls.pointed,
|
|
cls.confirmed,
|
|
cls.canceled
|
|
).join(
|
|
base_query, base_query.c.id == cls.id
|
|
).order_by(
|
|
db.desc(cls.operation_date),
|
|
cls.value,
|
|
cls.label,
|
|
)
|
|
|
|
if begin:
|
|
query = query.filter(cls.operation_date >= str(begin))
|
|
|
|
if end:
|
|
query = query.filter(cls.operation_date <= str(end))
|
|
|
|
return query
|
|
|
|
@classmethod
|
|
def get_categories_for_range(cls, account, begin, end):
|
|
"""Get category list for a specific time period."""
|
|
if isinstance(account, (int, str)):
|
|
account_id = account
|
|
else:
|
|
account_id = account.id
|
|
|
|
query = db.session.query(
|
|
cls.category,
|
|
db.func.sum(
|
|
db.case([(db.func.sign(cls.value) == -1, cls.value)], else_=0)
|
|
).label("expenses"),
|
|
db.func.sum(
|
|
db.case([(db.func.sign(cls.value) == 1, cls.value)], else_=0)
|
|
).label("revenues"),
|
|
db.func.sum(cls.value).label("balance")
|
|
).filter(
|
|
cls.account_id == account_id
|
|
).order_by(
|
|
cls.category
|
|
).group_by(
|
|
cls.category
|
|
)
|
|
|
|
if begin:
|
|
query = query.filter(cls.operation_date >= str(begin))
|
|
|
|
if end:
|
|
query = query.filter(cls.operation_date <= str(end))
|
|
|
|
return query
|
|
|
|
@classmethod
|
|
def query_daily_balances(cls, account, begin=None, end=None):
|
|
"""Get expenses, revenues, income and balance per day for a specific
|
|
range and a specific account."""
|
|
if isinstance(account, (int, str)):
|
|
account_id = account
|
|
else:
|
|
account_id = account.id
|
|
|
|
end = end if end else arrow.now().ceil('month').date()
|
|
|
|
query = db.session.query(
|
|
cls.operation_date,
|
|
db.func.coalesce(
|
|
db.func.sum(
|
|
cls.value
|
|
).filter(
|
|
db.func.sign(cls.value) == -1
|
|
).over(
|
|
partition_by=[cls.account_id, cls.operation_date],
|
|
),
|
|
0
|
|
).label("expenses"),
|
|
db.func.coalesce(
|
|
db.func.sum(
|
|
cls.value
|
|
).filter(
|
|
db.func.sign(cls.value) == 1
|
|
).over(
|
|
partition_by=[cls.account_id, cls.operation_date],
|
|
),
|
|
0
|
|
).label("revenues"),
|
|
db.func.coalesce(
|
|
db.func.sum(
|
|
cls.value
|
|
).over(
|
|
partition_by=[cls.account_id, cls.operation_date],
|
|
)
|
|
).label("income"),
|
|
db.func.coalesce(
|
|
db.func.sum(
|
|
cls.value
|
|
).over(
|
|
partition_by=cls.account_id,
|
|
order_by=cls.operation_date
|
|
)
|
|
).label("balance")
|
|
).distinct(
|
|
).order_by(
|
|
cls.operation_date
|
|
).filter(
|
|
cls.account_id == account_id
|
|
)
|
|
|
|
if begin:
|
|
base_query = query.subquery()
|
|
|
|
query = db.session.query(
|
|
base_query
|
|
).filter(
|
|
base_query.c.operation_date >= str(begin)
|
|
).order_by(
|
|
base_query.c.operation_date
|
|
)
|
|
|
|
if end:
|
|
query = query.filter(query.c.operation_date <= str(end))
|
|
|
|
elif end:
|
|
query = query.filter(cls.operation_date <= str(end))
|
|
|
|
return query
|