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
|
#!/usr/bin/env python3
import os
import sys
import json
CATEGORY_RUN = "run"
CATEGORY_WALK = "walk"
CATEGORY_TRAIL = "trail"
CATEGORY_KIDS = "kids"
CATEGORY_SAINT_SILVESTER = "saint-silvester"
CATEGORY_10K = "10k"
CATEGORY_15K = "15k"
CATEGORY_HALF_MARATHON = "half-marathon"
CATEGORY_MARATHON = "marathon"
CLASS_TO_CATEGORIES = {
"event_type-corrida": [CATEGORY_RUN],
"event_type-caminhada": [CATEGORY_WALK],
"event_type-corrida-10km": [CATEGORY_RUN, CATEGORY_10K],
"event_type-corrida-10-km": [CATEGORY_RUN, CATEGORY_10K],
"event_type-corrida-de-15-km": [CATEGORY_RUN, CATEGORY_15K],
"event_type-trail": [CATEGORY_TRAIL],
"event_type-kids-trail": [CATEGORY_TRAIL, CATEGORY_KIDS],
"event_type-trail-curto": [CATEGORY_TRAIL],
"event_type-trail-longo": [CATEGORY_TRAIL],
"event_type-trail-endurance": [CATEGORY_TRAIL],
"event_type-trail-ultra": [CATEGORY_TRAIL],
"event_type-sao-silvestre": [CATEGORY_SAINT_SILVESTER],
"event_type-outras": [],
"event_type-obstaculos": [CATEGORY_RUN],
"event_type-corta-mato": [CATEGORY_RUN],
"event_type-backyard-2": [CATEGORY_RUN],
"event_type-meiamaratona": [CATEGORY_RUN, CATEGORY_HALF_MARATHON],
"event_type-maratona": [CATEGORY_RUN, CATEGORY_MARATHON],
"event_type-skyrunning": [CATEGORY_RUN],
"event_type-corridas-inferior-10": [CATEGORY_RUN],
"event_type-kids": [CATEGORY_KIDS],
# ignored
"ajde_events": [],
"type-ajde_events": [],
"status-publish": [],
"has-post-thumbnail": [],
"hentry": [],
}
CLASS_IGNORE_PREFIXES = [
"post-",
"event_location-",
"event_organizer-",
"event_type_2-",
"event_type_3-",
"event_type_4-",
"event_type_5-",
]
for slug in sys.argv[1:]:
data_path = os.path.join("events", slug, "data.json")
with open(data_path, "r") as f:
data = json.load(f)
classes = data["class_list"]
categories = set()
categories_path = os.path.join("events", slug, "categories")
for class_ in classes:
if any(class_.startswith(p) for p in CLASS_IGNORE_PREFIXES):
continue
if class_ not in CLASS_TO_CATEGORIES:
raise Exception(f"unknown class: {class_}")
for category in CLASS_TO_CATEGORIES[class_]:
categories.add(category)
with open(categories_path, "w") as f:
for i, category in enumerate(sorted(categories)):
if i > 0:
f.write("\n")
f.write(category)
|
