beeminder_add_data.py

#!/usr/bin/env python3
# beeminder_add_data.py
# Created on: Apr 12 2021
# Created by: gojun077@gmail.com
# Last Updated: Aug 21 2021
# Contributors: gojun077@gmail.com
#
# A helper script to increment a beeminder goal by + 1.0 as well as send
# Brain Workshop session data in the comments field to the Beeminder API
# when a Dual N-Back session is completed. The Beeminder username,
# goalname, auth_token, and comment string will be read from a json file
# named "beeminder.json" located in the same dir as this script.


import datetime
import json
import requests
from requests.adapters import HTTPAdapter
from requests.packages.urllib3.util.retry import Retry


def submit(comment_moar: str):
    """
    Increment Beeminder goal by + 1.0
    """
    try:
        with open("beeminder.json","r") as f:
            bmndrD = json.load(f)
            url = "https://www.beeminder.com/api/v1/users"
            epoch_utc = int(datetime.datetime.utcnow().timestamp())
            myuser, mygoal = bmndrD["username"], bmndrD["goalname"]
            endpt = f"{url}/{myuser}/goals/{mygoal}/datapoints.json"
            comment = bmndrD["comment"]
            full_comment = f"{comment} | {comment_moar}"
            payload = {"auth_token": bmndrD["auth_token"],
                       "timestamp": epoch_utc,
                       "value": 1.0,
                       "comment": full_comment}
            header = {"Content-Type": "application/json"}

            retry_strategy = Retry(
                total=3,
                status_forcelist=[429, 500, 502, 503, 504],
                method_whitelist=["HEAD", "GET", "PUT", "DELETE", "OPTIONS",
                                  "POST", "TRACE"],
                backoff_factor = 2
            )
            adapter = HTTPAdapter(max_retries=retry_strategy)
            sess = requests.Session()
            sess.mount = ("https://", adapter)
            resp_sess = sess.post(endpt, headers=header,
                                  data=json.dumps(payload), timeout=10,
                                  )
            resp_sess.raise_for_status()
            print("Data successfully submitted to Beeminder")
            resp_data = resp_sess.json()
            print(resp_data)
    except requests.exceptions.ConnectionError as e:
        print(f"ConnError: {e}, payload: {full_comment}")
    except requests.exceptions.Timeout as e:
        print(f"cnxn TIMEOUT: {e}")
    # 4XX, 5XX HTTP errors
    except requests.exceptions.HTTPError as e:
        print(f"status code {resp_sess.status_code} from {url}: {e}")
    # Error Catch-all
    except Exception as ex:
        print(f"error occurred while trying to send data to Beeminder: {ex}")


def main():
    submit()


if __name__ == "__main__":
    main()