contest-scripts/generate_panels.py at master · PolyProg/contest-scripts · GitHub

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
#!/usr/bin/python3
import json
import html
import os
import pdfkit


# Get the config
config_path = input('Config file path: ')
with open(config_path, encoding='utf-8') as config_file:
  config = json.load(config_file)


# Validate data
if 'teams' not in config:
  sys.exit('Config must contain teams')
teams = config['teams']
for team in teams:
  if 'name' not in team or 'members' not in team or 'location' not in team :
    sys.exit('Teams must have name, members, location')


# Sort teams by location, to ease the work of people putting up the panels
teams.sort(key=lambda t: t['location'])


# Convert teams to HTML divs
body = ""
for idx, team in enumerate(teams):
  for n in range(2):
    div = ""

    # don't put a page break on the last page, to avoid a blank sheet at the end
    if n == 0 or idx < len(config['teams']) - 1:
      div += '<div style="page-break-after: always;">'
    else:
      div += '<div>'

    div += '<h1>' + html.escape(team['name']) + '</h1>'
    div += '<h2>' + html.escape(', '.join(team['members'])) + '</h2>'
    div += '<h3>' + team['location'] + '</h3>'
    div += '</div>'

    body += div


# Create the HTML
html = """<!DOCTYPE html>
<html>
<head>
  <meta charset="UTF-8" />
  <style>
  h1 {
    font-size: 12em;
    font-weight: 700;
  }

  h2 {
    font-size: 4em;
    font-weight: 500;
  }

  h3 {
    font-size: 1em;
    font-weight: normal;
  }
  </style>
</head>
<body>""" + body + """</body>
</html>
"""


# Set options (notably, A4 paper, and margins equivalent to MS Word's "narrow")
options = {
    'orientation': 'landscape',
    'page-size': 'A4',
    'margin-top': '0.3in',
    'margin-right': '0.3in',
    'margin-bottom': '0.3in',
    'margin-left': '0.3in',
    'encoding': 'UTF-8'
}

# Output the PDF
print('Outputting PDF, this may take a while...')
os.makedirs('out', exist_ok=True)
pdfkit.from_string(html, 'out/panels.pdf', options=options)


print('Done!')