import os
import fitz
import glob
import requests


os.environ['NLS_LANG'] = 'SIMPLIFIED CHINESE_CHINA.UTF8'

#剪切pdf為圖片
def CutPdf(pdfPath,savePath,pdfSavePath):
	try:
		doc = fitz.open(pdfPath) #pdf路徑
		for pg in range(doc.pageCount):
			page = doc[pg]
			rotate = int(0)
		    # 每個(gè)尺寸的縮放系數(shù)為2，這將為我們生成分辨率提高四倍的圖像。
			zoom_x = 2.0
			zoom_y = 2.0
			trans = fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)
			pm = page.getPixmap(matrix=trans, alpha=False)
			imagePath = savePath + '%s.png' % str(pg + 1)
			pm.writePNG(imagePath) #最終存儲路徑
			newpdfPath = pdfSavePath + '%s.pdf' % str(pg + 1)
			print("剪切一張圖片：" + str(pg + 1))
			frompic2pdf(imagePath,pdfSavePath,newpdfPath)
			print("轉(zhuǎn)換成功一張pdf：" + str(pg + 1))
	except Exception as e:
		print('CutPdf出現(xiàn)異常:' + str(e))

#圖片轉(zhuǎn)pdf
def frompic2pdf(img_path,pdfSavePath, pdf_path):
	# 使用glob讀圖
	for img in sorted(glob.glob(img_path)):
		if not os.path.exists(pdfSavePath):
			 os.makedirs(pdfSavePath)
		
		# 打開空文檔
		doc = fitz.open()
		# 打開指定圖片
		imgdoc = fitz.open(img)
		# 使用圖片創(chuàng)建單頁的PDF
		pdfbytes = imgdoc.convertToPDF()
		imgpdf = fitz.open("pdf", pdfbytes)
		# 將當(dāng)前頁寫入文檔
		doc.insertPDF(imgpdf)
		# 保存為指定名稱的PDF文件
		doc.save(pdf_path)
		# 關(guān)閉
		doc.close()


def main():
    pdfUrl = "cards"
    fileList = os.listdir(pdfUrl)
    index = 1
    for item in  fileList:
        print("開始裁剪" + item)
        url = pdfUrl + "/" + item

        foldStr = "images/" + item.split('.')[0]
        url2 = foldStr + "/" + item.split('.')[0]
        foldStr3 = "pdf/" + item.split('.')[0]
        url3 = foldStr3 + "/"
        if not os.path.exists(foldStr):
            os.makedirs(foldStr)

        CutPdf(url,url2,url3)
        print("完成裁剪")
        index+=1

if __name__ == "__main__":
    main()

# 獲取頁面尺寸
page_width = page.mediaBox.getWidth()
page_height = page.mediaBox.getHeight()

# 定義剪裁區(qū)域的位置和大小
x = 100  # 起始點(diǎn)的x坐標(biāo)
y = 100  # 起始點(diǎn)的y坐標(biāo)
width = 200  # 剪裁區(qū)域的寬度
height = 300  # 剪裁區(qū)域的高度

# 剪裁PDF頁面
page.trimBox.lowerLeft = (x, y)
page.trimBox.upperRight = (x + width, y + height)
page.cropBox.lowerLeft = (x, y)
page.cropBox.upperRight = (x + width, y + height)

from PIL import Image

# 將剪裁后的頁面保存為圖片
with open('output.jpg', 'wb') as image_file:
    page_data = page.extract_xobject().get('/Im0').getData()
    image = Image.open(io.BytesIO(page_data))
    image.save(image_file, 'JPEG')

from PyPDF2 import PdfFileReader, PdfFileWriter
 
 
# PDF文件分割
def split_pdf(read_file, out_detail):
    try:
        fp_read_file = open(read_file, 'rb')
        pdf_input = PdfFileReader(fp_read_file)  # 將要分割的PDF內(nèi)容格式話
        page_count = pdf_input.getNumPages()  # 獲取PDF頁數(shù)
        print(page_count)  # 打印頁數(shù)
 
        with open(out_detail, 'r',True,'utf-8')as fp:
            # print(fp)
            txt = fp.readlines()
            # print(txt)
            for detail in txt:  # 打開分割標(biāo)準(zhǔn)文件
                # print(type(detail))
                pages, write_file = detail.split()  # 空格分組
               #  write_file, write_ext = os.path.splitext(write_file)  # 用于返回文件名和擴(kuò)展名元組
                pdf_file = r'C:\Users\GZTSALFIEL\Desktop\excel問題文件\pdf\\'+f'{write_file}.pdf'
                # liststr=list(map(int, pages.split('-')))
                # print(type(liststr))
                start_page, end_page = list(map(int, pages.split('-')))  # 將字符串?dāng)?shù)組轉(zhuǎn)換成整形數(shù)組
                start_page -= 1
                try:
                    print(f'開始分割{start_page}頁-{end_page}頁，保存為{pdf_file}......')
                    pdf_output = PdfFileWriter()  # 實(shí)例一個(gè) PDF文件編寫器
                    for i in range(start_page, end_page):
                        pdf_output.addPage(pdf_input.getPage(i))
                    with open(pdf_file, 'wb') as sub_fp:
                        pdf_output.write(sub_fp)
                    print(f'完成分割{start_page}頁-{end_page}頁，保存為{pdf_file}!')
                except IndexError:
                    print(f'分割頁數(shù)超過了PDF的頁數(shù)')
        # fp.close()
    except Exception as e:
        print(e)
    finally:
        fp_read_file.close()
 
 
 
# def main():
#     fire.Fire(split_pdf)
#
# if __name__ == '__main__':
#     main()
 
split_pdf(r'..\pdfdocement\myfile.pdf', 'consult.txt')