forked from socallinuxexpo/scale-sync-scripts
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsync_guidebook.py
More file actions
executable file
·872 lines (794 loc) · 31 KB
/
sync_guidebook.py
File metadata and controls
executable file
·872 lines (794 loc) · 31 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
#!/usr/bin/env python3
#
# Copyright 2018-present Southern California Linux Expo
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#
# Author:: Phil Dibowitz <phil@ipm.com>
#
# Script to sync the website schedule to Guidebook complete with region
# mapping.
#
# By default it'll add only what's missing, but can optionally update all
# existing sessions.
#
# It automatically setups rooms ("Locations") and tracks. It has a hard-coded
# map of colors in the Guidebook class, so if you change tracks you'll need
# to update that.
#
from datetime import datetime
from dateutil import parser
from markdownify import markdownify as md
import click
import json
import logging
import os
import pytz
import re
import requests
import sys
try:
import xdg_base_dirs as xdg
except ImportError:
import xdg
DBASE_DEFAULT = "https://www.socallinuxexpo.org/scale/23x/app"
GUIDE_NAME = "SCaLE 23x"
class OurJSON:
rooms = set()
tracks = set()
sessions_by_name = {}
sessions_by_nid = {}
FIELD_MAPPING = {
"tracks": "Track",
"rooms": "Location",
}
def __init__(self, path, logger):
self.logger = logger
if path.startswith("http://") or path.startswith("https://"):
response = requests.get(path)
blob = response.text
else:
blob = open(path, "r").read()
self.sessions_by_name, self.sessions_by_nid = self.load_json(blob)
def load_json(self, raw):
self.logger.info("Loading JSON file")
raw = json.loads(raw)
data_by_name = {}
data_by_nid = {}
for session in raw:
# handle leading/trailing spaces in names
name = session["Name"].strip()
session["Name"] = name
track = session[self.FIELD_MAPPING["tracks"]].strip()
room = session[self.FIELD_MAPPING["rooms"]].strip()
if track != "":
self.tracks.add(track)
if room != "":
self.rooms.add(room)
clean_session = {k: v.strip() for k, v in session.items()}
data_by_name[name] = clean_session
data_by_nid[session["nid"]] = clean_session
return (data_by_name, data_by_nid)
class GuideBook:
URLS = {
"guide": "https://builder.guidebook.com/open-api/v1.1/guides/",
"tracks": "https://builder.guidebook.com/open-api/v1.1/schedule-tracks/",
"rooms": "https://builder.guidebook.com/open-api/v1.1/locations/",
"sessions": "https://builder.guidebook.com/open-api/v1.1/sessions/",
"x-rooms": "https://builder.guidebook.com/api/locations/",
"x-maps": "https://builder.guidebook.com/api/maps/",
"x-map-regions": "https://builder.guidebook.com/api/map-regions/",
"publish": "https://builder.guidebook.com/api/guides/{guide}/publish/",
}
COLOR_MAP = {
"Applied Science": "#dddddd",
"AstriCon": "#8b4789",
"BoFs": "#ffbc00",
"Career Day": "#dddddd", # Open Source Career Day
"Cloud Native": "#638dce", # Cloud Native Days
"Developer": "#d65c09",
"DevOpsDay LA": "#565448",
"Embedded Linux": "#004a4a",
"Entertainment": "#ff6f91",
"Fedora Hatch Day": "#294172",
"FOSS @ HOME": "#998876",
"General": "#97a67a",
"HAM Radio": "#96beef",
"Higher Education": "#fff8dc", # Open Source in Higher Education
"Kernel & Low Level Systems": "#ffa200",
"Keynote": "#d31111",
"Kwaai Summit": "#4b2e83",
"MySQL": "#0aaca0",
"Next Generation": "#96f74b", # The Next Generation
"Observability": "#ffbc00",
"Open Government": "#6c6c6c",
"Open Source AI": "#ffd672",
"PlanetNix": "#2d5d3f",
"PostgreSQL": "#0aaca0",
"Security": "#000000",
"SunSecCon": "#e63946",
"Systems & Infrastructure": "#c4c249",
"Workshops": "#774022",
}
ROOM_TO_MAP_REGION = {
"Ballroom A": {"h": 0.04, "w": 0.056, "x": 0.668, "y": 0.477},
"Ballroom B": {"h": 0.04, "w": 0.056, "x": 0.668, "y": 0.519},
"Ballroom C": {"h": 0.04, "w": 0.056, "x": 0.668, "y": 0.56},
"Ballroom DE": {"h": 0.122, "w": 0.082, "x": 0.729, "y": 0.477},
"Ballroom F": {"h": 0.04, "w": 0.056, "x": 0.816, "y": 0.56},
"Ballroom G": {"h": 0.04, "w": 0.056, "x": 0.816, "y": 0.519},
"Ballroom H": {"h": 0.04, "w": 0.056, "x": 0.816, "y": 0.477},
"Check-in": {"h": 0.09, "w": 0.06, "x": 0.608, "y": 0.301},
"Exhibit Hall": {"h": 0.141, "w": 0.209, "x": 0.675, "y": 0.189},
"Room 101": {"h": 0.039, "w": 0.042, "x": 0.58, "y": 0.843},
"Room 102": {"h": 0.039, "w": 0.042, "x": 0.535, "y": 0.843},
"Room 103": {"h": 0.039, "w": 0.042, "x": 0.488, "y": 0.843},
"Room 104": {"h": 0.039, "w": 0.042, "x": 0.443, "y": 0.843},
"Room 105": {"h": 0.039, "w": 0.042, "x": 0.396, "y": 0.843},
"Room 106": {"h": 0.048, "w": 0.077, "x": 0.396, "y": 0.713},
"Room 107": {"h": 0.048, "w": 0.077, "x": 0.545, "y": 0.713},
"Room 204": {"h": 0.042, "w": 0.03, "x": 0.231, "y": 0.836},
"Room 205": {"h": 0.021, "w": 0.025, "x": 0.201, "y": 0.836},
"Room 207": {"h": 0.042, "w": 0.03, "x": 0.154, "y": 0.836},
"Room 208": {"h": 0.042, "w": 0.03, "x": 0.121, "y": 0.836},
"Room 209": {"h": 0.021, "w": 0.025, "x": 0.079, "y": 0.858},
"Room 210": {"h": 0.021, "w": 0.025, "x": 0.079, "y": 0.836},
"Room 211": {"h": 0.039, "w": 0.065, "x": 0.079, "y": 0.713},
"Room 212": {"h": 0.039, "w": 0.03, "x": 0.237, "y": 0.713},
"Room 214": {"h": 0.039, "w": 0.03, "x": 0.273, "y": 0.713},
}
REGIONED_MAP = "Pasadena-Convention-Center-Map-1000-72-fs8"
def __init__(self, logger, update, dryrun, max_deletes, key, x_key=None):
self.logger = logger
self.update = update
self.dryrun = dryrun
self.max_deletes = max_deletes
self.headers = {"Authorization": "JWT " + key}
self.guide = self.get_guide()
self.tracks = self.get_things("tracks")
self.rooms = self.get_things("rooms")
self.sessions_by_nid = self.get_things("sessions", "import_id")
self.sessions_by_name = {
s["name"]: s for s in self.sessions_by_nid.values()
}
for nid, session in self.sessions_by_nid.items():
assert nid == session["import_id"]
self.x_rooms = []
self.nids_to_delete = []
if x_key:
self.x_headers = {"Authorization": "JWT " + x_key}
self.x_rooms = self.get_things("x-rooms")
self.x_map_id = self.get_x_map_id()
self.x_map_regions = self.get_things("x-map-regions")
def get_guide(self):
"""
We always have a single guide, and need it's IDs for most calls,
so we request all guides, check there's only one, and then return
it's ID.
"""
response = requests.get(self.URLS["guide"], headers=self.headers).json()
guide_id = None
for guide in response["results"]:
if guide["name"].lower() == GUIDE_NAME.lower():
guide_id = guide["id"]
break
if not guide_id:
self.logger.critical("ERROR: Could not determine guide ID")
sys.exit(1)
return guide_id
def get_x_map_id(self):
response = requests.get(
self.URLS["x-maps"] + "?guide=%d" % self.guide,
headers=self.x_headers,
).json()
for r in response["results"]:
if r["name"]["en-US"] == self.REGIONED_MAP:
self.logger.debug(
"Found map '%s' with id=%d for guide=%d",
self.REGIONED_MAP,
r["id"],
self.guide,
)
return r["id"]
self.logger.critical(
"ERROR: Did not find expected map '%s' for guide %d",
self.REGIONED_MAP,
self.guide,
)
sys.exit(1)
def get_things(self, thing, key="name"):
"""
Get the current set of <thing> from Guidebook, where <thing> is rooms,
tracks, sessions.
"""
msg = "Loading %s from Guidebook" % thing
ourthings = {}
url = self.URLS[thing] + "?guide=%d" % self.guide
page = 1
while url is not None:
self.logger.info("%s (page %d)" % (msg, page))
response = requests.get(
url,
headers=(
self.headers
if not thing.startswith("x-")
else self.x_headers
),
).json()
self.logger.debug("Response: %s" % response)
for ourthing in response["results"]:
# Fallback to id for things without names (e.g. map-regions)
name = ourthing.get(key) or ourthing.get("id")
if isinstance(name, dict):
# Things retrived from the internal API
# (i.e. x-* things) have names that are dicts like:
# 'name': { 'en-US': 'Thing name' }
# Assume first value is what we want
name = list(name.values())[0]
ourthings[name] = ourthing
url = response["next"]
page += 1
self.logger.debug("Loaded %s: %s things", thing, len(ourthings))
return ourthings
def add_thing(self, thing, name, data, update, tid):
"""
Implementation of adding objects to Guidebook. Wrapped by the
functions that know how to build the data and use it.
"""
verb = "Updating" if update else "Adding"
if self.dryrun:
self.logger.info(
"[DRYRUN] Would have: %s %s '%s' to Guidebook"
% (verb, thing, name)
)
return
self.logger.info("%s %s '%s' to Guidebook" % (verb, thing, name))
self.logger.debug("Data: %s" % data)
headers = self.headers if not thing.startswith("x-") else self.x_headers
if update:
response = requests.patch(
self.URLS[thing] + "%d/" % tid, data=data, headers=headers
).json()
else:
response = requests.post(
self.URLS[thing], data=data, headers=headers
).json()
self.logger.debug("Response: %s" % response)
if "id" not in response:
self.logger.error("Failed to import.")
self.logger.error("DATA: %s" % data)
self.logger.error("RESPONSE: %s" % response)
sys.exit(1)
return response
def add_track(self, track, update, tid):
"""
Track-specific wrapper around add_thing()
"""
if update and not self.update:
return
data = {
"guide": self.guide,
"name": track,
# NOTE WELL: Guidebook cannot handle lower-case letters
"color": self.COLOR_MAP[track].upper(),
}
self.tracks[track] = self.add_thing("tracks", track, data, update, tid)
def setup_tracks(self, tracks):
"""
Add all tracks passed in if missing.
"""
for track in tracks:
update = False
tid = None
if track in self.tracks:
orig = self.tracks[track]
# the only "info" about a track is the color (the name is
# our primary key), so if the color is correct, it's up to date.
if orig["color"].upper() == self.COLOR_MAP[track].upper():
self.logger.debug(
"Track '%s' exists in Guidebook and has correct color"
" %s. No update needed.",
track,
self.COLOR_MAP[track].upper(),
)
continue
update = True
tid = self.tracks[track]["id"]
self.add_track(track, update, tid)
def add_room(self, room, update, rid):
"""
Room-specific wrapper around add_thing()
"""
if update and not self.update:
return
data = {
"guide": self.guide,
"name": room,
"location_type": 2, # not google maps
}
self.rooms[room] = self.add_thing("rooms", room, data, update, rid)
def setup_rooms(self, rooms):
"""
Add all rooms passed in if missing.
"""
for room in rooms:
update = False
rid = None
if room in self.x_rooms:
continue
if room in self.rooms:
orig = self.rooms[room]
# Rooms can't really change, but just in case, we'll
# check it's location type is correct.
if orig["location_type"] == 2:
self.logger.debug(
"Room '%s' exists in Guidebook and has correct"
" location_type. No update needed.",
room,
)
continue
update = True
rid = self.rooms[room]["id"]
self.add_room(room, update, rid)
def add_x_map_region(self, map_region, update, rid, location_id):
if update and not self.update:
return
name = ("map-regions-%s" % rid,)
data = {
"map_object": self.x_map_id,
"location": location_id,
"relative_x": map_region["x"],
"relative_y": map_region["y"],
"relative_width": map_region["w"],
"relative_height": map_region["h"],
}
self.add_thing("x-map-regions", name, data, update, rid)
def get_x_map_region_for_room(self, location_id):
return next(
(
reg
for reg in self.x_map_regions.values()
if (
reg["location"] is not None
and "name" in reg["location"]
and reg["location"]["id"] == location_id
)
),
None,
)
def setup_x_map_regions(self):
for room, map_region in self.ROOM_TO_MAP_REGION.items():
if room not in self.x_rooms:
self.logger.warning(
'Room "%s" does not exist in Guidebook. '
"Skipping map region %s",
room,
map_region,
)
continue
update = False
rid = None
location_id = self.x_rooms[room]["id"]
guidebook_map_region = self.get_x_map_region_for_room(location_id)
if guidebook_map_region:
update = True
rid = guidebook_map_region["id"]
if self.update:
# Update room's Guidebook location to work the map region.
# NOTE: Changing the type to gb-interactive hides the location
# from the official API so it's might break other things.
self.add_thing(
"x-rooms",
room,
data={"location_type": "gb-interactive"},
update=True,
tid=self.x_rooms[room]["id"],
)
self.add_x_map_region(map_region, update, rid, location_id)
def to_utc(self, ts, fmt):
loc_dt = datetime.strptime(ts, fmt)
if not fmt.endswith("%z"):
pt_dt = pytz.timezone("America/Los_Angeles").localize(loc_dt)
else:
pt_dt = loc_dt
return pt_dt.astimezone(pytz.utc).isoformat(timespec="seconds")
def get_times(self, session):
"""
Helper function to build times for guidebook.
"""
fmt = "%Y-%m-%dT%H:%M:%S%z"
start_ts = session["StartTime"]
end_ts = session["EndTime"]
return (self.to_utc(start_ts, fmt), self.to_utc(end_ts, fmt))
def get_id(self, thing, session):
"""
Get the ID for <thing> where thing is a room or track
"""
key = OurJSON.FIELD_MAPPING[thing]
if session[key] == "":
return []
self.logger.debug(
"Thing: %s, Key: %s, Val: %s" % (thing, key, session[key])
)
# This is `ourlist = self.rooms` or `ourlist = self.tracks`
ourlist = getattr(self, thing)
self.logger.debug("List of %s's is %s" % (thing, ourlist.keys()))
ourid = None
if session[key] in ourlist and ourlist[session[key]] is not None:
ourid = ourlist[session[key]]["id"]
else:
ourlist = getattr(self, "x_%s" % thing)
ourid = ourlist[session[key]]["id"]
return [ourid]
def add_session(self, session, original_session=None):
"""
Sesssion-specific wrapper around add_thing()
"""
if original_session is not None and not self.update:
return
name = session["Name"]
start, end = self.get_times(session)
data = {
"name": name,
"start_time": start,
"end_time": end,
"guide": self.guide,
"description_html": session["LongAbstract"],
"schedule_tracks": self.get_id("tracks", session),
"locations": self.get_id("rooms", session),
"add_to_schedule": True,
"import_id": session["nid"],
}
update = False
sid = None
if original_session is not None:
if not self.session_needs_update(data, original_session):
self.logger.debug("Session '%s' does not need update" % name)
return
update = True
sid = original_session["id"]
self.logger.debug("Data: %s" % data)
s = self.add_thing("sessions", name, data, update, sid)
self.sessions_by_nid[session["nid"]] = s
self.sessions_by_name[name] = s
def normalize_html(self, html):
"""
The HTML supported by Drupal vs Guidebook is different and
GB normalizes it upon import, so we can get in a state where
we always detect a difference.
Stripping HTML is lossy, so instead we convert to MD and compare
that which gives us a lot of information about formatting without
being sensitive to exact HTML.
"""
markdown = md(html)
# Normalize whitespace and quotes
markdown = markdown.replace("\u2018", "'").replace("\u2019", "'")
markdown = markdown.replace("\u201c", '"').replace("\u201d", '"')
# collapse whitespace
markdown = " ".join(markdown.split())
return markdown
def normalize_time(self, time_str):
n = time_str.replace("+0000", "+00:00")
n = parser.isoparse(n)
n = n.astimezone(pytz.utc)
return n
def session_needs_update(self, new_data, original_session):
"""
Compare the new session data to the original session data, and return
True if we need to update. This is needed because some fields (e.g.
description) are not updated if they haven't changed, and we want to
avoid unnecessary updates.
"""
all_keys = [
"name",
"start_time",
"end_time",
"description_html",
"schedule_tracks",
"locations",
]
for key in all_keys:
if "time" in key:
a = self.normalize_time(new_data[key])
b = self.normalize_time(original_session[key])
elif "html" in key:
a = self.normalize_html(new_data[key])
b = self.normalize_html(original_session[key])
else:
a = new_data[key]
b = original_session[key]
if a != b:
self.logger.info(
"Session '%s' needs update because '%s' changed: '%s' !="
" '%s'",
new_data["name"],
key,
new_data[key],
original_session[key],
)
return True
return False
def backfill_session_nids(self, sessions_by_name, sessions_by_nid):
"""
We didn't always have a unique identifier, and this will backfill
missing ones. Probably can be nuked by 24X.
"""
for name, info in self.sessions_by_name.items():
if info["import_id"] is None:
if name in sessions_by_name.keys():
self.logger.info(
"Adding NID %s to session '%s'",
info["import_id"],
name,
)
session = sessions_by_name[name]
update = True
sid = info["id"]
self.add_session(session, update, sid, True)
else:
self.logger.warning(
"Session '%s' exists in Guidebook, but has no NID,"
" and we cannot find the name in our data. Deleting it."
% name,
)
self.delete_session(info)
else:
nid = info["import_id"]
if nid not in sessions_by_nid.keys():
self.logger.warning(
"Session '%s' with NID %s exists in Guidebook, but we"
" cannot find it in our data. Adding it to the delete"
" list." % (name, nid)
)
self.nids_to_delete.append(nid)
def setup_sessions(self, sessions_by_name, sessions_by_nid):
"""
Add all rooms passed in if missing.
"""
# First, make sure we have NIDs for all sessions in Guidebook
self.backfill_session_nids(sessions_by_name, sessions_by_nid)
# now loop through pass in sessions, and add/update as needed
for nid, session in sessions_by_nid.items():
original_session = None
name = session["Name"]
if session["StartTime"] == "":
self.logger.warning("Skipping %s - no date" % name)
continue
if nid in self.sessions_by_nid:
original_session = self.sessions_by_nid[nid]
self.add_session(session, original_session)
# Clean up sessions that should be deleted
num_deletes = len(self.nids_to_delete)
if num_deletes == 0:
return
if num_deletes > self.max_deletes:
self.logger.warning(
"Number of sessions to delete (%d) exceeds the max threshold"
" (%d). Not deleting any sessions.",
num_deletes,
self.max_deletes,
)
else:
self.logger.warning(
"Deleting %d sessions that are no longer in our data",
num_deletes,
)
for nid in self.nids_to_delete:
session = self.sessions_by_nid[nid]
self.delete_session(session)
def delete_session(self, session):
"""
Delete a session. Unlike "add" functions, this takes the object
from the Guidebook API, not our data.
"""
if self.dryrun:
self.logger.info(
"[DRYRUN] Would have deleted session '%s' from Guidebook"
% session["name"]
)
return
self.logger.debug(
"Deleting session %d [%s]" % (session["id"], session["name"])
)
response = requests.delete(
self.URLS["sessions"] + "%d/" % session["id"],
headers=self.headers,
)
self.logger.debug("Got %d" % response.status_code)
if not (response.status_code >= 200 and response.status_code < 300):
self.logger.error("Failed to delete")
self.logger.error("RESPONSE: %s" % response.json())
sys.exit(1)
def delete_sessions(self):
self.logger.warning("Deleting all sessions")
for session in self.sessions_by_nid.values():
self.delete_session(session)
def delete_track(self, track):
"""
Delete a track. Unlike "add" functions, this takes the object
from the Guidebook API, not our data.
"""
if self.dryrun:
self.logger.info(
"[DRYRUN] Would have deleted track '%s' from Guidebook"
% track["name"]
)
return
self.logger.debug(
"Deleting track %d [%s]" % (track["id"], track["name"])
)
response = requests.delete(
self.URLS["tracks"] + "%d/" % track["id"],
headers=self.headers,
)
if response.status_code != 204:
self.logger.error("Failed to delete")
self.logger.error("RESPONSE: %s" % response.json())
sys.exit(1)
def delete_tracks(self):
self.logger.warning("Deleting all tracks")
for track in self.tracks.values():
self.delete_track(track)
def delete_room(self, room):
"""
Delete a room. Unlike "add" functions, this takes the object
from the Guidebook API, not our data.
"""
if self.dryrun:
self.logger.info(
"[DRYRUN] Would have deleted room '%s' from Guidebook"
% room["name"]
)
return
self.logger.debug("Deleting room %d [%s]" % (room["id"], room["name"]))
response = requests.delete(
self.URLS["rooms"] + "%d/" % room["id"],
headers=self.headers,
)
if response.status_code != 204:
self.logger.error("Failed to delete")
self.logger.error("RESPONSE: %s" % response.json())
sys.exit(1)
def delete_rooms(self):
self.logger.warning("Deleting all rooms")
for room in self.rooms.values():
self.delete_room(room)
def delete_all(self):
self.delete_sessions()
self.delete_tracks()
self.delete_rooms()
def publish_updates(self):
"""
Publish pending updates. This is an internal/unpublished API, and
this may not be identical to the "publish" button in the Guidebook
builder. However, it does publish all pending session data at a minimum.
"""
if self.dryrun:
self.logger.info("[DRYRUN] Would have published pending updates.")
return
self.logger.info("Publishing changes")
response = requests.post(
self.URLS["publish"].format(guide=self.guide),
headers=self.x_headers,
)
if response.status_code == 202:
self.logger.debug("Publish accepted")
return
if response.status_code == 403:
resp_text = response.text.lower()
if "no new content" in resp_text:
self.logger.debug("No changes to publish")
return
elif "currently publishing" in resp_text:
self.logger.debug("Guidebook is already publishing")
return
self.logger.error("Failed to publish")
self.logger.error("Status: %s" % response.status_code)
self.logger.error("Body: %s" % response.text)
sys.exit(1)
def _get_token(fname, ename, logger):
env_token = os.getenv(ename)
if env_token is not None:
return env_token.strip()
for dir in xdg.xdg_config_dirs():
api_file = os.path.join(dir, fname)
if os.path.isfile(api_file):
logger.debug("Using %s from %s" % (ename, api_file))
return open(api_file, "r").read().strip()
def get_tokens(logger):
key = _get_token("guidebook_api_token", "GUIDEBOOK_API_TOKEN", logger)
if not key:
logger.critical("No API file specified. See help for details.")
sys.exit(1)
x_key = _get_token("guidebook_jwt_token", "GUIDEBOOK_JWT_TOKEN", logger)
return (key, x_key)
@click.command(context_settings={"help_option_names": ["-h", "--help"]})
@click.option(
"--debug/--no-debug", "-d", default=False, help="Print debug messages."
)
@click.option(
"--update/--no-update",
"-u",
default=True,
help="Update existing sessions. Does so idempotently, defaults to True.",
)
@click.option(
"--delete-all/--no-delete-all",
default=False,
help="Delete all tracks, rooms, and sessions",
)
@click.option(
"--json",
"feed",
metavar="FILE_OR_URL",
default=DBASE_DEFAULT,
help="JSON file or http(s) URL to JSON data.",
)
@click.option(
"--dryrun/--no-dryrun",
"-n",
default=False,
help="Don't actually make any changes to Guidebook.",
)
@click.option(
"--max-deletes",
default=0,
help="Max number of sessions to delete when syncing. Zero will not"
" delete any sessions. Ignored if --delete-all is used.",
)
def main(debug, update, delete_all, feed, dryrun, max_deletes):
"""
Sync the schedule data from our website to Guidebook.
AUTHENTICATION
The Guidebook API token must be provided either via the GUIDEBOOK_API_TOKEN
environment variable, or via a file named 'guidebook_api_token' located in
one of the standard XDG config directories (e.g. ~/.config/).
Optionally, a Guidebook JWT token may be provided via the
GUIDEBOOK_JWT_TOKEN environment variable or a file named
'guidebook_jwt_token' located in one of the standard XDG config
directories. This token is needed for certain operations such as setting up
X Map regions and publishing.
"""
level = logging.INFO
if debug:
level = logging.DEBUG
logger = logging.getLogger("genbook")
logger.setLevel(level)
ch = logging.StreamHandler()
formatter = logging.Formatter("%(levelname)s: %(message)s")
ch.setFormatter(formatter)
logger.addHandler(ch)
key, x_key = get_tokens(logger)
if delete_all:
print("WARNING!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!") # noqa: E999
print("This will cause any attendee who has saved any sessions")
print("into a schedule to lose all of that work.")
click.confirm("ARE YOU FUCKING SURE?!", abort=True)
else:
ourdata = OurJSON(feed, logger)
ourguide = GuideBook(logger, update, dryrun, max_deletes, key, x_key=x_key)
if delete_all:
ourguide.delete_all()
else:
ourguide.setup_tracks(ourdata.tracks)
ourguide.setup_rooms(ourdata.rooms)
ourguide.setup_sessions(
ourdata.sessions_by_name, ourdata.sessions_by_nid
)
if x_key:
ourguide.setup_x_map_regions()
# unclear exactly when this is needed.
ourguide.publish_updates()
if __name__ == "__main__":
main()