tariochbctools 1.3.0__py2.py3-none-any.whl → 1.4.1__py2.py3-none-any.whl

tariochbctools/importers/awardwalletimp/config.py

@@ -0,0 +1,93 @@
+import argparse
+import sys
+import uuid
+from typing import Any
+
+import yaml
+from awardwallet import AwardWalletClient
+from awardwallet.api import AccessLevel
+
+
+def get_link_url(client):
+    connection_url = client.get_connection_link(
+        platform="desktop",
+        access_level=AccessLevel.READ_ALL_EXCEPT_PASSWORDS,
+        state=str(uuid.uuid4()),
+    )
+    print(  # noqa: T201
+        "Redirect your user to this URL to authorize the connection."
+        "\nThis link expires in 10 minutes:"
+    )
+    print(connection_url)  # noqa: T201
+
+
+def generate(client):
+    """
+    Generate a config for a user including user_id and account_id list.
+    Output in yaml format.
+    """
+    config = {}
+    config["api_key"] = client.api_key
+    config["users"] = {}
+
+    connected_users = client.list_connected_users()
+
+    for user in connected_users:
+        user_id = user["userId"]
+        user_details = client.get_connected_user_details(user_id)
+        account_config = {}
+
+        for account in user_details.get("accounts", []):
+            account_config[account["accountId"]] = {
+                "provider": account["displayName"],
+                "account": "Assets:Current:Points",  # Placeholder, user should adjust
+                "currency": "POINTS",
+            }
+
+        config["users"][user_id] = {
+            "name": user["userName"],
+            "all_history": False,
+            "accounts": account_config,
+        }
+
+    yaml.dump(config, sys.stdout, sort_keys=False)
+
+
+def parse_args(args: Any) -> Any:
+    parser = argparse.ArgumentParser(description="awardwallet-config")
+    sub_parsers = parser.add_subparsers(dest="mode", required=True)
+
+    parser.add_argument(
+        "--api-key",
+        required=True,
+        help="API key, can be generated on AwardWallet Business interface",
+    )
+
+    sub_parsers.add_parser(
+        "get_link_url", help="Generate a connection link URL for user authorization"
+    )
+    sub_parsers.add_parser(
+        "generate", help="Generate a configuration template for connected users"
+    )
+
+    return parser.parse_args(args)
+
+
+def main(args: Any) -> None:
+    args = parse_args(args)
+
+    client = AwardWalletClient(args.api_key)
+
+    if args.mode == "get_link_url":
+        get_link_url(client)
+    elif args.mode == "generate":
+        generate(client)
+
+
+def run() -> None:
+    """Entry point for console_scripts"""
+    main(sys.argv[1:])
+
+
+if __name__ == "__main__":
+    main(sys.argv[1:])

tariochbctools/importers/awardwalletimp/importer.py

