from penny.models import session, User, TransactionUpload, BankAccount
from penny.common import filetypes
from penny.common.import_transactions import ImportTransactions, ImportTransactionsError
from penny.common.accountmatchrun import AccountMatchRun
from sqlalchemy.orm.exc import NoResultFound
import penny


def import_transactions(id, user_id):
    _app = penny.create_app()
    with _app.app_context():
        filetype = None

        try:
            transactionupload = (
                session.query(TransactionUpload).filter_by(id=id, user_id=user_id).one()
            )
        except NoResultFound:
            raise

        try:
            user = session.query(User).filter_by(id=user_id).one()
        except NoResultFound:
            raise
        else:
            _app.logger.info(
                "Attempting to import transactions; user={0}".format(user.id)
            )

        try:
            if filetypes.is_ofx(transactionupload.filepath):
                filetype = "ofx"
        except filetypes.FileTypesError:
            raise
        else:
            _app.logger.info("File type for import; filetype={0}".format(filetype))

        bankaccount_number = filetypes.get_bankaccount_number_from_ofx(
            transactionupload.filepath
        )

        try:
            bankaccount = (
                session.query(BankAccount)
                .filter_by(number=bankaccount_number, user=user)
                .one()
            )
        except NoResultFound:
            # FIXME: creating a new bank account now depends on an entity though
            # the UI has no way to set the entity when importing transactions.
            # The commit() below will raise an exception given there is no entity
            # with this BankAccount.
            bankaccount = BankAccount(user_id=user_id, number=bankaccount_number)
            session.add(bankaccount)
            session.commit()

        # Read all transactions into a list
        # transactions = None
        # with open(transactionupload.filepath, 'r') as fh:
        #    transactions = fh.read()

        # Instantiate the ImportTransactions object ready to import.
        import_transactions = ImportTransactions(transactionupload, bankaccount, user)

        # Import the transactions.
        imported_transactions = None
        if filetype == "ofx":
            try:
                imported_transactions = import_transactions.process_ofx()
            except ImportTransactionsError:
                raise

        return imported_transactions


def run_accountmatchrun(user_id):
    _app = penny.create_app()
    with _app.app_context():
        try:
            user = session.query(User).filter_by(id=user_id).one()
        except NoResultFound:
            raise
        else:
            _accountmatchrun = AccountMatchRun(user)
            return _accountmatchrun.run()