83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
import logging
|
|
import json
|
|
|
|
from flask.ext.sqlalchemy import SQLAlchemy
|
|
|
|
from accounting.exceptions import AccountingException
|
|
from accounting.storage import Storage
|
|
from accounting.models import Transaction, Posting, Amount
|
|
|
|
_log = logging.getLogger(__name__)
|
|
db = SQLAlchemy()
|
|
|
|
|
|
class SQLStorage(Storage):
|
|
def __init__(self, app=None):
|
|
|
|
if not app:
|
|
raise Exception('Missing app keyword argument')
|
|
|
|
self.app = app
|
|
db.init_app(app)
|
|
|
|
from .models import Transaction as SQLTransaction, \
|
|
Posting as SQLPosting, Amount as SQLAmount
|
|
|
|
db.create_all()
|
|
|
|
self.Transaction = SQLTransaction
|
|
self.Posting = SQLPosting
|
|
self.Amount = SQLAmount
|
|
|
|
def get_transactions(self, *args, **kw):
|
|
transactions = []
|
|
|
|
for transaction in self.Transaction.query.all():
|
|
dict_transaction = transaction.as_dict()
|
|
dict_postings = dict_transaction.pop('postings')
|
|
|
|
postings = []
|
|
|
|
for dict_posting in dict_postings:
|
|
dict_amount = dict_posting.pop('amount')
|
|
posting = Posting(**dict_posting)
|
|
posting.amount = Amount(**dict_amount)
|
|
|
|
postings.append(posting)
|
|
|
|
dict_transaction.update({'postings': postings})
|
|
|
|
transactions.append(Transaction(**dict_transaction))
|
|
|
|
return transactions
|
|
|
|
def update_transaction(self, transaction):
|
|
if transaction.id is None:
|
|
raise AccountingException('The transaction id must be set for'
|
|
' update_transaction calls')
|
|
|
|
_log.debug('DUMMY: Update transaction: %s', transaction)
|
|
|
|
def add_transaction(self, transaction):
|
|
if transaction.id is None:
|
|
transaction.generate_id()
|
|
|
|
_t = self.Transaction()
|
|
_t.uuid = transaction.id
|
|
_t.date = transaction.date
|
|
_t.payee = transaction.payee
|
|
_t.meta = json.dumps(transaction.metadata)
|
|
|
|
db.session.add(_t)
|
|
|
|
for posting in transaction.postings:
|
|
_p = self.Posting()
|
|
_p.transaction_uuid = transaction.id
|
|
_p.account = posting.account
|
|
_p.meta = json.dumps(posting.metadata)
|
|
_p.amount = self.Amount(symbol=posting.amount.symbol,
|
|
amount=posting.amount.amount)
|
|
|
|
db.session.add(_p)
|
|
|
|
db.session.commit()
|