@@ -0,0 +1,160 @@
+import logging
+from os import path
+from typing import Any
+
+import beangulp
+import dateutil.parser
+import yaml
+from awardwallet import AwardWalletClient
+from beancount.core import amount, data
+from beancount.core.number import D
+
+
+class Importer(beangulp.Importer):
+    """An importer for AwardWallet"""
+
+    def _configure(self, filepath: str, existing: data.Entries) -> None:
+        with open(filepath, "r") as f:
+            self.config = yaml.safe_load(f)
+        self.api_key = self.config["api_key"]
+
+    def identify(self, filepath: str) -> bool:
+        return path.basename(filepath).endswith("awardwallet.yaml")
+
+    def account(self, filepath: str) -> data.Account:
+        return ""
+
+    def extract(self, filepath: str, existing: data.Entries = None) -> data.Entries:
+        self._configure(filepath, existing)
+        client = AwardWalletClient(self.api_key)
+        entries = []
+
+        for user_id, user in self.config["users"].items():
+            user_details = client.get_connected_user_details(user_id)
+
+            if user.get("accounts"):
+                if user.get("all_history", False):
+                    entries.extend(self._extract_account_history(user, client))
+                else:
+                    entries.extend(self._extract_user_history(user, user_details))
+            else:
+                logging.warning(
+                    "Ignoring user ID %s: no accounts configured",
+                    user_id,
+                )
+
+        return entries
+
+    def _extract_user_history(
+        self, user: dict, user_details: dict
+    ) -> list[data.Transaction]:
+        """
+        User history is limited to the last 10 history elements per account
+        """
+        entries = []
+        for account in user_details["accounts"]:
+            account_id = account["accountId"]
+
+            if account_id in user["accounts"]:
+                logging.info("Extracting account ID %s", account_id)
+                account_config = user["accounts"][account_id]
+
+                entries.extend(
+                    self._extract_transactions(
+                        account["history"], account_config, account_id
+                    )
+                )
+            else:
+                logging.warning(
+                    "Ignoring account ID %s: %s", account_id, account["displayName"]
+                )
+        return entries
+
+    def _extract_account_history(
+        self, user: dict, client: AwardWalletClient
+    ) -> list[data.Transaction]:
+        entries = []
+        for account_id, account_config in user["accounts"].items():
+            logging.info("Extracting account ID %s", account_id)
+            account = client.get_account_details(account_id)["account"]
+
+            entries.extend(
+                self._extract_transactions(
+                    account["history"], account_config, account_id
+                )
+            )
+        return entries
+
+    def _extract_transactions(
+        self,
+        history: list,
+        account_config: dict,
+        account_id: str,
+    ) -> list[data.Transaction]:
+        local_account = account_config["account"]
+        currency = account_config["currency"]
+
+        logging.debug(
+            "Extracting %i transactions for account %s",
+            len(history),
+            account_id,
+        )
+
+        entries = []
+        for trx in history:
+            entries.extend(
+                self._extract_transaction(trx, local_account, currency, account_id)
+            )
+        return entries
+
+    def _extract_transaction(
+        self,
+        trx: dict[str, Any],
+        local_account: data.Account,
+        currency: str,
+        account_id: str,
+    ) -> list[data.Transaction]:
+        entries = []
+        metakv = {"account-id": str(account_id)}
+
+        trx_date = None
+        trx_description = None
+        trx_amount = None
+
+        for f in trx.get("fields", []):
+            if f["code"] == "PostingDate":
+                trx_date = dateutil.parser.parse(f["value"]["value"]).date()
+            if f["code"] == "Description":
+                trx_description = f["value"]["value"].replace("\n", " ")
+            if f["code"] == "Miles":
+                trx_amount = D(f["value"]["value"])
+            if f["code"] == "Info":
+                name = f["name"].lower().replace(" ", "-")
+                metakv[name] = f["value"]["value"].replace("\n", " ")
+
+        assert trx_date
+        assert trx_description
+        assert trx_amount is not None, f"No amount in trx: {trx}"
+
+        meta = data.new_metadata("", 0, metakv)
+        entry = data.Transaction(
+            meta,
+            trx_date,
+            "*",
+            "",
+            trx_description,
+            data.EMPTY_SET,
+            data.EMPTY_SET,
+            [
+                data.Posting(
+                    local_account,
+                    amount.Amount(trx_amount, currency),
+                    None,
+                    None,
+                    None,
+                    None,
+                ),
+            ],
+        )
+        entries.append(entry)
+        return entries

