diff --git a/tab b/tab index 535aca5..27f67bb 100755 --- a/tab +++ b/tab @@ -6,11 +6,17 @@ import os import requests -ME = os.getenv("TAB_ME") -TOKEN = os.getenv("TAB_TOKEN") +ME = os.getenv("TAB_ME") +TOKEN = os.getenv("TAB_TOKEN") +BASE_URI = os.getenv("TAB_BASE_URI", default="https://tab.zeus.gent") -TAP_ACCOUNT = os.getenv("LEDGER_TAP_ACCOUNT", "Expenses:Zeuskelder:Drinks and snacks") -FOOD_ACCOUNT = os.getenv("LEDGER_FOOD_ACCOUNT", "Expenses:Zeuskelder:Food") +TAP_ACCOUNT = os.getenv("LEDGER_TAP_ACCOUNT", default="Expenses:Zeuskelder:Drinks and snacks") +FOOD_ACCOUNT = os.getenv("LEDGER_FOOD_ACCOUNT", default="Expenses:Zeuskelder:Food") + +TAB2LEDGER_NAME = "tab2ledger" +TAB2LEDGER_VERSION = "1.0" +TAB2LEDGER_URL = "https://git.zeus.gent/midgard/tab2ledger" +USER_AGENT = os.getenv("TAB2LEDGER_USER_AGENT", default=f"{TAB2LEDGER_NAME} {TAB2LEDGER_VERSION} ({TAB2LEDGER_URL})") def formatted_price(cents): @@ -70,8 +76,11 @@ def formatted_transactions(transactions, final_balance, me): def main(me, token, file=sys.stdout): def get(path): return requests.get( - f"https://tab.zeus.gent{path}", - headers={"Accept": "application/json", "Authorization": f"Token token={token}"} + f"{BASE_URI}{path}", + headers={ + "Accept": "application/json", "Authorization": f"Token token={token}", + "User-Agent": USER_AGENT + } ).json() balance = get(f"/users/{me}")["balance"]