def command(args):
import os
from subprocess import check_call
from sr.tools import spending
from sr.tools.budget import BudgetTree
root = spending.find_root()
budget = spending.load_budget_with_spending(root)
line = budget.path(args.budgetline)
if isinstance(line, BudgetTree):
print("Summary for tree", args.budgetline)
budgeted = sum([x.cost for x in line.walk()])
spent = sum([x.spent for x in line.walk()])
else:
print("Summary for line", args.budgetline)
budgeted = line.cost
spent = line.spent
print("Budgeted:\t", budgeted)
print("Spent:\t\t", spent)
if spent > budgeted:
print("OVER BUDGET")
print("Transactions:")
account = spending.budget_line_to_account(args.budgetline)
check_call([
"ledger", "--file",
os.path.join(root, "spending.dat"), "reg", account
])
def command(args):
import os
from subprocess import check_call
from sr.tools import spending
from sr.tools.budget import BudgetTree
root = spending.find_root()
budget = spending.load_budget_with_spending(root)
line = budget.path(args.budgetline)
if isinstance(line, BudgetTree):
print("Summary for tree", args.budgetline)
budgeted = sum([x.cost for x in line.walk()])
spent = sum([x.spent for x in line.walk()])
else:
print("Summary for line", args.budgetline)
budgeted = line.cost
spent = line.spent
print("Budgeted:\t", budgeted)
print("Spent:\t\t", spent)
if spent > budgeted:
print("OVER BUDGET")
print("Transactions:")
account = spending.budget_line_to_account(args.budgetline)
check_call(["ledger",
"--file", os.path.join(root, "spending.dat"),
"reg", account])
def command(args):
import re
import sys
import six.moves.xmlrpc_client as xmlrpclib
from sr.tools import spending
from sr.tools.trac import TracProxy
try:
root = spending.find_root()
except spending.NotSpendingRepo:
print(
"Please run in spending.git top level directory", file=sys.stderr)
exit(1)
spends = spending.load_transactions(root)
spendsumgrp = {}
for s in spends:
if s.trac in spendsumgrp:
spendsumgrp[s.trac] += float(s.cost)
else:
spendsumgrp[s.trac] = float(s.cost)
server = TracProxy(anon=True)
mserver = xmlrpclib.MultiCall(server)
tickets = server.ticket.query("status!=closed&component=Purchasing")
for ticket in tickets:
mserver.ticket.get(ticket)
costsumgrp = {}
for ticket in mserver():
match = re.search(
'Total cost: \xa3([0-9.]+)', ticket[3]['description'])
if match is None:
print("Unable to determine cost for ticket " +
str(ticket[0]) + ". Invalid formatting")
continue
if ticket[0] in costsumgrp:
costsumgrp[ticket[0]] += float(match.groups()[0])
else:
costsumgrp[ticket[0]] = float(match.groups()[0])
for val in costsumgrp:
if spendsumgrp[val] != costsumgrp[val]:
print("Ticket " + str(val) + " does not match transactions")
print("\tTicket cost: £" + str(costsumgrp[val]))
print("\tTransactions: £" + str(spendsumgrp[val]))
def command(args):
from decimal import Decimal as D
import sys
from sr.tools import spending
from sr.tools.budget import BudgetTree
try:
root = spending.find_root()
except spending.NotSpendingRepo:
print("Please run in spending.git", file=sys.stderr)
sys.exit(1)
bud = spending.load_budget_with_spending(root)
if args.path != "/":
try:
budline = bud.path(args.path)
except KeyError:
print("Budget line '" + args.path + "' not found", file=sys.stderr)
sys.exit(1)
else:
budline = bud
total = D(0)
spent = D(0)
total_unspent = D(0)
if isinstance(budline, BudgetTree):
for line in budline.walk():
spent += line.spent
total += line.cost
unspent = line.cost - line.spent
if unspent != D(0):
print(line.name, unspent)
total_unspent += unspent
print("-" * 10)
else:
total_unspent = budline.cost - budline.spent
print("Total budget:\t", total)
print("Total unspent:\t", total_unspent)
def command(args):
import os
import sys
from sr.tools import spending
from sr.tools.config import Config
from sr.tools.environment import get_config_filename
# Invoke ledger on the SR spending repo
# Check that we are indeed invoking it on spending.git
config = Config()
# Default to using the spending.git specified in the config
root = config["spending"]
if root is not None:
root = os.path.expanduser(root)
else:
# Not specified in the config
root = os.getcwd()
try:
# Check that it's actually spending.git
root = spending.find_root(path=root)
except spending.NotSpendingRepo:
print("This isn't SR spending.git", file=sys.stderr)
print("Solve this by either:", file=sys.stderr)
print(" - Changing working directory to spending.git", file=sys.stderr)
print(" - Set the 'spending' config option in {}".format(
get_config_filename()),
file=sys.stderr)
sys.exit(1)
ledger_args = ['ledger'] + args.command
if "--file" not in args:
# Tell ledger where to look
ledger_args = ['ledger', "--file", os.path.join(root, "spending.dat")] \
+ args.command
os.execvp("ledger", ledger_args)
def command(args):
import os
import sys
from sr.tools import spending
from sr.tools.config import Config
from sr.tools.environment import get_config_filename
# Invoke ledger on the SR spending repo
# Check that we are indeed invoking it on spending.git
config = Config()
# Default to using the spending.git specified in the config
root = config["spending"]
if root is not None:
root = os.path.expanduser(root)
else:
# Not specified in the config
root = os.getcwd()
try:
# Check that it's actually spending.git
root = spending.find_root(path=root)
except spending.NotSpendingRepo:
print("This isn't SR spending.git", file=sys.stderr)
print("Solve this by either:", file=sys.stderr)
print(" - Changing working directory to spending.git", file=sys.stderr)
print(" - Set the 'spending' config option in {}"
.format(get_config_filename()), file=sys.stderr)
sys.exit(1)
ledger_args = ['ledger'] + args.command
if "--file" not in args:
# Tell ledger where to look
ledger_args = ['ledger', "--file", os.path.join(root, "spending.dat")] \
+ args.command
os.execvp("ledger", ledger_args)
def command(args):
import re
import sys
import sr.tools.spending as srspending
import sr.tools.budget as srbudget
try:
spending_root = srspending.find_root()
except srspending.NotSpendingRepo:
print("Please run in spending.git repository", file=sys.stderr)
exit(1)
budget = srspending.load_budget_with_spending(spending_root)
path = budget.path(args.LINE)
if isinstance(path, srbudget.BudgetItem):
items = [path]
else:
items = path.walk()
for item in items:
if item.closed:
# Line is already closed; nothing to do
continue
if item.spent > (item.cost * srbudget.FUDGE_FACTOR):
# More has been spent than the budget line's value
print(
"Cannot close {0}: More has been spent than budgeted.".format(
item.name),
file=sys.stderr)
print("\tSpent: £{0}\n\tBudgeted: £{1}".format(
item.spent, item.cost * srbudget.FUDGE_FACTOR),
file=sys.stderr)
# Skip this item
continue
orig_contents = open(item.fname, 'r').read()
contents = re.sub(r'^(\s*cost\s*:\s*).+$',
r'\g<1>{0}'.format(item.spent), orig_contents, 0,
re.MULTILINE)
# Check that the cost has actually been changed
# within the string if it actually needed to be
if item.spent != item.cost and contents == orig_contents:
print("Warning: Failed to set the cost of {0} -- skipping it.".
format(item.name),
file=sys.stderr)
continue
# Mark as closed
m = re.search(r"^\s*closed\s*:.*$", contents, flags=re.MULTILINE)
if m is not None:
# There's a closed line in there already
contents = re.sub(r"^(\s*closed\s*:\s*).*$",
r"\1true",
contents,
flags=re.MULTILINE)
else:
contents += "\nclosed: true\n"
with open(item.fname, 'w') as file:
file.write(contents)
def command(args):
import re
import sys
import sr.tools.spending as srspending
import sr.tools.budget as srbudget
try:
spending_root = srspending.find_root()
except srspending.NotSpendingRepo:
print("Please run in spending.git repository", file=sys.stderr)
exit(1)
budget = srspending.load_budget_with_spending(spending_root)
path = budget.path(args.LINE)
if isinstance(path, srbudget.BudgetItem):
items = [path]
else:
items = path.walk()
for item in items:
if item.closed:
# Line is already closed; nothing to do
continue
if item.spent > (item.cost * srbudget.FUDGE_FACTOR):
# More has been spent than the budget line's value
print("Cannot close {0}: More has been spent than budgeted."
.format(item.name), file=sys.stderr)
print("\tSpent: £{0}\n\tBudgeted: £{1}"
.format(item.spent, item.cost * srbudget.FUDGE_FACTOR),
file=sys.stderr)
# Skip this item
continue
orig_contents = open(item.fname, 'r').read()
contents = re.sub(r'^(\s*cost\s*:\s*).+$',
r'\g<1>{0}'.format(item.spent),
orig_contents, 0, re.MULTILINE)
# Check that the cost has actually been changed
# within the string if it actually needed to be
if item.spent != item.cost and contents == orig_contents:
print("Warning: Failed to set the cost of {0} -- skipping it."
.format(item.name), file=sys.stderr)
continue
# Mark as closed
m = re.search(r"^\s*closed\s*:.*$", contents, flags=re.MULTILINE)
if m is not None:
# There's a closed line in there already
contents = re.sub(r"^(\s*closed\s*:\s*).*$",
r"\1true",
contents,
flags=re.MULTILINE)
else:
contents += "\nclosed: true\n"
with open(item.fname, 'w') as file:
file.write(contents)
def command(args):
import datetime
import sr.tools.spending as srspending
import sr.tools.budget as srbudget
from sr.tools.trac import TracProxy, WrongServer
try:
spending_root = srspending.find_root()
except srspending.NotSpendingRepo:
print("Please run in spending.git top level directory",
file=sys.stderr)
sys.exit(1)
try:
budget = srspending.load_budget_with_spending(spending_root)
except srbudget.NoBudgetConfig as nbc:
print("Error:", nbc.message, file=sys.stderr)
print("Have you initialised the budget submodule?", file=sys.stderr)
print("If not, run 'git submodule update --init'.", file=sys.stderr)
sys.exit(1)
if args.spend_file is not None:
spend_request = SpendRequest(args.spend_file)
else:
spend_request = SpendRequest.from_editor()
ticket_text = "Payee: {} \\\\\n".format(spend_request.username)
if spend_request.supplier_url is None:
ticket_text += "Supplier: {} \\\\\n".format(spend_request.supplier)
else:
ticket_text += "Supplier: [{url} {supplier}] \\\\\n" \
.format(url=spend_request.supplier_url,
supplier=spend_request.supplier)
budget_line_totals = {}
for purchase in spend_request.purchases:
try:
budget_line = budget.path(purchase.budget_line)
except KeyError:
# TODO: Move this check up into the parser so it's caught and the
# user can fix it
print('Budget line "{0}" not found', file=sys.stderr)
sys.exit(1)
bl_request_total = D(0)
ticket_text += """
=== Items from [budget:{budget_line}] ===
{summary} \\\\
||= '''Item''' =||= '''Cost''' =||
""".format(budget_line=purchase.budget_line,
summary=purchase.summary)
for item in purchase.items:
ticket_text += "|| {desc} || £{cost} ||\n".format(desc=item.desc,
cost=item.cost)
bl_request_total += item.cost
# How much has already been spent against this budget line
spent = budget_line.spent
req_total = spent + bl_request_total
# It is over the limit?
if req_total > budget_line.cost:
print("Warning: This purchase exceeds the budget line '{0}'"
.format(purchase.budget_line))
print("\tBudget line's value: £{0}".format(budget_line.cost))
print("\tRequested Expenditure: £{0} ({1}%)"
.format(req_total, 100 * (req_total) / budget_line.cost))
if not input("Continue anyway? [y/N] ").lower() == 'y':
sys.exit()
if purchase.budget_line not in budget_line_totals:
budget_line_totals[purchase.budget_line] = D(0)
budget_line_totals[purchase.budget_line] += bl_request_total
ticket_text += """
=== Budget Line Totals ===
||= '''Budget Line''' =||= '''Total''' =||
"""
for line, total in budget_line_totals.items():
ticket_text += "|| [budget:{line}] || £{total} ||\n" \
.format(line=line, total=total)
print(ticket_text)
if args.dry_run:
print("Stopping before actually creating ticket.")
sys.exit(0)
try:
server = TracProxy(server=args.server, port=args.port)
except WrongServer:
print("Error: The specified server is not a Trac instance",
file=sys.stderr)
sys.exit(1)
ticketNum = server.ticket.create(spend_request.summary,
ticket_text,
{'component': "Purchasing",
'owner': "treasurer",
'type': "task"})
if not ticketNum > 0:
print("Unable to create a valid ticket")
sys.exit()
if args.port == 443:
hostname = args.server
else:
hostname = "{0}:{1}".format(args.server, args.port)
print("Purchasing ticket created: https://{0}/trac/ticket/{1}".format(
hostname, ticketNum))
print("Spending Entries:")
for purchase in spend_request.purchases:
print()
today = datetime.date.today()
i = "{date} ! {summary}\n"
i += " {account}\t£{amount}\n"
i += " Liabilities:{payee}\n"
i += " ; trac: #{trac}"
print(i.format(date=today.isoformat(),
summary=purchase.summary,
account=srspending.budget_line_to_account(
purchase.budget_line),
amount=sum([x.cost for x in purchase.items]),
payee=spend_request.username,
trac=ticketNum))
def command(args):
import datetime
import sr.tools.spending as srspending
import sr.tools.budget as srbudget
from sr.tools.trac import TracProxy, WrongServer
try:
spending_root = srspending.find_root()
except srspending.NotSpendingRepo:
print("Please run in spending.git top level directory",
file=sys.stderr)
sys.exit(1)
try:
budget = srspending.load_budget_with_spending(spending_root)
except srbudget.NoBudgetConfig as nbc:
print("Error:", nbc.message, file=sys.stderr)
print("Have you initialised the budget submodule?", file=sys.stderr)
print("If not, run 'git submodule update --init'.", file=sys.stderr)
sys.exit(1)
if args.spend_file is not None:
spend_request = SpendRequest(args.spend_file)
else:
spend_request = SpendRequest.from_editor()
ticket_text = "Payee: {} \\\\\n".format(spend_request.username)
if spend_request.supplier_url is None:
ticket_text += "Supplier: {} \\\\\n".format(spend_request.supplier)
else:
ticket_text += "Supplier: [{url} {supplier}] \\\\\n" \
.format(url=spend_request.supplier_url,
supplier=spend_request.supplier)
budget_line_totals = {}
for purchase in spend_request.purchases:
try:
budget_line = budget.path(purchase.budget_line)
except KeyError:
# TODO: Move this check up into the parser so it's caught and the
# user can fix it
print('Budget line "{0}" not found', file=sys.stderr)
sys.exit(1)
bl_request_total = D(0)
ticket_text += """
=== Items from [budget:{budget_line}] ===
{summary} \\\\
||= '''Item''' =||= '''Cost''' =||
""".format(budget_line=purchase.budget_line, summary=purchase.summary)
for item in purchase.items:
ticket_text += "|| {desc} || £{cost} ||\n".format(desc=item.desc,
cost=item.cost)
bl_request_total += item.cost
# How much has already been spent against this budget line
spent = budget_line.spent
req_total = spent + bl_request_total
# It is over the limit?
if req_total > budget_line.cost:
print(
"Warning: This purchase exceeds the budget line '{0}'".format(
purchase.budget_line))
print("\tBudget line's value: £{0}".format(budget_line.cost))
print("\tRequested Expenditure: £{0} ({1}%)".format(
req_total, 100 * (req_total) / budget_line.cost))
if not input("Continue anyway? [y/N] ").lower() == 'y':
sys.exit()
if purchase.budget_line not in budget_line_totals:
budget_line_totals[purchase.budget_line] = D(0)
budget_line_totals[purchase.budget_line] += bl_request_total
ticket_text += """
=== Budget Line Totals ===
||= '''Budget Line''' =||= '''Total''' =||
"""
for line, total in budget_line_totals.items():
ticket_text += "|| [budget:{line}] || £{total} ||\n" \
.format(line=line, total=total)
print(ticket_text)
if args.dry_run:
print("Stopping before actually creating ticket.")
sys.exit(0)
try:
server = TracProxy(server=args.server, port=args.port)
except WrongServer:
print("Error: The specified server is not a Trac instance",
file=sys.stderr)
sys.exit(1)
ticketNum = server.ticket.create(spend_request.summary, ticket_text, {
'component': "Purchasing",
'owner': "treasurer",
'type': "task"
})
if not ticketNum > 0:
print("Unable to create a valid ticket")
sys.exit()
if args.port == 443:
hostname = args.server
else:
hostname = "{0}:{1}".format(args.server, args.port)
print("Purchasing ticket created: https://{0}/trac/ticket/{1}".format(
hostname, ticketNum))
print("Spending Entries:")
for purchase in spend_request.purchases:
print()
today = datetime.date.today()
i = "{date} ! {summary}\n"
i += " {account}\t£{amount}\n"
i += " Liabilities:{payee}\n"
i += " ; trac: #{trac}"
print(
i.format(date=today.isoformat(),
summary=purchase.summary,
account=srspending.budget_line_to_account(
purchase.budget_line),
amount=sum([x.cost for x in purchase.items]),
payee=spend_request.username,
trac=ticketNum))
