-
Notifications
You must be signed in to change notification settings - Fork 447
/
merge_badges.py
69 lines (56 loc) · 2.26 KB
/
merge_badges.py
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
#!usr/bin/python3
import os
import argparse
from generate_badges import GenerateBadges
from cairosvg import svg2pdf
from PyPDF2 import PdfFileMerger
parser = argparse.ArgumentParser(description='Argument Parser for merge_badges')
parser.add_argument('-p', dest='pdf', action='store_true')
parser.set_defaults(pdf=True)
arguments = parser.parse_args()
_pdf = arguments.pdf
APP_ROOT = os.path.dirname(os.path.abspath(__file__))
BADGES_FOLDER = os.path.join(APP_ROOT, 'static/badges')
badge_generator = GenerateBadges()
badge_generator.run_generator()
input_folders = [file for file in os.listdir(BADGES_FOLDER) if file.lower().endswith(".badges")]
def generate_pdfs(folder_path):
"""
Function to generate the PDF for the badge
:param `folder_path` - Path of the folder for saving of the PDF's
"""
svgs = [file for file in os.listdir(folder_path) if file.lower().endswith('.svg')]
for svg in svgs:
svg_path = os.path.join(folder_path, svg)
pdf_path = os.path.splitext(svg_path)[0] + '.pdf'
print('svg: {}'.format(svg_path))
print('pdf: {}'.format(pdf_path))
try:
svg2pdf(url=svg_path, write_to=pdf_path)
except Exception as e:
pass
# Generating PDF files from svg.
if _pdf:
for folder in input_folders:
folder_path = os.path.join(BADGES_FOLDER, folder)
generate_pdfs(folder_path)
# Merge badges of different types
input_folders = [file for file in os.listdir(BADGES_FOLDER) if file.lower().endswith(".badges")]
print('Merging badges of different types.')
for folder in input_folders:
folder_path = os.path.join(BADGES_FOLDER, folder)
merger = PdfFileMerger()
pdfs = [file for file in os.listdir(folder_path) if file.lower().endswith('.pdf')]
for pdf in pdfs:
merger.append(open(os.path.join(folder_path, pdf), 'rb'))
out = folder + '.pdf'
out_path = os.path.join(BADGES_FOLDER, out)
with open(out_path, 'wb') as fout:
merger.write(fout)
final_path = os.path.join(BADGES_FOLDER, 'all-badges.pdf')
pdfs = [file for file in os.listdir(BADGES_FOLDER) if file.lower().endswith('.pdf')]
merger = PdfFileMerger()
for pdf in pdfs:
merger.append(open(os.path.join(BADGES_FOLDER, pdf), 'rb'))
with open(final_path, 'wb') as fout:
merger.write(fout)