『退屈なことはPythonにやらせよう』その11(13章:PDFとWord)

#! python3

#モジュールの読み込み
import PyPDF2

#ファイルの読み込み(バイナリ読み込みモード)
pdf_file = open("pdfファイルのパス", "rb")
pdf_reader = PyPDF2.PdfFileReader(pdf_file)

#ページ指定
page_obj = pdf_reader.getPage(1) #最初のページは 0

#テキスト抽出
page_obj.extractText()

pdf_reader.isEncrypted #パスワードが必要なら、Trueが返る

pdf_reader.decrypt("パスワード") #パスワードの入力(残りの作業は上と同じ)

import PyPDF2

#準備
pdf_file = open("pdfファイルのパス", "rb")    #pdfをバイナリ読み込みモードで読み込む
pdf_reader = PyPDF2.PdfFileReader(pdf_file)  #PdfFileReaderオブジェクトにする
pdf_writer = PyPDF2.PdfFileWriter()          #出力先のPDF

#pdfページのコピーと書き出し
page_obj = pdf_reader.getPage(page_number)   #page_numberには取得したいページ番号をいれる
pdf_writer.addPage(page_obj)

#pdfページを新規ページとして出力
pdf_output_file = open("保存先 pdf ファイル", "wb")
pdf_writer.write(pdf_output_file)

#開いたファイル達を閉じる(これをやらないと、他のとこで開けなくなったりする)
pdf_output_file.close()
pdf_file_close()

import PyPDF2

#準備
pdf_file = open("pdfファイルのパス", "rb")    #pdfをバイナリ読み込みモードで読み込む
pdf_reader = PyPDF2.PdfFileReader(pdf_file)  #PdfFileReaderオブジェクトにする

#pdfページの回転
page_obj = pdf_reader.getPage(page_number)   #page_numberには取得したいページ番号をいれる
page_obj.rotateClockwise(90)

import PyPDF2

#準備
pdf_file_main = open("pdfファイルのパス", "rb")    #pdfをバイナリ読み込みモードで読み込む
pdf_file_ue = open("pdfファイルのパス", "rb")    #pdfをバイナリ読み込みモードで読み込む
pdf_reader_main = PyPDF2.PdfFileReader(pdf_file_main)  #PdfFileReaderオブジェクトにする
pdf_reader_ue = PyPDF2.PdfFileReader(pdf_file_ue)  #PdfFileReaderオブジェクトにする

#pdfページの重ね合わせ
page_obj_main = pdf_reader_main.getPage(page_number)   #page_numberには取得したいページ番号をいれる(重ね合わせの下)
page_obj_main.mergePage(pdf_reader_ue.getPage(page_number))  #重ね合わせたいページの番号(上になる側)

import PyPDF2

#準備
pdf_file = open("pdfファイルのパス", "rb")    #pdfをバイナリ読み込みモードで読み込む
pdf_reader = PyPDF2.PdfFileReader(pdf_file)  #PdfFileReaderオブジェクトにする
pdf_writer = PyPDF2.PdfFileWriter()          #出力先のPDF

#pdfページのコピーと書き出し
page_obj = pdf_reader.getPage(page_number)   #page_numberには取得したいページ番号をいれる
pdf_writer.addPage(page_obj)


#ここで暗号化
pdf_writer.encrypt("パスワード")

#pdfページを新規ページとして出力
pdf_output_file = open("保存先 pdf ファイル", "wb")
pdf_writer.write(pdf_output_file)

#開いたファイル達を閉じる(これをやらないと、他のとこで開けなくなったりする)
pdf_output_file.close()
pdf_file.close()

import docx

#白紙のワードファイル作成
doc = docx.Document()

#段落の追加
para_obj1 = doc.add_paragraph("一段落目のテキストを書き始める。")
para_obj2 = doc.add_paragraph("二段落目のテキストを書き始める。")

#段落にテキストを足したい。
para_obj1.add_run("1段落目に追加したいテキストを書く。")

#ページ区切りを入れたい

#ここまで前のページ
doc.add_page_break()
#ここから次のページ

#見出しの追加
doc.add_heading("見出しの文字",　0～4)) #見出しは0が大きくて、4が小さい感じになります。

#ワードファイルを保存する
doc.save("保存したいワードファイルの名前")

#! python3
# check13_1.py  - フォルダの中のpdfを全て見つけて暗号化する

