Update schedule

scripts/gen_schedule.py

@@ -8,6 +8,7 @@ import sys
 import json
 
 from xml.dom import minidom
+from dateutil.relativedelta import relativedelta
 
 MAPS = ['FRANCE', 'INNSBRUCK', 'LONDON', 'MAKURIISLANDS', 'RICHMOND', 'SCOTLAND']
 
@@ -15,9 +16,13 @@ dom = minidom.parseString('<MapSchedule><appointments></appointments><VERSION>1<
 appts = dom.getElementsByTagName('appointments')[0]
 
 now = datetime.datetime.now(datetime.timezone.utc).replace(day=1)
+maps_loop = []
 prev_map = None
-for i in range(0, 500):
-    map_choice = random.choice(MAPS)
+for _ in range(0, 200):
+    if not maps_loop:
+        maps_loop = list(MAPS)
+        random.shuffle(maps_loop)
+    map_choice = maps_loop.pop()
     while map_choice == prev_map:
         map_choice = random.choice(MAPS)
     prev_map = map_choice
@@ -38,20 +43,46 @@ CLIMBS = [x['road'] for x in data]
 dom = minidom.parseString('<PortalRoads><PortalRoadSchedule><appointments></appointments><VERSION>1</VERSION></PortalRoadSchedule></PortalRoads>')
 appts = dom.getElementsByTagName('appointments')[0]
 
-now = datetime.datetime.now(datetime.timezone.utc)
+month = datetime.datetime.now(datetime.timezone.utc).replace(day=1)
+climbs_loop = []
+climbs_loop_month = []
 prev_climb = None
-for i in range(0, 100):
-    climb_choice = random.choice(CLIMBS)
-    while climb_choice == prev_climb:
-        climb_choice = random.choice(CLIMBS)
-    prev_climb = climb_choice
+prev_climb_month = None
+for _ in range(0, 13):
+    if not climbs_loop_month:
+        climbs_loop_month = list(CLIMBS)
+        random.shuffle(climbs_loop_month)
+    climb_choice_month = climbs_loop_month.pop()
+    while climb_choice_month == prev_climb_month:
+        climb_choice_month = random.choice(CLIMBS)
+    prev_climb_month = climb_choice_month
     appt = dom.createElement('appointment')
-    appt.setAttribute('road', climb_choice)
+    appt.setAttribute('road', climb_choice_month)
+    appt.setAttribute('world', '10')
+    appt.setAttribute('portal_of_month', 'true')
     appt.setAttribute('portal', '0')
-    appt.setAttribute('start', now.strftime("%Y-%m-%dT00:01-04"))
+    appt.setAttribute('start', month.strftime("%Y-%m-%dT00:01-04"))
 
     appts.appendChild(appt)
-    now += datetime.timedelta(days=2)
+    day = month
+    month += relativedelta(months=+1)
+
+    for _ in range(0, 10):
+        if not climbs_loop:
+            climbs_loop = list(CLIMBS)
+            random.shuffle(climbs_loop)
+        climb_choice = climbs_loop.pop()
+        while climb_choice == prev_climb or climb_choice == climb_choice_month:
+            climb_choice = random.choice(CLIMBS)
+        prev_climb = climb_choice
+        appt = dom.createElement('appointment')
+        appt.setAttribute('road', climb_choice)
+        appt.setAttribute('world', '1')
+        appt.setAttribute('portal', '0')
+        appt.setAttribute('start', day.strftime("%Y-%m-%dT00:01-04"))
+
+        appts.appendChild(appt)
+        day += datetime.timedelta(days=3)
 
 with open('PortalRoadSchedule_v1.xml', 'w') as f:
     f.write(dom.toprettyxml())