Merge pull request #4 from luisrx7/dev

Add tests and autobet script

Author: luisrx7

.gitignore

@@ -10,3 +10,5 @@ results1.txt
 results2.txt
 .txt
 creds.py
+slices.pickle
+autobet.log

autobet.py

@@ -0,0 +1,66 @@
+import logging
+import time
+
+import rich
+import rich.console
+import rich.traceback
+import sys
+
+from aviator import Aviator
+from strats.custom_strats import DAlembertStrat
+
+rich.traceback.install()
+console = rich.console.Console()
+
+
+logging.basicConfig(format='%(asctime)s %(levelname)s:%(message)s', level=logging.INFO, datefmt='%m/%d/%Y %I:%M:%S', filename="autobet.log")
+
+logging.getLogger().addHandler(logging.StreamHandler(sys.stdout))
+
+
+def signal_handler(sig, frame):
+    global stop
+    print('You pressed Ctrl+C!')
+    stop = True
+    # sys.exit(0)
+
+
+stop = False
+
+
+
+
+
+def main():
+    strat = DAlembertStrat(
+        description="DAlembertStrat",
+        start_balance=3000,
+        base_bet=0.1,
+        max_bet=1,
+        multiplier=2,
+        max_bets=10000,
+    )
+    
+    while stop is False:
+        try:
+            aviator = Aviator(debug=True, strat=strat)
+            aviator.login()
+            aviator.go_to_game()
+            time.sleep(3)
+            aviator.setup_auto_bet()
+
+
+            while aviator.in_game() and stop is False:
+                aviator.wait_for_game_to_finish()
+                last_result = aviator.get_last_game_result()
+                aviator.process_bet(float(last_result))
+                aviator.add_to_log(last_result)
+                print(f"balance: {aviator.get_balance()}")
+        except Exception as e:
+            console.print_exception(show_locals=True)
+            logging.error(e)
+        finally:
+            aviator.close()
+
+if __name__ == "__main__":
+    main()

aviator/aviator.py

@@ -6,6 +6,8 @@
 import time
 from datetime import datetime 
 
+from strats.strats import Strat
+
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.support.wait import WebDriverWait
 from selenium.common.exceptions import StaleElementReferenceException, NoSuchElementException
@@ -18,12 +20,16 @@ class Aviator(Browser):
 
     def __init__(self, headless=False, test=False, remote_driver=True,
                 remote_address="127.0.0.1",remote_port=4446, use_cookies=False,
-                debug=False):
+                debug=False,strat: Strat = None):
         super().__init__(headless= headless, test=test, remote_driver=remote_driver,
                 remote_address=remote_address,remote_port=remote_port,
                   use_cookies=use_cookies, profile_path=vars.profile_path)
 
         self.debug = debug
+        self.strat = strat
+        
+        if self.strat is not None:
+            self.strat.reset()
 
         helium.set_driver(self.driver)
 
@@ -78,11 +84,46 @@ def get_last_game_result(self):
         '''
         # if self.debug:
         #     print("getting last game result")
+
+        element = self.find_elements(By.XPATH, vars.last_game_result, timeout=0.5)
 
+        if element is not None:
+            return element.text.strip().replace("x", "")
+
+
         results = self.get_game_results()
         if len(results) > 0:
             return results[0]
 
+    def process_bet(self, result):
+        '''
+        check if a strat is defined
+        if it is, use it to calculate the next bet
+        if not warn the user that a strat is not defined
+        '''
+        if self.debug:
+            print(f"processing bet for result {result}")
+        if self.strat is None:
+            print("WARNING: no strat defined")
+            return False
+
+        self.strat.calculate_bet(result)
+        if self.debug:
+            print(f"bet: {self.strat.bet}, multiplier: {self.strat.multiplier}")
+
+        if self.strat.bet == 0 or self.strat.multiplier == 0:
+            if self.debug:
+                print("bet or multiplier is 0, not placing bet")
+            return False        
+
+        if self.place_bet(self.strat.bet, self.strat.multiplier) is False:
+            if self.debug:
+                print("could not place bet")
+            return False
+        
+        self.strat.gamble()
+
+
     def get_game_results(self):
         '''
         get game result
@@ -150,15 +191,13 @@ def wait_for_game_to_finish(self):
             print("waiting for game to finish")
         last_result = self.get_last_game_result()
         while True:
-
             if self.get_last_game_result() != last_result:
                 break
             if self.debug:
                 print(".", end="")
-            time.sleep(0.5)
+            time.sleep(0.1)
         if self.debug:
