can choose store

workers/commandor_page.py

@@ -47,6 +47,15 @@ class CommandorPage:
         self.logger = logging.getLogger("CommandorPage")
         self.is_finished = False
         self.contact = contact
+        # 0: random
+        # 1: faubourg
+        # 2: George
+        # 3: Sèvres
+        self.store_map = {
+            1: "faubourg",
+            2: "georgev",
+            3: "sevres"
+        }
         self.store_type = store_type
 
     def on_success(self, result: ReserveResultPojo):
@@ -159,10 +168,11 @@ class CommandorPage:
                 document.getElementById("phone_country").focus();
                 document.getElementById("phone_country").value = \"FR\" }""")
             else:
-                self.page.evaluate("""()=>{
-                document.getElementById("prefer").value = \"faubourg\";
+                store_to_choose = self.store_map[self.store_type]
+                self.page.evaluate("""(store_to_choose)=>{
+                document.getElementById("prefer").value = store_to_choose;
                 document.getElementById("phone_country").focus();
-                document.getElementById("phone_country").value = \"FR\" }""")
+                document.getElementById("phone_country").value = \"FR\" }""", store_to_choose)
         except Exception as error:
             self.logger.error(error)