check for fidelity margin account

NelsonDane · Dec 22, 2023 · 2d37faf · 2d37faf
1 parent 76996a2
commit 2d37faf
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/fidelityAPI.py b/fidelityAPI.py
@@ -456,6 +456,17 @@ def fidelity_transaction(fidelity_o: Brokerage, orderObj: stockOrder, loop=None)
                             )
                         price_box.clear()
                         price_box.send_keys(wanted_price)
+                    # Check for margin account
+                    try:
+                        print("Checking for margin account...")
+                        margin_cash = driver.find_element(
+                            by=By.ID, value="tradetype-cash"
+                        )
+                        margin_cash.click()
+                        print("Margin account found!")
+                    except NoSuchElementException:
+                        print("Not a margin account")
+                        pass
                     # Preview order
                     WebDriverWait(driver, 10).until(check_if_page_loaded)
                     sleep(1)