from os import chmod, environ, path as os_path
from subprocess import call as call_subprocess
from tempfile import NamedTemporaryFile


def generate_pdf(html='', url=''):
    # Validate input
    if not html and not url:
        raise ValueError('Must pass HTML or specify a URL')
    if html and url:
        raise ValueError('Must pass HTML or specify a URL, not both')

    wkhtmltopdf_default = (
            os_path.abspath(os_path.split(__file__)[0]) +
            '/bin/wkhtmltopdf-heroku')

    # Make sure wkhtmltopdf-heroku is executable
    chmod(wkhtmltopdf_default, 0755)

    # Reference command
    wkhtmltopdf_cmd = environ.get('WKHTMLTOPDF_CMD', wkhtmltopdf_default)

    # Set up return file
    pdf_file = NamedTemporaryFile(suffix='.pdf')

    if html:
        # Save the HTML to a temp file
        html_file = NamedTemporaryFile(suffix='.html')
        html_file.write(html)

        # wkhtmltopdf
        call_subprocess([wkhtmltopdf_cmd, '-q', html_file.name, pdf_file.name])

        # Clean up
        html_file.close()
    else:
        # wkhtmltopdf, using URL
        call_subprocess([wkhtmltopdf_cmd, '-q', url, pdf_file.name])

    return pdf_file

from os import chmod, environ, path as os_path
from subprocess import call as call_subprocess
from tempfile import NamedTemporaryFile


def generate_pdf(html='', url=''):
    # Validate input
    if not html and not url:
        raise ValueError('Must pass HTML or specify a URL')
    if html and url:
        raise ValueError('Must pass HTML or specify a URL, not both')

    wkhtmltopdf_default = 'wkhtmltopdf-heroku'

    # Make sure wkhtmltopdf-heroku is executable
    chmod(wkhtmltopdf_default, 0755)

    # Reference command
    wkhtmltopdf_cmd = environ.get('WKHTMLTOPDF_CMD', wkhtmltopdf_default)

    # Set up return file
    pdf_file = NamedTemporaryFile(suffix='.pdf')

    if html:
        # Save the HTML to a temp file
        html_file = NamedTemporaryFile(suffix='.html')
        html_file.write(html)

        # wkhtmltopdf
        call_subprocess([wkhtmltopdf_cmd, '-q', html_file.name, pdf_file.name])

        # Clean up
        html_file.close()
    else:
        # wkhtmltopdf, using URL
        call_subprocess([wkhtmltopdf_cmd, '-q', url, pdf_file.name])

    return pdf_file

Update default location