Fixed all of the last known bugs

Author: BenTimor

__init__.py

@@ -1,2 +1,2 @@
 from .api import CFD, Invest
-from .utils import Panel
+from .utils import BuyStockMethod, Panel

api.py

@@ -5,7 +5,7 @@
 from selenium.webdriver.support import expected_conditions
 from selenium.webdriver.support.wait import WebDriverWait
 
-from .utils import Mode, Panel, force_click, script_click_xpath, type_sleep
+from .utils import *
 
 
 class Trading212:
@@ -93,14 +93,15 @@ def setup(self, username, password, panel):
         elem = self.driver.find_element_by_class_name("account-menu-button")
         elem.click()
 
-    def buy_stock(self, stock, amount):
+    def buy_stock(self, stock, amount, method=BuyStockMethod.Shares):
         """
         Buying a stock
         :param stock: Stock (Display name)
         :param amount: It may be the amount of the shares and it may be the value of the money. It depends on your Trading212 defaults and I may fix it in the future. For now, You can debug it with headless=False parameter in the constructor.
+        :param method: Shares - For buying the stock by its shares amount | Value - For buying the stock by its value
         """
-        # It's just opening a stock and buying it
         self.open_stock_dialog(stock)
+        self.switch_buying_method(method)
         self.buy(amount)
         sleep(self.long_sleep)
 
@@ -124,6 +125,17 @@ def buy(self, amount):
         else:
             self.driver.find_elements_by_xpath("//div[contains(@class,'confirm-button')]")[0].click()
 
+    def switch_buying_method(self, method):
+        if self.mode == Mode.Invest:
+            force_click(WebDriverWait(self.driver, self.timeout).until(expected_conditions.element_to_be_clickable((By.XPATH, "//div[@class='invest-by-content']"))))
+            if method == BuyStockMethod.Value:
+                # item item-invest-by-items-value
+                # item item-invest-by-items-quantity
+                self.driver.find_element_by_xpath("//div[@class='item item-invest-by-items-value']").click()
+            else:
+                self.driver.find_element_by_xpath("//div[@class='item item-invest-by-items-quantity']").click()
+
+
     def open_stock_dialog(self, stock):
         """
         Opening stock dialog
@@ -194,6 +206,7 @@ def close_position(self, stock):
         # God knows why selenium can't click it. So I'm using javascript hacks.
         quantity = self.position_info(stock, "quantity")
         script_click_xpath(self.driver, f"//td[@class='name' and text()='{stock}']/following::div[@class='button-wrapper']/child::div[@class='sell-button']")
+        self.switch_buying_method(BuyStockMethod.Shares)
         elem = WebDriverWait(self.driver, self.timeout).until(expected_conditions.visibility_of_element_located((By.XPATH,"//div[@class='visible-input']//input")))
         elem.clear()
         type_sleep(elem, quantity, self.short_sleep)

utils.py

@@ -1,5 +1,6 @@
 from time import sleep
 
+
 class Panel:
     """
     Allows you to select between practice and real
@@ -11,6 +12,10 @@ class Mode:
     Invest = "Invest"
     CFD = "CFD"
 
+class BuyStockMethod:
+    Shares = "Shares"
+    Value = "Value"
+
 def force_click(element, sleep_time=1):
     """
     Clicks something until it works
@@ -30,4 +35,4 @@ def type_sleep(element, text, time):
         sleep(time)
 
 def script_click_xpath(driver, xpath):
-    driver.execute_script(f"document.evaluate(\"{xpath}\", document, null, XPathResult.FIRST_ORDERED_NODE_TYPE, null).singleNodeValue.click()")
+    driver.execute_script(f"document.evaluate(\"{xpath}\", document, null, XPathResult.FIRST_ORDERED_NODE_TYPE, null).singleNodeValue.click()")