additional fidelity brokerage format

beancount_reds_importers/importers/fidelity/fidelity_brokerage2_csv.py

@@ -0,0 +1,63 @@
+"""Fidelity Brokerage csv importer for beancount."""
+
+import re
+
+from beancount_reds_importers.libreader import csvreader
+from beancount_reds_importers.libtransactionbuilder import investments
+
+
+class Importer(investments.Importer, csvreader.Importer):
+    IMPORTER_NAME = "Fidelity Cash Management Account 2"
+
+    def custom_init(self):
+        self.max_rounding_error = 0.04
+        self.filename_pattern_def = ".*History"
+        self.date_format = "%m/%d/%Y"
+        self.header_identifier = ""
+        self.column_labels_line = "Run Date,Action,Symbol,Description,Type,Exchange Quantity,Exchange Currency,Quantity,Currency,Price,Exchange Rate,Commission,Fees,Accrued Interest,Amount,Cash Balance,Settlement Date"
+        self.header_map = {
+            "Run Date": "date",
+            "Action": "memo",
+            "Symbol": "security",
+            "Amount": "amount",
+            "Settlement Date": "settleDate",
+            "Quantity": "units",
+            "Accrued Interest": "accrued_interest",
+            "Fees": "fees",
+            "Commission": "commission",
+            "Cash Balance": "balance",
+            "Price": "unit_price",
+        }
+        self.transaction_type_map = {
+            "DIVIDEND RECEIVED": "dividends",
+            "TRANSFERRED FROM": "cash",
+            "YOU BOUGHT": "buystock",
+            "YOU SOLD": "sellstock",
+            "INTEREST EARNED": "income",
+            "REINVESTMENT FIDELITY": "buystock",
+            "SHORT-TERM CAP": "capgainsd_st",
+            "LONG-TERM CAP": "capgainsd_lt",
+        }
+        self.skip_transaction_types = []
+        # fmt: on
+
+    def deep_identify(self, file):
+        last_four = self.config.get("account_number", "")[-4:]
+        return (
+            re.match(self.header_identifier, file.head(), flags=re.DOTALL)
+            and f"{last_four}" in file.name
+        )
+
+    def prepare_table(self, rdr):
+        for field in ["Action", "Symbol", "Description"]:
+            rdr = rdr.convert(field, lambda x: x.lstrip())
+
+        rdr = rdr.addfield("total", lambda x: x["Amount ($)"])
+        rdr = rdr.addfield("tradeDate", lambda x: x["Run Date"])
+        rdr = rdr.cutout("Type")
+        rdr = rdr.capture("Action", "(\\S+(?:\\s+\\S+)?)", ["type"], include_original=True)
+
+        # for field in ["memo"]:
+        #     rdr = rdr.convert(field, lambda x: x.lstrip())
+
+        return rdr