first commit

src/rewards_tasks.py

@@ -0,0 +1,196 @@
+import os
+import random
+import time
+from typing import Callable
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.remote.webelement import WebElement
+from selenium.webdriver.common.action_chains import ActionChains
+import tab_utils
+import llm_utils
+import mouse_trajectory
+import mimic_typing
+import element_selectors
+
+class RewardsTaskUtils:
+	def __init__(self, driver: webdriver.Edge):
+		self.driver = driver
+		self.tab_utils = tab_utils.TabUtils(driver)
+		self.mouse = mouse_trajectory.MouseUtils(driver)
+		self.keyboard = mimic_typing.KeyboardUtils(driver)
+		self.elements = element_selectors.ElementSelectionUtils(driver)
+
+	def find_element(self, xpath: str):
+		return self.driver.find_element(By.XPATH, xpath)
+
+	def wait_for_element(self, element_getter: Callable[[], WebElement | list[WebElement]], timeout: int = 10) -> WebElement | list[WebElement]:
+		def condition(_: webdriver.Edge):
+			try:
+				element_or_elements = element_getter()
+
+				return element_or_elements
+			except:
+				return False
+
+		return WebDriverWait(self.driver, timeout).until(condition)
+
+	def switch_to_earn_page(self):
+		self.move_to_and_click(self.elements.get_earn_tab())
+
+	def switch_to_dashboard(self):
+		self.move_to_and_click(self.elements.get_dashboard_tab())
+
+	def move_to_and_click(self, elem: WebElement):
+		self.mouse.move_to_element(elem)
+		self.mouse.human_like_click()
+
+	def wait_for_then_click(self, element_getter: Callable[[], WebElement], timeout: int = 10):
+		elem = self.wait_for_element(element_getter, timeout)
+		self.move_to_and_click(elem)
+
+	def complete_bing_daily_set(self):
+		self.switch_to_earn_page()
+
+		self.wait_for_then_click(self.elements.get_open_daily_set_button)
+
+		daily_set_links = self.wait_for_element(self.elements.get_daily_set_elements)
+
+		self.move_to_and_click(daily_set_links[0])
+		time.sleep(random.uniform(2, 3))
+		self.driver.switch_to.window(self.driver.current_window_handle) # refocus on the main tab
+
+		self.move_to_and_click(daily_set_links[1])
+		time.sleep(random.uniform(2, 3))
+		self.driver.switch_to.window(self.driver.current_window_handle)
+
+		self.move_to_and_click(daily_set_links[2])
+		time.sleep(random.uniform(2, 3))
+		self.driver.switch_to.window(self.driver.current_window_handle)
+
+		self.tab_utils.close_all_other_tabs()
+
+	def complete_explore_on_bing_tasks(self):
+		self.switch_to_earn_page()
+
+		explore_on_bing_links = self.wait_for_element(self.elements.get_explore_on_bing_elements)
+
+		for card in explore_on_bing_links:
+			desc = self.elements.extract_card_descriptions(card)
+			query = llm_utils.get_search_query_from_task_description(desc)
+
+			self.move_to_and_click(card)
+			self.tab_utils.switch_to_other_tab()
+
+			self.wait_for_element(self.elements.get_bing_search_bar)
+
+			# search bar should be auto-focused
+
+			self.keyboard.send_keys(query+Keys.ENTER)
+
+			time.sleep(random.uniform(2, 3))
+
+			self.tab_utils.switch_to_other_tab()
+			self.tab_utils.close_all_other_tabs()
+
+		for card in explore_on_bing_links:
+			if not self.elements.card_is_complete(card):
+				print(f"WARNING: Explore on Bing Card [desc={self.elements.extract_card_descriptions(card)!r}] is not complete after searching. Please check manually.")
+
+	def complete_visual_search(self):
+		self.switch_to_earn_page()
+
+		self.wait_for_then_click(self.elements.get_open_visual_search_sidebar)
+
+		self.wait_for_then_click(self.elements.get_search_now_link_from_visual_search_sidebar)
+
+		self.tab_utils.switch_to_other_tab()
+
+		self.mouse.reinitialize()
+
+		self.wait_for_then_click(self.elements.get_visual_search_button)
+
+		file_input = self.wait_for_element(self.elements.get_visual_search_file_input)
+
+		file_input.send_keys(os.path.abspath("keypress_times.png"))
+
+		time.sleep(random.uniform(3, 5))
+
+		self.tab_utils.switch_to_other_tab()
+		self.tab_utils.close_all_other_tabs()
+
+		self.mouse.reinitialize()
+
+	def complete_misc_cards(self):
+		self.switch_to_earn_page()
+
+		misc_cards: list[WebElement] = self.wait_for_element(self.elements.get_all_misc_cards)
+
+		for card in misc_cards:
+			while not self.elements.element_is_fully_in_viewport(card): # this should work for top-down iteration
+				ActionChains(self.driver).scroll_by_amount(0, 100).perform()
+
+			if not self.elements.card_is_complete(card) and self.elements.get_card_point_value(card) > 0:
+				self.move_to_and_click(card)
+				time.sleep(random.uniform(1, 2))
+				self.driver.switch_to.window(self.driver.current_window_handle)
+
+		for card in misc_cards:
+			if not self.elements.card_is_complete(card) and self.elements.get_card_point_value(card) > 0:
+				print(f"WARNING: Misc Card [desc={self.elements.extract_card_descriptions(card)!r}] is not complete after clicking. Please check manually.")
+
+		self.tab_utils.close_all_other_tabs()
+
+	def complete_twenty_searches(self):
+		self.driver.get("https://www.bing.com/")
+
+		search_bar = self.wait_for_element(self.elements.get_bing_search_bar)
+
+		# search bar should be auto-focused
+
+		for query in llm_utils.get_related_search_queries(
+			llm_utils.get_random_noun(), num_queries=20
+		):
+			self.keyboard.send_keys(query+Keys.ENTER)
+
+			time.sleep(random.uniform(2, 3))
+
+			# clear search bar for next query
+			# the 't' can be any character, it just needs to be there to
+			# auto-focus the search bar so that the backspaces will work
+			self.keyboard.send_keys('t'+Keys.BACKSPACE*(len(query)+1))
+
+		self.driver.get("https://rewards.bing.com/")
+
+		self.mouse.reinitialize()
+
+		self.switch_to_earn_page()
+
+		self.wait_for_then_click(self.elements.get_points_breakdown_button)
+
+		close_btn = self.wait_for_element(self.elements.get_close_button_on_points_breakdown)
+
+		points_earned, max_pts = self.elements.get_points_earned_from_searches_on_points_breakdown()
+
+		self.move_to_and_click(close_btn)
+
+		print(f"Points earned from 20 searches: {points_earned}/{max_pts}")
+
+	def claim_bonus_points(self):
+		self.switch_to_dashboard()
+
+		self.wait_for_then_click(self.elements.get_bonus_button_on_dashboard)
+
+		try:
+			self.move_to_and_click(self.elements.get_claim_bonus_points_button())
+		except IndexError:
+			print("[WARNING] Could not find the 'Claim Bonus Points' button. There are likely no bonus points to claim at this time.")
+
+	def complete_all_tasks(self):
+		self.complete_bing_daily_set()
+		self.complete_explore_on_bing_tasks()
+		self.complete_visual_search()
+		self.complete_misc_cards()
+		self.complete_twenty_searches()
+		self.claim_bonus_points()