#debug
import logging
logging.basicConfig(level=logging.DEBUG,
format = " %(asctime)s - %(levelname)s - %(message)s")
logging.debug("プログラム開始")
#logging.disable(logging.CRITICAL)

#walk
import os, PyPDF2

for foldername, subfolders, filenames in os.walk(os.getcwd()):
    logging.debug("The current folder is " + foldername)

    for subfolder in subfolders:
        logging.debug("SUBFOLDER OF " + foldername + ": " +subfolder)
    
    for filename in filenames:
        if filename.endswith(".pdf"):
            logging.debug("FILE INSIDE " + foldername + ": " + filename)
            logging.debug("{}の暗号化を開始します".format(filename))
            pdf_file = open(filename, "rb")  #pdfを開く
            pdf_reader = PyPDF2.PdfFileReader(pdf_file)
            if not pdf_reader.isEncrypted:
                pdf_writer = PyPDF2.PdfFileWriter()   #新規のpdfを作成
                for page_num in range(pdf_reader.numPages):
                    pdf_writer.addPage(pdf_reader.getPage(page_num))
                pdf_writer.encrypt("bananafish") #暗号化
                result_pdf = open(filename[:-4] + "_encrypt.pdf", "wb")
                pdf_writer.write(result_pdf)
                result_pdf.close()
            else:
                print("{}は既に暗号化されています".format(filename))

            logging.debug("{}の暗号化が完了しました".format(filename))  
    print("")

logging.debug("プログラム終了")

#! python3
# check13_2.py  -  テキストファイルから名前を受け取り、それに合った招待状を作成する。

#debug
import logging
logging.basicConfig(level=logging.DEBUG,
format = " %(asctime)s - %(levelname)s - %(message)s")
logging.debug("プログラム開始")
#logging.disable(logging.CRITICAL)

import docx
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.shared import Pt

to_send_file = open("guests.txt", "r")
guests = to_send_file.readlines()
to_send_file.close()

doc = docx.Document()

for guest in guests:
    #text
    #para1
    para_obj1 = doc.add_paragraph("It would be a pleasure to have the company of")
    #para2
    para_obj2 = doc.add_paragraph(guest.rstrip("\n"))
    para_obj2.runs[0].underline = True
    #para3
    para_obj3 = doc.add_paragraph("")
    para_obj3.add_run("at")
    para_obj3.add_run(" 11010 Memory of Lane on the Evening of")
    para_obj3.runs[0].underline = True
    #para4
    para_obj4 = doc.add_paragraph("April 1st")
    #para5
    para_obj5 = doc.add_paragraph("")
    para_obj5.add_run("at")
    para_obj5.add_run(" 7 o'clock")
    para_obj5.runs[0].underline = True
    doc.add_page_break()
    #alignment
    for para in doc.paragraphs:
        para.alignment = WD_ALIGN_PARAGRAPH.CENTER
    #font-size
        for run in para.runs:
            font = run.font
            font.name = 'Calibri'
            font.size = Pt(20)

doc.save("invitations.docx")
logging.debug("プログラム終了")

#! python3
# check13_3.py  -  総当たり法でpdfをアンロックする
#debug
import logging
logging.basicConfig(level=logging.DEBUG,
format = " %(asctime)s - %(levelname)s - %(message)s")
logging.debug("プログラム開始")
logging.disable(logging.CRITICAL)

import PyPDF2
en_file = "allminutes_encrypt.pdf"

dic_file = open("dictionary.txt", "r")
key_dict = []
for keys in dic_file.readlines():
    key_dict.append(keys)
logging.debug("ket_dictの長さは{}".format(len(key_dict)))
dic_file.close()
logging.debug(key_dict[:5])

logging.debug("pdfを準備しています。")
pdf_file = open(en_file, "rb")
pdf_reader = PyPDF2.PdfFileReader(pdf_file)

#pdfの暗号を解読
print("pdfの暗号を解読します。")
for count, key in enumerate(key_dict):
    trial1 = pdf_reader.decrypt(key.strip("\n").lower())
    trial2 = pdf_reader.decrypt(key.strip("\n").upper())

    if trial1 == 1:
        print("キーワードは{}でした。".format(key.strip("\n").lower()))
        break
    elif trial2 == 1:
        print("キーワードは{}でした".format(key.strip("\n").upper()))
        break
    else:
        logging.debug("{}番目の{}はキーワードではありませんでした".format(count, key))