-            print("")
-            print("game finished")
+            print("\ngame finished")
 
     def add_to_log(self, result):
         '''
@@ -170,6 +209,50 @@ def add_to_log(self, result):
         with open("results.txt", "a") as f:
             f.write(f"{datetime.now().strftime('%d-%m-%Y %H:%M:%S')},{result}\n")
 
+
+    def setup_auto_bet(self):
+        '''
+        click the buttons to setup auto cashout
+        '''
+        if self.debug:
+            print("setting up auto bet")
+
+        if self.click_button(vars.bet_type_button) is False:
+            if self.debug:
+                print("could not click bet type button")
+            return False
+        if self.click_button(vars.auto_cashout_button) is False:
+            if self.debug:
+                print("could not click auto cashout button")
+            return False
+        
+
+    def place_bet(self,amount, multiplier):
+        '''
+        place bet with amount and multiplier
+        '''
+        if self.debug:
+            print(f"setting bet amount to {amount} at multiplier {multiplier}")
+
+
+        if self.send_keys(vars.bet_amount_input_box, str(amount)) is False:
+            if self.debug:
+                print("could not set bet amount")
+            return False
+
+        if self.send_keys(vars.multiplier_input_box, str(multiplier)) is False:
+            if self.debug:
+                print("could not set multiplier")
+            return False
+        
+        if self.click_button(vars.place_bet_button) is False:
+            if self.debug:
+                print("could not click place bet button")
+            return False
+        
+        return True
+
+
     def go_to_game(self):
         wait = WebDriverWait(self.driver, 10)
 

aviator/vars.py

@@ -8,9 +8,20 @@
 
 game_name = '//*[@id="slots"]/div[2]/div[1]/div[1]'
 
-last_game_result = 'body > app-root > app-game > div > div.main-container > div.w-100.h-100 > div > div.game-play > div.result-history.disabled-on-game-focused.my-2 > app-stats-widget > div > div.payouts-wrapper > div > app-bubble-multiplier:nth-child($ROUND$)'
-
+last_game_results = 'body > app-root > app-game > div > div.main-container > div.w-100.h-100 > div > div.game-play > div.result-history.disabled-on-game-focused.my-2 > app-stats-widget > div > div.payouts-wrapper > div > app-bubble-multiplier:nth-child($ROUND$)'
+last_game_result = "/html/body/app-root/app-game/div/div[1]/div[2]/div/div[2]/div[1]/app-stats-widget/div/div[1]/div/app-bubble-multiplier[1]/div"
 
 open_new_window_button = '//*[@id="wrapper_game_area"]/div/ul/li[3]/a'
 
-balance = '/html/body/app-root/app-game/div/div[1]/div[1]/app-header/div/div[2]/div[1]/div[1]/div/span[1]'
+balance = '/html/body/app-root/app-game/div/div[1]/div[1]/app-header/div/div[2]/div[1]/div[1]/div/span[1]'
+
+
+#bets
+
+
+bet_type_button = '/html/body/app-root/app-game/div/div[1]/div[2]/div/div[2]/div[3]/app-bet-controls/div/app-bet-control[1]/div/app-navigation-switcher/div/button[2]'
+auto_cashout_button = '/html/body/app-root/app-game/div/div[1]/div[2]/div/div[2]/div[3]/app-bet-controls/div/app-bet-control[1]/div/div[3]/div[2]/div[1]/app-ui-switcher'
+place_bet_button = '/html/body/app-root/app-game/div/div[1]/div[2]/div/div[2]/div[3]/app-bet-controls/div/app-bet-control[1]/div/div[1]/div[2]/button'
+
+bet_amount_input_box = '/html/body/app-root/app-game/div/div[1]/div[2]/div/div[2]/div[3]/app-bet-controls/div/app-bet-control[1]/div/div[1]/div[1]/app-spinner/div/div[2]/input'
+multiplier_input_box = '/html/body/app-root/app-game/div/div[1]/div[2]/div/div[2]/div[3]/app-bet-controls/div/app-bet-control[1]/div/div[3]/div[2]/div[2]/div/app-spinner/div/div[2]/input'

browser/browser.py

@@ -6,6 +6,7 @@
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.support.wait import WebDriverWait
+from selenium.webdriver.common.keys import Keys
 
 import os 
 
@@ -40,6 +41,18 @@ def __init__(self,headless = True,test = False,remote_driver = True,
         self.chrome_options.add_experimental_option('excludeSwitches',
                                                      ['enable-logging'])
 
+        # Adding argument to disable the AutomationControlled flag 
+        self.chrome_options.add_argument("--disable-blink-features=AutomationControlled") 
+        
+        # Exclude the collection of enable-automation switches 
+        self.chrome_options.add_experimental_option("excludeSwitches", ["enable-automation"]) 
+        
+        # Turn-off userAutomationExtension 
+        self.chrome_options.add_experimental_option("useAutomationExtension", False) 
+        
+        # Setting the driver path and requesting a page 
+        
+        
         if self.profile_path != "":
             print("profile path",self.profile_path)
             os.makedirs(self.profile_path, exist_ok=True)
@@ -51,6 +64,9 @@ def __init__(self,headless = True,test = False,remote_driver = True,
             self.driver = webdriver.Chrome(service=ChromeDriverManager().install(), # type: ignore  # noqa: E501
                                            options=self.chrome_options) 
 
+        # Changing the property of the navigator value for webdriver to undefined 
+        self.driver.execute_script("Object.defineProperty(navigator, 'webdriver', {get: () => undefined})") 
+        
         self.driver.implicitly_wait(5)
 
     def find_elements(self,by, value, timeout=5):
@@ -116,4 +132,16 @@ def click_button(self, button_id):
             buttons[0].click()
             return True
         return False
+    
+    def send_keys(self,element_id,keys,clear_first = True):
+        element = self.find_elements(By.XPATH, element_id,timeout=1)
+        if element is None:
+            return False
+        
+        if clear_first:
+            element.send_keys(Keys.CONTROL + "a")
+        
+        element.send_keys(keys)
+        
+        return True
 

docker-compose.yml

@@ -5,3 +5,6 @@ services:
       - "4446:4444"
     # env_file:
     #   - ./.env
+
+    environment:
+      - VNC_NO_PASSWORD=1