135 lines
3.5 KiB
Python
135 lines
3.5 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()
|
|
)
|
|
|
|
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):
|
|
"""Return query for this class."""
|
|
|
|
query = db.session.query(
|
|
cls
|
|
)
|
|
|
|
return query
|
|
|
|
@classmethod
|
|
def query_with_balance(cls, account_id, begin=None, end=None):
|
|
"""Get operations with cumulated balance for a speciific account,
|
|
optionally for a specific time period."""
|
|
query = db.session.query(
|
|
*cls.__table__.columns.keys(),
|
|
db.func.sum(
|
|
cls.value
|
|
).filter(
|
|
db.not_(cls.canceled)
|
|
).over(
|
|
partition_by=[cls.account_id],
|
|
order_by=[
|
|
cls.operation_date,
|
|
db.desc(cls.value),
|
|
db.desc(cls.label),
|
|
cls.id
|
|
]
|
|
).label("balance")
|
|
).filter(
|
|
cls.account_id == account_id
|
|
)
|
|
|
|
if begin:
|
|
base_query = query.subquery()
|
|
|
|
query = db.session.query(
|
|
base_query
|
|
).filter(
|
|
cls.operation_date >= str(begin)
|
|
).join(
|
|
cls, base_query.c.id == cls.id
|
|
)
|
|
|
|
if end:
|
|
query = query.filter(cls.operation_date <= str(end))
|
|
|
|
query = query.order_by(
|
|
cls.operation_date,
|
|
db.desc(cls.value),
|
|
db.desc(cls.label),
|
|
cls.id
|
|
)
|
|
|
|
return query
|