brayvasq/python_pdf.py

## python_pdf.py
# main.py
# import section ....
from xhtml2pdf import pisa             # import python module
import sys

# Constants section ....
# Content to write in our PDF file.
SOURCE = "<html><body><p>PDF from string</p></body></html>"

# Filename for our PDF file.
OUTPUT_FILENAME = "test.pdf"

# Template file name
TEMPLATE_FILE = "template.html"

# Methods section ....
def html_to_pdf(content, output):
    """
    Generate a pdf using a string content

    Parameters
    ----------
    content : str
	    content to write in the pdf file
	output  : str
	    name of the file to create
    """
    # Open file to write
    result_file = open(output, "w+b") # w+b to write in binary mode.

    # convert HTML to PDF
    pisa_status = pisa.CreatePDF(
            content,                   # the HTML to convert
            dest=result_file	       # file handle to recieve result
    )

    # close output file
    result_file.close()

    result = pisa_status.err

    if not result:
        print("Successfully created PDF")
    else:
        print("Error: unable to create the PDF")

    # return False on success and True on errors
    return result

def from_text(source, output):
    """
    Generate a pdf from a plain string

    Parameters
    ----------
    source : str
	    content to write in the pdf file
	output  : str
	    name of the file to create
    """
    html_to_pdf(source, output)

def from_template(template, output):
    """
    Generate a pdf from a html file

    Parameters
    ----------
    source : str
	    content to write in the pdf file
	output  : str
	    name of the file to create
    """
    # Reading our template
    source_html = open(template, "r")
    content = source_html.read() # the HTML to convert
    source_html.close() # close template file

    html_to_pdf(content, output)

# Main section ...
if __name__ == "__main__":
    if len(sys.argv)> 1 :
        if sys.argv[1] == '--help':
            print('Info: ')
            print('--help List the options to send an email')
            print('--text Create a PDF file from a string')
            print('--template Create a PDF file from a template')
        elif sys.argv[1] == '--text':
            print("Creating a PDF file from a string")
            from_text(SOURCE, OUTPUT_FILENAME)
        elif sys.argv[1] == '--template':
            print("Creating a PDF file from a template")
            from_template(TEMPLATE_FILE, OUTPUT_FILENAME)
    else:
        print("Please give the type of message to send.")
        print("For help execute `python main.py --help`")
	# main.py
	# import section ....
	from xhtml2pdf import pisa # import python module
	import sys

	# Constants section ....
	# Content to write in our PDF file.
	SOURCE = "<html><body><p>PDF from string</p></body></html>"

	# Filename for our PDF file.
	OUTPUT_FILENAME = "test.pdf"

	# Template file name
	TEMPLATE_FILE = "template.html"

	# Methods section ....
	def html_to_pdf(content, output):
	"""
	Generate a pdf using a string content

	Parameters
	----------
	content : str
	content to write in the pdf file
	output : str
	name of the file to create
	"""
	# Open file to write
	result_file = open(output, "w+b") # w+b to write in binary mode.

	# convert HTML to PDF
	pisa_status = pisa.CreatePDF(
	content, # the HTML to convert
	dest=result_file # file handle to recieve result
	)

	# close output file
	result_file.close()

	result = pisa_status.err

	if not result:
	print("Successfully created PDF")
	else:
	print("Error: unable to create the PDF")

	# return False on success and True on errors
	return result

	def from_text(source, output):
	"""
	Generate a pdf from a plain string

	Parameters
	----------
	source : str
	content to write in the pdf file
	output : str
	name of the file to create
	"""
	html_to_pdf(source, output)

	def from_template(template, output):
	"""
	Generate a pdf from a html file

	Parameters
	----------
	source : str
	content to write in the pdf file
	output : str
	name of the file to create
	"""
	# Reading our template
	source_html = open(template, "r")
	content = source_html.read() # the HTML to convert
	source_html.close() # close template file

	html_to_pdf(content, output)

	# Main section ...
	if __name__ == "__main__":
	if len(sys.argv)> 1 :
	if sys.argv[1] == '--help':
	print('Info: ')
	print('--help List the options to send an email')
	print('--text Create a PDF file from a string')
	print('--template Create a PDF file from a template')
	elif sys.argv[1] == '--text':
	print("Creating a PDF file from a string")
	from_text(SOURCE, OUTPUT_FILENAME)
	elif sys.argv[1] == '--template':
	print("Creating a PDF file from a template")
	from_template(TEMPLATE_FILE, OUTPUT_FILENAME)
	else:
	print("Please give the type of message to send.")
	print("For help execute `python main.py --help`")