update scripts

Author: githubfarid1

main.py

@@ -31,6 +31,11 @@
 	from subprocess import CREATE_NEW_CONSOLE
 import json
 
+def savejson(filename, list):
+	with open(filename, "w") as final:
+		json.dump(list.get(0, END), final)
+
+
 class Window(Tk):
 	def __init__(self) -> None:
 		super().__init__()
@@ -105,8 +110,8 @@ def __init__(self, window) -> None:
 
 		titleLabel = TitleLabel(self, 'Main Menu')
 		resybotv1Button = FrameButton(self, window, text="Resy Bot v1", class_frame=ResyBotv1Frame)
-		reservationlistButton = FrameButton(self, window, text="Add Reservation Type", class_frame=AddReservationFrame)
-		periodlistButton = FrameButton(self, window, text="Add Period", class_frame=AddPeriodFrame)
+		reservationlistButton = FrameButton(self, window, text="Update Reservation Type", class_frame=AddReservationFrame)
+		periodlistButton = FrameButton(self, window, text="Update Period", class_frame=AddPeriodFrame)
 
 		# extractButton = FrameButton(self, window, text="Extract PDF Diagram", class_frame=ExtractPdfFrame)
 		# graburlButton = FrameButton(self, window, text="Grab URLs", class_frame=GrabUrlsFrame)
@@ -154,7 +159,7 @@ def __init__(self, window) -> None:
 		self.rowconfigure(6, weight=1)
 		self.rowconfigure(7, weight=1)
 		self.rowconfigure(8, weight=1)
-		titleLabel = TitleLabel(self, text="Add Reservation Type")
+		titleLabel = TitleLabel(self, text="Update Reservation Type")
 		valuentry = Entry(self, width=80)
 		dlist = StringVar(value=RESERVATION_LIST)
 		self.valueslist = Listbox(self, width=80, height=10, listvariable=dlist)
@@ -167,14 +172,14 @@ def __init__(self, window) -> None:
 		valuentry.grid(column = 0, row = 1, sticky=(W))
 		addButton.grid(column = 0, row = 1, sticky = (E))
 		self.valueslist.grid(column = 0, row = 2, sticky=(W))
-		saveButton.grid(column = 0, row = 3, sticky = (W,N))
+		# saveButton.grid(column = 0, row = 3, sticky = (W,N))
 		closeButton.grid(column = 0, row = 8, sticky = (E))
 	def removeValue(self, event):
 		selection = self.valueslist.curselection()
 		for i in self.valueslist.curselection():
 				messagebox.showinfo("Message box", f"`{self.valueslist.get(i)}` deleted..")
 		self.valueslist.delete(selection)
-		
+		savejson(filename="reservationlist.json", list=self.valueslist)
 
 	def savelist(self, **kwargs):
 		with open("reservationlist.json", "w") as final:
@@ -184,6 +189,7 @@ def savelist(self, **kwargs):
 	def addlist(self, **kwargs):
 		kwargs['list'].insert(0, kwargs['entry'].get())
 		kwargs['entry'].delete(0, END)
+		savejson(filename="reservationlist.json", list=kwargs['list'])
 		messagebox.showinfo("Message box","New Reservation added..")
 
 class AddPeriodFrame(ttk.Frame):
@@ -211,7 +217,7 @@ def __init__(self, window) -> None:
 		self.rowconfigure(6, weight=1)
 		self.rowconfigure(7, weight=1)
 		self.rowconfigure(8, weight=1)
-		titleLabel = TitleLabel(self, text="Add Period")
+		titleLabel = TitleLabel(self, text="Update Period")
 		valuentry = Entry(self, width=80)
 		dlist = StringVar(value=PERIOD_LIST)
 		self.valueslist = Listbox(self, width=80, height=10, listvariable=dlist)
@@ -224,14 +230,15 @@ def __init__(self, window) -> None:
 		valuentry.grid(column = 0, row = 1, sticky=(W))
 		addButton.grid(column = 0, row = 1, sticky = (E))
 		self.valueslist.grid(column = 0, row = 2, sticky=(W))
-		saveButton.grid(column = 0, row = 3, sticky = (W,N))
+		# saveButton.grid(column = 0, row = 3, sticky = (W,N))
 		closeButton.grid(column = 0, row = 8, sticky = (E))
+	
 	def removeValue(self, event):
 		selection = self.valueslist.curselection()
 		for i in self.valueslist.curselection():
 				messagebox.showinfo("Message box", f"`{self.valueslist.get(i)}` deleted..")
 		self.valueslist.delete(selection)
-		
+		savejson(filename="periodlist.json", list=self.valueslist)	
 
 	def savelist(self, **kwargs):
 		with open("periodlist.json", "w") as final:
@@ -241,6 +248,7 @@ def savelist(self, **kwargs):
 	def addlist(self, **kwargs):
 		kwargs['list'].insert(0, kwargs['entry'].get())
 		kwargs['entry'].delete(0, END)
+		savejson(filename="periodlist.json", list=kwargs['list'])
 		messagebox.showinfo("Message box","New Period added..")
 
 class ResyBotv1Frame(ttk.Frame):

modules/resybotv1.py

@@ -55,14 +55,16 @@ def random_delay(min_seconds, max_seconds):
 def reserve_restaurant(page, selected_reservation):
     """Reserve the restaurant with improved error handling and explicit waits."""
     try:
+        # breakpoint()
         selected_reservation.click()
         frame_element = page.wait_for_selector('iframe[title="Resy - Book Now"]', timeout=10000)
         frame = frame_element.content_frame()
         frame.wait_for_selector('[data-test-id="order_summary_page-button-book"]', timeout=30000)
+        # time.sleep(2)
         for i in range(5):
             page.mouse.wheel(0, 15000)
-            time.sleep(2)        
-
+            time.sleep(1)        
+        # page.evaluate("() => window.scrollTo(0, document.body.scrollHeight)")
         frame.query_selector('[data-test-id="order_summary_page-button-book"]').click()
         time.sleep(5)
         if frame.query_selector('.StripeForm__header'):
@@ -168,8 +170,9 @@ def main():
                 logging.info(message)
                 print(message)
                 random_delay(2, 5)
+                # breakpoint()
                 page.goto(restaurant_link, wait_until='domcontentloaded')
-                page.wait_for_timeout(20000)
+                # page.wait_for_timeout(20000)
                 page.evaluate("() => document.fonts.ready")
                 random_delay(2, 5)
                 # page.screenshot(path="debugging_photos/screenshot1.png", timeout=120000)
@@ -185,6 +188,7 @@ def main():
                     input(" ".join([message, CLOSE_MESSAGE]))
                     sys.exit()
                 else:
+                    # breakpoint()
                     menu = page.wait_for_selector(f'//div[contains(@class,"ShiftInventory__shift")][h2[text()="{period_wanted.lower()}"]]', timeout=30000)
                     if page.query_selector('//button[contains(@class,"AnnouncementModal__icon-close")]'):
                         page.query_selector('//button[contains(@class,"AnnouncementModal__icon-close")]').click()