{tariochbctools-1.3.0.dist-info → tariochbctools-1.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tariochbctools
-Version: 1.3.0
+Version: 1.4.1
 Summary: Importers, plugins and price fetchers for Beancount
 Home-page: https://github.com/tarioch/beancounttools/
 Author: Patrick Ruckstuhl
@@ -16,10 +16,10 @@ Classifier: Intended Audience :: Developers
 Classifier: Topic :: Office/Business :: Financial
 Classifier: Topic :: Office/Business :: Financial :: Accounting
 Classifier: Topic :: Office/Business :: Financial :: Investment
-Classifier: License :: OSI Approved :: MIT License
 Description-Content-Type: text/x-rst; charset=UTF-8
 License-File: LICENSE.txt
 Requires-Dist: importlib-metadata; python_version < "3.8"
+Requires-Dist: awardwallet
 Requires-Dist: beancount>=3
 Requires-Dist: beangulp
 Requires-Dist: beanprice
@@ -29,7 +29,7 @@ Requires-Dist: pyyaml
 Requires-Dist: ibflex
 Requires-Dist: requests
 Requires-Dist: camelot-py
-Requires-Dist: opencv-python
+Requires-Dist: opencv-python-headless
 Requires-Dist: blockcypher
 Requires-Dist: imap-tools
 Requires-Dist: undictify

{tariochbctools-1.3.0.dist-info → tariochbctools-1.4.1.dist-info}/RECORD

@@ -1,5 +1,8 @@
 tariochbctools/__init__.py,sha256=pH5i4Fj1tbXLqLtTVIdoojiplZssQn0nnud8-HXodRE,577
 tariochbctools/importers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tariochbctools/importers/awardwalletimp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tariochbctools/importers/awardwalletimp/config.py,sha256=EPwvUE9WBdCk9xxleJ8qDLfPy8GwcgAtG1uGSShyt4w,2460
+tariochbctools/importers/awardwalletimp/importer.py,sha256=0G3DcxYP_chy1zUYMCijDOwNZBBBdX8hR2sPKiVHZS8,5071
 tariochbctools/importers/bcge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tariochbctools/importers/bcge/importer.py,sha256=bAQpkBLURatdsfSoMQdiha-8QfExAXXvpzAgnFFMwoE,1176
 tariochbctools/importers/bitst/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -50,9 +53,9 @@ tariochbctools/plugins/check_portfolio_sum.py,sha256=naJ2j6BFpQhJhT2c-gfjyIdcYe0
 tariochbctools/plugins/generate_base_ccy_prices.py,sha256=4CDzUosjMWCZfsBJMLrf-i5WNCHexI2rdm5vIDYW-AI,1314
 tariochbctools/plugins/prices/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tariochbctools/plugins/prices/ibkr.py,sha256=GYCjnlF-MK-ZFPEr0M6T4iO3Etq0tMmrMlsHGInXUO8,1405
-tariochbctools-1.3.0.dist-info/licenses/LICENSE.txt,sha256=VR2hkz3p9Sw4hSXc7S5iZTOXGeV4h-i8AO_q0zEmtkE,1074
-tariochbctools-1.3.0.dist-info/METADATA,sha256=qafViFxIs-0JxGrPrvw9iHUCxRKHch1hdh7G1IXXwpM,2190
-tariochbctools-1.3.0.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
-tariochbctools-1.3.0.dist-info/entry_points.txt,sha256=9xrCCY1wx2zCIsQUOWZelLHDmHw9Oc-ZBAKUIY9lKTA,88
-tariochbctools-1.3.0.dist-info/top_level.txt,sha256=CiA_NepCI6zDNsaORA55zmpuJFSnTvLESraIL13xiOQ,15
-tariochbctools-1.3.0.dist-info/RECORD,,
+tariochbctools-1.4.1.dist-info/licenses/LICENSE.txt,sha256=VR2hkz3p9Sw4hSXc7S5iZTOXGeV4h-i8AO_q0zEmtkE,1074
+tariochbctools-1.4.1.dist-info/METADATA,sha256=x5jAZ6K1do4OMGOoJYbteq1nf3Qw4dW1DR7693B1XCk,2175
+tariochbctools-1.4.1.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
+tariochbctools-1.4.1.dist-info/entry_points.txt,sha256=bo7wO1u-PIDHNuqsTEekC56VCAmn2i2vTRcKXxqc770,158
+tariochbctools-1.4.1.dist-info/top_level.txt,sha256=CiA_NepCI6zDNsaORA55zmpuJFSnTvLESraIL13xiOQ,15
+tariochbctools-1.4.1.dist-info/RECORD,,

{tariochbctools-1.3.0.dist-info → tariochbctools-1.4.1.dist-info}/entry_points.txt

@@ -1,2 +1,3 @@
 [console_scripts]
+awardwallet-conf = tariochbctools.importers.awardwalletimp.config:run
 nordigen-conf = tariochbctools.importers.nordigen.nordigen_config:run