Extract the first page of pdf document in python

python

import PyPDF2

def extract_first_page(pdf_path, output_path):
    pdf_reader = PyPDF2.PdfFileReader(open(pdf_path, "rb"))
    if pdf_reader.numPages > 0:
        pdf_writer = PyPDF2.PdfFileWriter()
        pdf_writer.addPage(pdf_reader.getPage(0))

        with open(output_path, "wb") as output_file:
            pdf_writer.write(output_file)