up reqs

budgetmail.py

@@ -0,0 +1,39 @@
+import calendar
+import datetime
+
+import firefly.budgets
+
+
+def _get_remaining_days():
+    today = datetime.datetime.combine(datetime.date.today(), datetime.datetime.min.time())
+    endofmonth = today.replace(day=calendar.monthrange(today.year, today.month)[1])
+    return (endofmonth - today).days + 1
+
+
+def _collect_budget_data():
+    budgets = []
+    for budget in firefly.budgets._get_budgets():
+        budgets.append({
+            "id": budget.get('id'),
+            "name": budget.get('attributes').get('name'),
+            "limit": firefly.budgets._get_current_limit(budget.get('id')).get('attributes').get('amount'),
+            "spent": firefly.budgets._get_current_spent_amount(budget.get('id')),
+        })
+        currbudget = budgets[-1]
+        currbudget['remaining'] = (float(currbudget.get('limit')) - float(currbudget.get('spent')))
+        remaining_per_day = currbudget.get('remaining') / _get_remaining_days()
+        currbudget['remaining_per_day'] = 0 if remaining_per_day < 0 else remaining_per_day
+    return budgets
+
+
+def main():
+    for budget in _collect_budget_data():
+        print("Budget: " + budget.get('name'))
+        print("  Limit: " + budget.get('limit'))
+        print("  Spent: {:.2f}".format(budget.get('spent')))
+        print("  Remaining: {:.2f}".format(budget.get('remaining')))
+        print("  Remaining per day: {:.2f}".format(budget.get('remaining_per_day')))
+
+
+if __name__ == "__main__":
+    main()

budgetxmpp.py

@@ -0,0 +1,59 @@
+import calendar
+import datetime
+import xmpp
+import time
+import os
+
+import firefly.budgets
+
+
+def _get_remaining_days():
+    today = datetime.datetime.combine(datetime.date.today(), datetime.datetime.min.time())
+    endofmonth = today.replace(day=calendar.monthrange(today.year, today.month)[1])
+    return (endofmonth - today).days + 1
+
+
+def _collect_budget_data():
+    budgets = []
+    for budget in firefly.budgets._get_budgets():
+        budgets.append({
+            "id": budget.get('id'),
+            "name": budget.get('attributes').get('name'),
+            "limit": firefly.budgets._get_current_limit(budget.get('id')).get('attributes').get('amount'),
+            "spent": firefly.budgets._get_current_spent_amount(budget.get('id')),
+        })
+        currbudget = budgets[-1]
+        currbudget['remaining'] = (float(currbudget.get('limit')) - float(currbudget.get('spent')))
+        remaining_per_day = currbudget.get('remaining') / _get_remaining_days()
+        currbudget['remaining_per_day'] = 0 if remaining_per_day < 0 else remaining_per_day
+    return budgets
+
+
+def main():
+    jid = xmpp.protocol.JID(os.environ.get("FIREFLY_JABBER_ID"))
+    connection = xmpp.Client(server=jid.getDomain())
+    connection.connect()
+    connection.auth(
+        user=jid.getNode(),
+        password=os.environ.get("FIREFLY_JABBER_PASSWORD"),
+        resource=jid.getResource(),
+    )
+
+    for budget in _collect_budget_data():
+        message = "Budget: " + budget.get('name')
+        message += "\n  Limit: " + budget.get('limit')
+        message += "\n  Spent: {:.2f}".format(budget.get('spent'))
+        message += "\n  Remaining: {:.2f}".format(budget.get('remaining'))
+        message += "\n  Remaining per day: {:.2f}".format(budget.get('remaining_per_day'))
+
+        connection.send(
+            xmpp.protocol.Message(
+                to=os.environ.get('FIREFLY_JABBER_RECEIVER'),
+                body=message,
+            ),
+        )
+        time.sleep(1)
+
+
+if __name__ == "__main__":
+    main()

firefly/budgets.py

@@ -40,7 +40,7 @@ def _get_current_limit(budget_id):
     today = datetime.datetime.combine(datetime.date.today(), datetime.datetime.min.time()).replace(tzinfo=pytz.UTC)
     return [
         budget for budget in budgets
-        if today >= dateutil.parser.parse(budget.get('attributes').get('start')) and today <= dateutil.parser.parse(budget.get('attributes').get('end'))
+        if today > dateutil.parser.parse(budget.get('attributes').get('start')) and today <= dateutil.parser.parse(budget.get('attributes').get('end'))
     ][0]
 
 

requirements-xmpp.txt

@@ -0,0 +1 @@
+xmpppy==0.6.1