import sys
import argparse
import json
import logging
import locale
from datetime import datetime
from decimal import Decimal
import requests
from accounting.models import Transaction, Posting, Amount
from accounting.transport import AccountingDecoder, AccountingEncoder
locale.setlocale(locale.LC_ALL, '')
_log = logging.getLogger(__name__)
class Client:
def __init__(self, host=None, json_encoder=None,
json_decoder=None):
@@ -40,21 +43,26 @@ class Client:
def post(self, path, payload, **kw):
kw.update({'headers': {'Content-Type': 'application/json'}})
kw.update({'data': json.dumps(payload, cls=self.json_encoder)})
return self._decode_response(requests.post(self.host + path, **kw))
def simple_transaction(self, from_acc, to_acc, amount):
def simple_transaction(self, from_acc, to_acc, amount, symbol=None,
payee=None):
if symbol is None:
# Get the currency from the environment locale
symbol = locale.localeconv()['int_curr_symbol'].strip()
t = Transaction(
date=datetime.today(),
payee='PayPal donation',
payee=payee,
postings=[
Posting(account=from_acc,
amount=Amount(symbol='$', amount=-amount)),
amount=Amount(symbol=symbol, amount=-amount)),
Posting(account=to_acc,
amount=Amount(symbol='$', amount=amount))
amount=Amount(symbol=symbol, amount=amount))
]
)
return self.post('/transaction', {'transactions': [t]})
def get_register(self):
@@ -83,13 +91,13 @@ def print_balance_accounts(accounts, level=0):
' ' + '-' * (80 - len(str(account.name)) - level))
for amount in account.amounts:
print(' ' * level + ' {amount.symbol} {amount.amount}'.format(
amount=amount))
print_balance_accounts(account.accounts, level+1)
print_balance_accounts(account.accounts, level + 1)
def main(argv=None, prog=None):
global HOST
if argv is None:
prog = sys.argv[0]
@@ -97,15 +105,22 @@ def main(argv=None, prog=None):
parser = argparse.ArgumentParser(prog=prog)
actions = parser.add_subparsers(title='Actions', dest='action')
insert = actions.add_parser('insert',
aliases=['in'])
insert.add_argument('payee',
help='The payee line of the transaction')
insert.add_argument('from_account')
insert.add_argument('to_account')
insert.add_argument('amount', type=Decimal)
insert.add_argument('amount', type=Decimal,
help='The amount deducted from from_account and added'
' to to_account')
insert.add_argument('-s', '--symbol',
help='The symbol for the amount, e.g. $ or USD for'
' USD. Defaults to your locale\'s setting.')
actions.add_parser('balance', aliases=['bal'])
actions.add_parser('register', aliases=['reg'])
parser.add_argument('-v', '--verbosity',
@@ -119,13 +134,14 @@ def main(argv=None, prog=None):
logging.basicConfig(level=getattr(logging, args.verbosity))
client = Client(args.host)
if args.action in ['insert', 'in']:
print(client.simple_transaction(args.from_account, args.to_account,
args.amount))
args.amount, payee=args.payee,
symbol=args.symbol))
elif args.action in ['balance', 'bal']:
print_balance_accounts(client.get_balance())
elif args.action in ['register', 'reg']:
print_transactions(client.get_register())
else:
parser.print_help()