um-cseg / chez-betty

# Call this when the cash box gets emptied

def reconcile_safe(amount, admin):

    assert(amount>=0)

    e = event.EmptySafe(admin)

    DBSession.add(e)

    DBSession.flush()

    safe_c = account.get_cash_account("safe")

    expected_amount = safe_c.balance

    amount_missing = expected_amount - amount

    if amount_missing != 0.0:

        # If the amount in the safe doesn't match what we expected there to

        # be, we need to adjust the amount in the cash box be transferring

        # to or from a null account

        if amount_missing > 0:

            # We got less in the box than we expected

            # Move money from the safe account to null with transaction type

            # "lost"

            t1 = transaction.Lost(e, account.get_cash_account("safe"), amount_missing)

            DBSession.add(t1)

        else:

            # We got more in the box than expected! Use a found transaction

            # to reconcile the difference

            t1 = transaction.Found(e, account.get_cash_account("safe"), abs(amount_missing))

            DBSession.add(t1)

    # Now move all the money from the safe to chezbetty

    t2 = transaction.EmptySafe(e, amount)

    DBSession.add(t2)

    return e

# Call this to move all of the money from the cash box to the safe.

# Call this when bitcoins are converted to USD

def reconcile_bitcoins(amount, admin, expected_amount=None):

    assert(amount>0)

    e = event.EmptyBitcoin(admin)

    DBSession.add(e)

    DBSession.flush()

    btcbox_c = account.get_cash_account("btcbox")

    if expected_amount == None:

        expected_amount = btcbox_c.balance

    amount_missing = expected_amount - amount

    if amount_missing != 0.0:

        # Value of bitcoins fluctated and we didn't make as much as we expected

        if amount_missing > 0:

            # We got less in bitcoins than we expected

            # Move money from the btcbox account to null with transaction type

            # "lost"

            t1 = transaction.Lost(e, account.get_cash_account("btcbox"), amount_missing)

            DBSession.add(t1)

        else:

            # We got more in bitcoins than expected! Use a found transaction

            # to reconcile the difference

            t1 = transaction.Found(e, account.get_cash_account("btcbox"), abs(amount_missing))

            DBSession.add(t1)

    # Now move all the money from the bitcoin box to chezbetty

    t2 = transaction.EmptyBitcoin(e, amount)

    DBSession.add(t2)

    return expected_amount

# Call this to make a miscellaneous adjustment to the chezbetty account