From d742fd8d71bc4d90552df9904a74a7d7be510b75 Mon Sep 17 00:00:00 2001
From: Carl Meyer <carl@oddbird.net>
Date: Mon, 21 Dec 2020 14:04:49 -0700
Subject: [PATCH] Schwab CSV: parse per-account positions CSV too.

---
 beancount_import/source/schwab_csv.py            | 16 ++++++++++++----
 .../test_basic/import_results.beancount          | 15 +++++++++++++++
 .../Intelligent-Positions-2020-11-30.CSV         |  6 ++++++
 3 files changed, 33 insertions(+), 4 deletions(-)
 create mode 100644 testdata/source/schwab_csv/test_basic/positions/Intelligent-Positions-2020-11-30.CSV

diff --git a/beancount_import/source/schwab_csv.py b/beancount_import/source/schwab_csv.py
index 17186055..50aa028c 100644
--- a/beancount_import/source/schwab_csv.py
+++ b/beancount_import/source/schwab_csv.py
@@ -915,7 +915,7 @@ def _load_transactions(filename: str) -> List[RawEntry]:
 
 
 POSITIONS_TITLE_RE = re.compile(
-    r'"Positions for All-Accounts as of (?P<time>.+), (?P<date>.+)"'
+    r'"Positions for (account )?(?P<account>.+) as of (?P<time>.+), (?P<date>.+)"'
 )
 POSITIONS_ACCT_RE = re.compile(r'"(?P<account>.+)"')
 
@@ -930,13 +930,21 @@ def _load_positions(filename: str) -> Sequence[RawPosition]:
         title = csvfile.readline()
         match = POSITIONS_TITLE_RE.match(title)
         assert match, title
+        groups = match.groupdict()
         date = _convert_date(match.groupdict()["date"])
+        if groups["account"] == "All-Accounts":
+            expect_account_headers = True
+            looking_for_account = True
+            account = None
+        else:
+            expect_account_headers = False
+            looking_for_account = False
+            account = groups["account"]
+            line_count += 1
 
-        account = None
-        looking_for_account = True
         for line in csvfile:
             if not line.strip():
-                looking_for_account = True
+                looking_for_account = expect_account_headers
                 empty_lines += 1
                 continue
             if looking_for_account:
diff --git a/testdata/source/schwab_csv/test_basic/import_results.beancount b/testdata/source/schwab_csv/test_basic/import_results.beancount
index 941d0a47..55c5817d 100644
--- a/testdata/source/schwab_csv/test_basic/import_results.beancount
+++ b/testdata/source/schwab_csv/test_basic/import_results.beancount
@@ -294,3 +294,18 @@
 ;; info: {"filename": "<testdata>/test_basic/positions/All-Accounts-Positions-2020-11-15.CSV", "line": 12, "type": "text/csv"}
 
 2020-11-15 price SCHA                                78.97 USD
+
+;; date: 2020-12-21
+;; info: {"filename": "<testdata>/test_basic/positions/Intelligent-Positions-2020-11-30.CSV", "line": 4, "type": "text/csv"}
+
+2020-12-21 balance Assets:Schwab:Intelligent-4321:SCHA             110 SCHA
+
+;; date: 2020-12-21
+;; info: {"filename": "<testdata>/test_basic/positions/Intelligent-Positions-2020-11-30.CSV", "line": 5, "type": "text/csv"}
+
+2020-12-21 balance Assets:Schwab:Intelligent-4321:Cash             2953.34 USD
+
+;; date: 2020-12-21
+;; info: {"filename": "<testdata>/test_basic/positions/Intelligent-Positions-2020-11-30.CSV", "line": 4, "type": "text/csv"}
+
+2020-12-21 price SCHA                                88.71 USD
diff --git a/testdata/source/schwab_csv/test_basic/positions/Intelligent-Positions-2020-11-30.CSV b/testdata/source/schwab_csv/test_basic/positions/Intelligent-Positions-2020-11-30.CSV
new file mode 100644
index 00000000..f35a3df7
--- /dev/null
+++ b/testdata/source/schwab_csv/test_basic/positions/Intelligent-Positions-2020-11-30.CSV
@@ -0,0 +1,6 @@
+"Positions for account Intelligent XXXX-4321 as of 02:45 AM ET, 12/21/2020"
+
+"Symbol","Description","Quantity","Price","Price Change $","Price Change %","Market Value","Day Change $","Day Change %","Cost Basis","Gain/Loss $","Gain/Loss %","Reinvest Dividends?","Capital Gains?","% Of Account","Dividend Yield","Last Dividend","Ex-Dividend Date","P/E Ratio","52 Week Low","52 Week High","Volume","Intrinsic Value","In The Money","Security Type",
+"SCHA","SCHWAB US SMALL CAP ETF","110","$88.71","-$0.26","-0.29%","$9,758.10","-$2.86","-0.29%","$7,500.00","$2,258.10","+43.7%","No","--","5.24%","+1%","$0.30","12/10/2020","--","$43.05","$89.52","431,291","--","--","ETFs & Closed End Funds",
+"Cash & Cash Investments","--","--","--","--","--","$2,953.34","$0.00","0%","--","--","--","--","--","13.97%","--","--","--","--","--","--","--","--","--","Cash and Money Market",
+"Account Total","--","--","--","--","--","$12,711.44","-$2.86","-0.21%","$7,500.00","$2,258.10","+14.73%","--","--","--","--","--","--","--","--","--","--",