From 4617d2d9baa508244a56ea026789fd462b2085c6 Mon Sep 17 00:00:00 2001
From: zevav <zev@averba.ch>
Date: Tue, 23 May 2017 23:09:47 -0400
Subject: [PATCH] get all transactions before the while loop, un-PEP8'd

---
 splurgebot.md |  5 +++++
 splurgebot.py | 41 ++++++++++++++++++++++++++---------------
 2 files changed, 31 insertions(+), 15 deletions(-)
 create mode 100644 splurgebot.md

diff --git a/splurgebot.md b/splurgebot.md
new file mode 100644
index 0000000..0393f1e
--- /dev/null
+++ b/splurgebot.md
@@ -0,0 +1,5 @@
+Budgeting.  You do it, then you stop doing it.  You wake up in a cold sweat and you start tracking your spending again.  It sucks: You stop.
+
+There are tools!  Horrible, awful tools that are glorified spreadsheets which think all your Amazon purchases go under `books`.  Much data entered, many hairs pulled out.
+
+Surely this is fodder for a bot.  
\ No newline at end of file
diff --git a/splurgebot.py b/splurgebot.py
index baef422..1b92a77 100644
--- a/splurgebot.py
+++ b/splurgebot.py
@@ -3,6 +3,7 @@ import os
 import time
 from typing import List
 
+import math
 from twilio.rest import Client as TwilioClient
 
 from plaid import Client as PlaidClient
@@ -12,44 +13,54 @@ ALERT_FOR_TRANSACTIONS_GTE = 500
 MY_CELL = os.getenv('MY_CELL')
 MY_TWILIO_NUM = os.getenv('MY_TWILIO_NUM')
 PLAID_ENV = os.getenv('PLAID_ENV') or 'sandbox'
+
 # this is an access ID from Plaid for a specific user and bank/credit account
 PLAID_ACCESS_ID = os.getenv('PLAID_ACCESS_ID')
 
-plaid_client = PlaidClient(client_id=os.getenv('PLAID_CLIENT_ID'),
-                           secret=os.getenv('PLAID_SECRET'),
-                           public_key=os.getenv('PLAID_PUBLIC_KEY'),
-                           environment=PLAID_ENV)
+plaid_client = PlaidClient(client_id=os.getenv('PLAID_CLIENT_ID'), secret=os.getenv('PLAID_SECRET'),
+                           public_key=os.getenv('PLAID_PUBLIC_KEY'), environment=PLAID_ENV)
 
-twilio_client = TwilioClient(os.getenv('TWILIO_SID'),
-                             os.getenv('TWILIO_TOKEN'))
+twilio_client = TwilioClient(os.getenv('TWILIO_SID'), os.getenv('TWILIO_TOKEN'))
 
 transaction_ids = set()
 
 
+def get_all_transactions() -> List[dict]:
+    today = datetime.date.today().strftime('%Y-%m-%d')
+    total_transactions: int = plaid_client.Transactions.get(PLAID_ACCESS_ID, '1972-01-01', today)['total_transactions']
+    all_transactions = []
+    for page in range(math.ceil(total_transactions / 500)):
+        all_transactions += plaid_client.Transactions.get(PLAID_ACCESS_ID, '1972-01-01', today,
+                                                          offset=page * 500, count=500)['transactions']
+    return all_transactions
+
+
 def get_latest_transactions() -> List[dict]:
     """get 100 most recent transactions"""
     today = datetime.date.today().strftime('%Y-%m-%d')
     return [transaction
-            for transaction in plaid_client.Transactions.get(
-                    PLAID_ACCESS_ID, '1972-01-01', today)['transactions']
+            for transaction in
+            plaid_client.Transactions.get(PLAID_ACCESS_ID, '1972-01-01', today)['transactions']
             if transaction['transaction_id'] not in transaction_ids]
 
 
 def alert(transaction: dict) -> None:
-    message = (
-       'hey, a transaction hit your account that exceeds '
-       f'${ALERT_FOR_TRANSACTIONS_GTE}: '
-       f'{transaction["date"]} {transaction["name"]} ${transaction["amount"]}')
+    message = (f'hey, a transaction hit your account that exceeds ${ALERT_FOR_TRANSACTIONS_GTE}: '
+               f'{transaction["date"]} {transaction["name"]} ${transaction["amount"]}')
 
-    twilio_client.api.account.messages.create(to=MY_CELL,
-                                              from_=MY_TWILIO_NUM,
-                                              body=message)
+    twilio_client.api.account.messages.create(to=MY_CELL, from_=MY_TWILIO_NUM, body=message)
 
 
 def main() -> None:
+    if len(transaction_ids) == 0:
+        transaction_ids.update(transaction['transaction_id'] for transaction in get_all_transactions())
+
     while True:
+
         for transaction in get_latest_transactions():
+
             transaction_ids.add(transaction['transaction_id'])
+
             if transaction['amount'] >= ALERT_FOR_TRANSACTIONS_GTE:
                 alert(transaction)