248 lines
9.4 KiB
Python
248 lines
9.4 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
openslides.motion.pdf
|
||
~~~~~~~~~~~~~~~~~~~~~
|
||
|
||
Functions to generate the PDFs for the motion app.
|
||
|
||
:copyright: (c) 2011–2013 by the OpenSlides team, see AUTHORS.
|
||
:license: GNU GPL, see LICENSE for more details.
|
||
"""
|
||
|
||
import random
|
||
from bs4 import BeautifulSoup
|
||
|
||
from reportlab.lib import colors
|
||
from reportlab.lib.units import cm
|
||
from reportlab.platypus import (
|
||
SimpleDocTemplate, PageBreak, Paragraph, Spacer, Table, TableStyle)
|
||
from django.utils.translation import ugettext as _
|
||
|
||
from openslides.config.api import config
|
||
from openslides.utils.pdf import stylesheet
|
||
from .models import Motion
|
||
|
||
|
||
def motions_to_pdf(pdf):
|
||
"""Create a PDF with all motions."""
|
||
|
||
motions = Motion.objects.all()
|
||
all_motion_cover(pdf, motions)
|
||
for motion in motions:
|
||
pdf.append(PageBreak())
|
||
motion_to_pdf(pdf, motion)
|
||
|
||
|
||
def motion_to_pdf(pdf, motion):
|
||
"""Create a PDF for one motion."""
|
||
|
||
pdf.append(Paragraph(_("Motion: %s") % motion.title, stylesheet['Heading1']))
|
||
|
||
motion_data = []
|
||
|
||
# submitter
|
||
cell1a = []
|
||
cell1a.append(Spacer(0, 0.2 * cm))
|
||
cell1a.append(Paragraph("<font name='Ubuntu-Bold'>%s:</font>" % _("Submitter"),
|
||
stylesheet['Heading4']))
|
||
cell1b = []
|
||
cell1b.append(Spacer(0, 0.2 * cm))
|
||
cell1b.append(Paragraph(unicode(motion.submitter), stylesheet['Normal']))
|
||
motion_data.append([cell1a, cell1b])
|
||
|
||
# TODO: choose this in workflow
|
||
if motion.state.allow_submitter_edit:
|
||
# Cell for the signature
|
||
cell2a = []
|
||
cell2b = []
|
||
cell2a.append(Paragraph("<font name='Ubuntu-Bold'>%s:</font>" %
|
||
_("Signature"), stylesheet['Heading4']))
|
||
cell2b.append(Paragraph(42 * "_", stylesheet['Signaturefield']))
|
||
cell2b.append(Spacer(0, 0.1 * cm))
|
||
cell2b.append(Spacer(0, 0.2 * cm))
|
||
motion_data.append([cell2a, cell2b])
|
||
|
||
# supporters
|
||
if config['motion_min_supporters']:
|
||
cell3a = []
|
||
cell3b = []
|
||
cell3a.append(Paragraph("<font name='Ubuntu-Bold'>%s:</font><seqreset id='counter'>"
|
||
% _("Supporters"), stylesheet['Heading4']))
|
||
supporters = motion.supporter.all()
|
||
for supporter in supporters:
|
||
cell3b.append(Paragraph("<seq id='counter'/>. %s" % unicode(supporter),
|
||
stylesheet['Signaturefield']))
|
||
if motion.state.allow_support:
|
||
for count in range(config['motion_min_supporters'] - supporters.count()):
|
||
cell3b.append(Paragraph("<seq id='counter'/>. " + 42 * "_",
|
||
stylesheet['Signaturefield']))
|
||
cell3b.append(Spacer(0, 0.2 * cm))
|
||
motion_data.append([cell3a, cell3b])
|
||
|
||
## # status
|
||
## cell4a = []
|
||
## cell4b = []
|
||
## note = " ".join(motion.notes)
|
||
## cell4a.append(Paragraph("<font name='Ubuntu-Bold'>%s:</font>" % _("Status"), stylesheet['Heading4']))
|
||
## if note != "":
|
||
## if motion.status == "pub":
|
||
## cell4b.append(Paragraph(note, stylesheet['Normal']))
|
||
## else:
|
||
## cell4b.append(Paragraph("%s | %s" % (motion.get_status_display(), note), stylesheet['Normal']))
|
||
## else:
|
||
## cell4b.append(Paragraph("%s" % motion.get_status_display(), stylesheet['Normal']))
|
||
## data.append([cell4a, cell4b])
|
||
|
||
# Motion state
|
||
cell4a = []
|
||
cell4b = []
|
||
cell4a.append(Paragraph("<font name='Ubuntu-Bold'>%s:</font>" % _("State"),
|
||
stylesheet['Heading4']))
|
||
cell4b.append(Paragraph(motion.state.name, stylesheet['Normal']))
|
||
motion_data.append([cell4a, cell4b])
|
||
|
||
# Version number (aid)
|
||
if motion.versions.count() > 1:
|
||
cell5a = []
|
||
cell5b = []
|
||
cell5a.append(Paragraph("<font name='Ubuntu-Bold'>%s:</font>" % _("Version"),
|
||
stylesheet['Heading4']))
|
||
cell5b.append(Paragraph("%s" % motion.version.version_number, stylesheet['Normal']))
|
||
motion_data.append([cell5a, cell5b])
|
||
|
||
# voting results
|
||
polls = []
|
||
for poll in motion.polls.all():
|
||
if not poll.has_votes():
|
||
continue
|
||
polls.append(poll)
|
||
|
||
if polls:
|
||
cell6a = []
|
||
cell6b = []
|
||
cell6a.append(Paragraph("<font name='Ubuntu-Bold'>%s:</font>" %
|
||
_("Vote results"), stylesheet['Heading4']))
|
||
ballotcounter = 0
|
||
for poll in polls:
|
||
ballotcounter += 1
|
||
option = poll.get_options()[0]
|
||
yes, no, abstain, invalid, votecast = (
|
||
option['Yes'], option['No'], option['Abstain'],
|
||
poll.print_voteinvalid(), poll.print_votecast())
|
||
|
||
if len(polls) > 1:
|
||
cell6b.append(Paragraph("%s. %s" % (ballotcounter, _("Vote")),
|
||
stylesheet['Bold']))
|
||
cell6b.append(Paragraph(
|
||
"%s: %s <br/> %s: %s <br/> %s: %s <br/> %s: %s <br/> %s: %s" %
|
||
(_("Yes"), yes, _("No"), no, _("Abstention"), abstain, _("Invalid"),
|
||
invalid, _("Votes cast"), votecast), stylesheet['Normal']))
|
||
cell6b.append(Spacer(0, 0.2 * cm))
|
||
motion_data.append([cell6a, cell6b])
|
||
|
||
# Creating Table
|
||
table = Table(motion_data)
|
||
table._argW[0] = 4.5 * cm
|
||
table._argW[1] = 11 * cm
|
||
table.setStyle(TableStyle([('BOX', (0, 0), (-1, -1), 1, colors.black),
|
||
('VALIGN', (0, 0), (-1, -1), 'TOP')]))
|
||
pdf.append(table)
|
||
pdf.append(Spacer(0, 1 * cm))
|
||
|
||
# motion title
|
||
pdf.append(Paragraph(motion.title, stylesheet['Heading3']))
|
||
|
||
# motion text
|
||
convert_html_to_reportlab(pdf, motion.text)
|
||
pdf.append(Spacer(0, 1 * cm))
|
||
|
||
# motion reason
|
||
if motion.reason:
|
||
pdf.append(Paragraph(_("Reason:"), stylesheet['Heading3']))
|
||
convert_html_to_reportlab(pdf, motion.reason)
|
||
return pdf
|
||
|
||
|
||
def convert_html_to_reportlab(pdf, text):
|
||
# parsing and replacing not supported html tags for reportlab...
|
||
soup = BeautifulSoup(text)
|
||
# read all list elements...
|
||
for element in soup.find_all('li'):
|
||
# ... and replace ul list elements with <para><bullet>•</bullet>...<para>
|
||
if element.parent.name == "ul":
|
||
if element.ul:
|
||
# for nested ul lists use simple spaces (pragmatic solution)
|
||
element.li.insert(0, ' ')
|
||
element.insert_before(element.find_all('li'))
|
||
element.clear()
|
||
else:
|
||
element.name = "para"
|
||
bullet_tag = soup.new_tag("bullet")
|
||
bullet_tag.string = "•"
|
||
element.insert(0, bullet_tag)
|
||
# ... and replace ol list elements with <para><bullet><seq id="%id"></seq>.</bullet>...</para>
|
||
if element.parent.name == "ol":
|
||
# set list id if element is the first of numbered list
|
||
if not element.find_previous_sibling():
|
||
id = random.randrange(0, 101)
|
||
if element.ol:
|
||
# nested ol list
|
||
element.li.insert(0, ' ')
|
||
element.insert_before(element.find_all('li'))
|
||
element.clear()
|
||
else:
|
||
element.name = "para"
|
||
element.insert(0, soup.new_tag("bullet"))
|
||
element.bullet.insert(0, soup.new_tag("seq"))
|
||
element.bullet.seq['id'] = id
|
||
element.bullet.insert(1, ".")
|
||
# remove tags which are not supported by reportlab (replace tags with their children tags)
|
||
for tag in soup.find_all('ul'):
|
||
tag.unwrap()
|
||
for tag in soup.find_all('ol'):
|
||
tag.unwrap()
|
||
for tag in soup.find_all('li'):
|
||
tag.unwrap()
|
||
# print paragraphs with numbers
|
||
text = soup.body.contents
|
||
paragraph_number = 1
|
||
for paragraph in text:
|
||
paragraph = str(paragraph)
|
||
# ignore empty paragraphs (created by newlines/tabs of ckeditor)
|
||
if paragraph == '\n' or paragraph == '\n\n' or paragraph == '\n\t':
|
||
continue
|
||
if "<pre>" in paragraph:
|
||
pdf.append(Paragraph(paragraph.replace('\n', '<br/>'), stylesheet['InnerMonotypeParagraph'], str(paragraph_number)))
|
||
paragraph_number += 1
|
||
elif "<para>" in paragraph:
|
||
pdf.append(Paragraph(paragraph, stylesheet['InnerListParagraph']))
|
||
elif "<seqreset" in paragraph:
|
||
pass
|
||
elif "<h1>" in paragraph:
|
||
pdf.append(Paragraph(paragraph, stylesheet['InnerH1Paragraph']))
|
||
elif "<h2>" in paragraph:
|
||
pdf.append(Paragraph(paragraph, stylesheet['InnerH2Paragraph']))
|
||
elif "<h3>" in paragraph:
|
||
pdf.append(Paragraph(paragraph, stylesheet['InnerH3Paragraph']))
|
||
else:
|
||
pdf.append(Paragraph(str(paragraph), stylesheet['InnerParagraph'], str(paragraph_number)))
|
||
paragraph_number += 1
|
||
|
||
|
||
def all_motion_cover(pdf, motions):
|
||
"""Create a coverpage for all motions."""
|
||
pdf.append(Paragraph(config["motion_pdf_title"], stylesheet['Heading1']))
|
||
|
||
preamble = config["motion_pdf_preamble"]
|
||
if preamble:
|
||
pdf.append(Paragraph("%s" % preamble.replace('\r\n', '<br/>'), stylesheet['Paragraph']))
|
||
|
||
pdf.append(Spacer(0, 0.75 * cm))
|
||
|
||
if not motions:
|
||
pdf.append(Paragraph(_("No motions available."), stylesheet['Heading3']))
|
||
else:
|
||
for motion in motions:
|
||
pdf.append(Paragraph(motion.title, stylesheet['Heading3']))
|