pattern_exporter.py

#--------------------------------------------------------------
#   pattern_exporter.py
#
#   Created for LearnMYOG.com
#   Automates creation of PDF pattern sheets from Inkscape PNG export
#   From large format PNG create A0, tabloid/A3 and letter/A4 merged patterns
#   
#   virtualenv: source .venv/bin/activate
#   use python3 pattern_exporter.py input.png paperSize
#
#   For best results:
#   'import.png' is PNG size in multiple of 7.5 inches wide and 10 inches tall @ 300dpi.
#   
#   pattern_tile.py performs image operations:
#       - tiles for letter (a4), tabloid (a3) and 36 x 48 inch (a0)
#       - adds cut and alignment guides
#       - adds page numbers and copyright
#       - removes alpha layer and adds border
#       - converts to pdfs
#
#   pdf_merge.py finds all pdfs in dir_path and merges into one pdf
#
#   Next:
#   Add argparse for automated help
#
#--------------------------------------------------------------

import sys, os, shutil, time
from pattern_tile import pattern_tile
from pdf_merge import merge_pdfs

# Initialize process timer
start_time = time.time()


def pattern_exporter(input_png, paperSize = 'all'):

    # Set variables
    if len(sys.argv) <= 2:
        input_png = sys.argv[1]
        formats = ['letter','tabloid']

    elif len(sys.argv) > 2:
        input_png = sys.argv[1]
        paperSize = sys.argv[2]
        if paperSize == 'all':
            formats = ['letter','tabloid','a0']
        else: 
            formats = [paperSize]
    
    dir_path = 'pdfs/'
    
    
    # 1 check if pattern.PNG exists, create or cleanup dir_paths folder

    if os.path.exists(input_png):
        prepare_temp_dir(dir_path)
        
    else:
        print(input_png,"does not exist in ",os.path.abspath('.'))
        exit()


    # 2 convert individual PNGs to PDFs
    # formats = ['letter','tabloid','a0']
    for f in formats:
        print("Working on",f,"pattern sheets.")
        pattern_tile(input_png, f, dir_path)

   
        # 4 merge the individual PDFs with pdf_merge.py
        merged_temp = f + 'merged.pdf'
        merge_pdfs(
            dir_path, 
            merged_temp)

        # 5 final output file named and copied to base directory
        timestr = time.strftime("%Y%m%d")
        fname = os.path.basename(input_png).split('.')[0]
        output_pdf = fname + '_' + f + '_' + timestr + '.pdf'
        shutil.copy(dir_path + merged_temp, output_pdf)

        # output messages
        print()
        print(">>>>>>>>> Created: " + output_pdf +" <<<<<<<<<")

        print("--- %s seconds elapsed---" % (time.time() - start_time))
        print()

        prepare_temp_dir(dir_path)  # clean up after yourself


#### HELPER FUNCTIONS ####

def prepare_temp_dir(dir_path):
    if os.path.exists(dir_path):
        try:
            for i in os.listdir(dir_path):
                os.remove(dir_path+i)
        except OSError as e:
            print("Error: %s : %s" % (dir_path, e.strerror))
    else: 
        os.mkdir(dir_path)



if __name__ == '__main__':
    pattern_exporter(
        input_png = 'pattern.png',
        paperSize = 'all')