【Python】PDF編集アプリを開発

from pdf2image import convert_from_path

# 表示用にPDFをPIL画像データのリストに変換
pil_images = convert_from_path(
	pdf_path,
	poppler_path='/usr/local/bin',
	dpi=200
)

# self.canvasはCanvasクラスのインスタンス
self.canvas.create_image(
	0, 0,
	image=tk_image,
	anchor=tkinter.NW,
	tag="show_image"
)

# イベントを再設定
self.canvas.bind("<ButtonPress>", self.click)
self.canvas.bind("<Motion>", self.motion)
self.canvas.bind("<ButtonRelease>", self.release)

# canvasはCanvasクラスのインスタンス
# tagは描画した図形に付けられたタグ
canvas.delete(tag)

# textはEntryクラスのインスタンス
text.place_forget()

# textはEntryクラスのインスタンス
text.place(x, y)

for page in self.pages:# ページに対するループ

	# 〜略〜

	# text_info_list は各ページに描画したテキストの情報を記録するリスト
	for text_info in text_info_list[i]:# テキストに対するループ

# 座標を取得
# line_infoは線の情報を格納したリスト
x1, y1, x2, y2 = line_info["place"]

# PDFの座標に変換（pp.wはPDFのページの幅、pp.hはPDFのページの高さ）
target_x1 = x1 * pp.w / self.images[0].width()
target_x2 = x2 * pp.w / self.images[0].width()
target_y1 = pp.h - y1 * pp.h / self.images[0].height()
target_y2 = pp.h - y2 * pp.h / self.images[0].height()

# 線描画
cc.line(target_x1, target_y1, target_x2, target_y2)

import tkinter
import tkinter.filedialog
from PIL import Image, ImageTk
import os
from pdfrw import PdfReader
from pdfrw.buildxobj import pagexobj
from pdfrw.toreportlab import makerl
from reportlab.pdfgen import canvas
from reportlab.pdfbase.cidfonts import UnicodeCIDFont
from reportlab.pdfbase import pdfmetrics
from reportlab.lib.units import mm
from pdf2image import convert_from_path

# キャンバスのサイズ設定
CANVAS_WIDTH = 400
CANVAS_HEIGHT = 600

# 図形の色の設定
TEXT_COLOR = "red"
LINE_COLOR = "green"
RECT_COLOR = "blue"

class Figure():
	'''図形を描画するクラス'''

	def __init__(self, canvas, num_page):
		'''
		初期化メソッド
		canvas:キャンバスウィジェット
		num_page：ページ数
		'''

		# 図形の描画先ウィジェット設定
		self.canvas = canvas

		# 図形情報記憶用のリスト作成
		self.all_list = list(range(num_page))
		for i in range(len(self.all_list)):
			self.all_list[i] = []

		# マウス位置の初期化
		self.sx = 0
		self.sy = 0
		self.ex = 0
		self.ey = 0

	def remove(self, num_page):
		'''
		図形を削除するメソッド
		num_page:図形を削除するページ番号
		'''

		# 図形を削除するページのリストを取得
		page_list = self.all_list[num_page]

		for figure in page_list:

			# タグを指定して図形を削除
			tag = figure["tag"]
			self.canvas.delete(tag)

	def place(self, num_page):
		'''
		図形を再描画するメソッド
		num_page:図形を再描画するページ番号
		'''

		# 図形を描画するページのリストを取得
		page_list = self.all_list[num_page]
		for figure in page_list:
			tag = figure["tag"]
			place = figure["place"]

			# 図形を際描画
			self.create_method(place, tag=tag)

	def motion(self, sx, sy, ex, ey):
		'''
		マウス移動時に図形を一時的に描画するメソッド
		sx, sy, ex, ey:図形の座標
		'''

		# 一旦図形を削除
		self.canvas.delete("drawing_figure")

		# 指定された座標に図形を描画
		self.create_method(
			sx, sy, ex, ey,
			tag="drawing_figure"
		)

	def create(self, sx, sy, ex, ey, num_page):
		'''
		マウスクリック終了時に図形を描画するメソッド
		sx, sy, ex, ey:図形の座標
		num_pag:図形の描画先のページのページ番号
		'''

		# 図形を一旦削除
		self.canvas.delete("drawing_line")

		# タグを生成
		tag = self.prefix + str(num_page) + "_" + str(len(self.all_list[num_page]))
		
		# 図形を描画
		self.create_method(
			sx, sy, ex, ey,
			tag=tag
		)
		
		# 図形の情報をリストに記憶
		info = {
			"tag":tag,
			"place":(sx, sy, ex, ey),
		}
		self.all_list[num_page].append(info)
	
	def get_list(self):
		'''
		図形の情報を格納したリストを取得するメソッド
		'''

		return self.all_list

class Text(Figure):
	'''テキストを描画するクラス'''
	def __init__(self, canvas, num_page):
		super().__init__(canvas, num_page)

		# テキストならではの情報を設定
		self.prefix = "rect_"
		self.create_method = self.canvas.create_rectangle

	def remove(self, num_page):
		'''
		テキストを削除するメソッド
		num_page:テキストを削除するページ番号
		'''

		# テキストを削除するページのリストを取得
		page_list = self.all_list[num_page]
		for text in page_list:

			# ウィジェットを取得してアプリから除去
			obj = text["obj"]
			obj.place_forget()

	def place(self, num_page):
		'''
		テキストを再描画するメソッド
		num_page:テキストを再描画するページ番号
		'''

		# テキストを再描画するページのリストを取得
		page_list = self.all_list[num_page]
		for text in page_list:

			# ウィジェットを取得して再配置
			obj = text["obj"]
			place = text["place"]
			obj.place(x=place[0], y=place[1])

	def motion(self, sx, sy, ex, ey):
		'''
		マウス移動時に図形を一時的に描画するメソッド
		sx, sy, ex, ey:図形の座標
		'''

		# 長方形を一旦削除
		self.canvas.delete("text_box")

		# 長方形を指定された座標に描画
		self.canvas.create_rectangle(
			sx, sy, ex, ey,
			tag="text_box"
		)

	def create(self, sx, sy, ex, ey, num_page):
		'''
		テキストを描画するメソッド
		sx, sy, ex, ey:テキストの座標
		num_page:テキストの描画先のページのページ番号
		'''

		# テキストを表していた長方形を削除
		self.canvas.delete("text_box")

		# フォントサイズを取得
		font_size = self.font_size_adjust(max(sy, ey) - min(sy, ey))
		
		# 横幅（文字数）をザックリ計算
		if font_size != 0:
			width = (max(sx, ex) - min(sx, ex)) // font_size * 2
		else:
			width = 1

		# Entryウィジェットを作成
		text = tkinter.Entry(
			self.canvas,
			foreground=TEXT_COLOR,
			width=width,
			highlightthickness=0,
			font=("", font_size)
		)

		# Entryウィジェットを配置
		text.place(x=min(sx, ex), y=min(sy, ey))

		# テキストの情報をリストに記録
		info = {
			"obj":text,
			"place":(sx, sy, ex, ey),
			"size":font_size
		}
		self.all_list[num_page].append(info)
	
	def get_list(self):
		'''
		テキストの情報を格納したリストを取得するメソッド
		'''

		return self.all_list

	def font_size_adjust(self, size):
		'''
		フォントサイズの微調整を行うメソッド
		'''

		return int(size / 1.2)

class Line(Figure):
	'''線を描画するクラス'''

	def __init__(self, canvas, num_page):
		super().__init__(canvas, num_page)

		# 線ならではの情報を設定
		self.prefix = "line_"
		self.create_method = self.canvas.create_line

class Rect(Figure):
	'''長方形を描画するクラス'''

	def __init__(self, canvas, num_page):
		super().__init__(canvas, num_page)

		# 長方形ならではの情報を設定
		self.prefix = "rect_"
		self.create_method = self.canvas.create_rectangle

class PdfEditor():
	'''
	図形編集アプリクラス
	'''

	def __init__(self, master):
		self.master = master


		self.canvas_width = CANVAS_WIDTH
		self.canvas_height = CANVAS_HEIGHT

		self.pressing = False
		
		self.create_widgets()

		self.text = None
		self.line = None
		self.rect = None
		self.figure = None

	def create_widgets(self):
		'''
		ウィジェットを作成として配置するメソッド
		'''

		self.canvas = tkinter.Canvas(
			self.master,
			width=self.canvas_width,
			height=self.canvas_height,
			highlightthickness=0
		)
		self.canvas.grid(
			column=0,
			row=0
		)

		self.operation_frame = tkinter.Frame(
			self.master
		)
		self.operation_frame.grid(
			column=1,
			row=0
		)

		self.read_button = tkinter.Button(
			self.operation_frame,
			text="読み込み",
			command=self.file_read
		)
		self.read_button.grid(
			column=0,
			row=0
		)

		self.write_button = tkinter.Button(
			self.operation_frame,
			text="保存",
			command=self.write,
			state=tkinter.DISABLED
		)
		self.write_button.grid(
			column=0,
			row=1
		)
		
		self.text_button = tkinter.Button(
			self.operation_frame,
			text="テキスト",
			command=self.text_mode,
			state=tkinter.DISABLED
		)
		self.text_button.grid(
			column=0,
			row=2
		)

		self.line_button = tkinter.Button(
			self.operation_frame,
			text="線",
			command=self.line_mode,
			state=tkinter.DISABLED
		)
		self.line_button.grid(
			column=0,
			row=3
		)

		self.rect_button = tkinter.Button(
			self.operation_frame,
			text="長方形",
			command=self.rect_mode,
			state=tkinter.DISABLED
		)
		self.rect_button.grid(
			column=0,
			row=4
		)

		self.prev_button = tkinter.Button(
			self.operation_frame,
			text="前ページ",
			command=self.prev_page,
			state=tkinter.DISABLED,
		)
		self.prev_button.grid(
			column=0,
			row=5
		)

		self.next_button = tkinter.Button(
			self.operation_frame,
			text="次ページ",
			command=self.next_page,
			state=tkinter.DISABLED,
		)
		self.next_button.grid(
			column=0,
			row=6
		)

	def change_prev_next_button_state(self):
		'''
		前ページ・次ページボタンの状態を更新するメソッド
		'''

		# 前ページボタンの状態設定
		if self.show_page > 0:
			self.prev_button.config(state=tkinter.NORMAL)
		else:
			self.prev_button.config(state=tkinter.DISABLED)

		# 次ページボタンの状態設定
		if self.show_page < self.num_page - 1:
			self.next_button.config(state=tkinter.NORMAL)
		else:
			self.next_button.config(state=tkinter.DISABLED)

	def prev_page(self):
		'''
		前ページボタンが押された時の処理
		'''

		# 表示中の図形を削除
		self.text.remove(self.show_page)
		self.line.remove(self.show_page)
		self.rect.remove(self.show_page)

		# 表示中のページ番号を更新
		self.show_page -= 1

		# 前ページをキャンバスに描画
		self.canvas.create_image(
			0, 0,
			image=self.pdf.get_image(self.show_page),
			anchor=tkinter.NW,
			tag="show_image"
		)

		# 前ページに既に描画している図形を再描画
		self.text.place(self.show_page)
		self.line.place(self.show_page)
		self.rect.place(self.show_page)

		# 前ページ・後ページの状態を更新
		self.change_prev_next_button_state()

	def next_page(self):
		'''
		前ページボタンが押された時の処理
		'''

		# 表示中の図形を削除
		self.text.remove(self.show_page)
		self.line.remove(self.show_page)
		self.rect.remove(self.show_page)

		# 表示中のページ番号を更新
		self.show_page += 1

		# 次ページをキャンバスに描画
		self.canvas.create_image(
			0, 0,
			image=self.pdf.get_image(self.show_page),
			anchor=tkinter.NW,
			tag="show_image"
		)

		# 次ページに既に描画している図形を再描画
		self.text.place(self.show_page)
		self.line.place(self.show_page)
		self.rect.place(self.show_page)

		# 前ページ・後ページの状態を更新
		self.change_prev_next_button_state()

	def change_figure_state(self, text, line, rect):
		'''
		図形選択ボタンの状態を更新するメソッド
		text:テキストボタンの状態
		line:線ボタンの状態
		rect:長方形ボタンの状態
		'''

		# 各ボタンの状態を設定
		self.text_button.config(
			state=text
		)
		self.line_button.config(
			state=line
		)
		self.rect_button.config(
			state=rect
		)
		

	def text_mode(self):
		'''図形をテキストに設定するメソッド'''

		# 一旦イベントを全てアンバインドする
		self.canvas.unbind("<ButtonPress>")
		self.canvas.unbind("<Motion>")
		self.canvas.unbind("<ButtonRelease>")
		
		if type(self.figure) == Text:
			# 現在選択されている図形がテキストの場合

			# 全ボタンを通常状態に設定
			self.change_figure_state(
				tkinter.NORMAL,
				tkinter.NORMAL,
				tkinter.NORMAL,
			)

			# 選択されている図形をNoneに設定
			self.figure = None
		else:
			# 現在選択されている図形がテキスト以外の場合

			# イベントを再設定
			self.canvas.bind("<ButtonPress>", self.click)
			self.canvas.bind("<Motion>", self.motion)
			self.canvas.bind("<ButtonRelease>", self.release)
		
			# テキストボタンをアクティブに設定
			self.change_figure_state(
				tkinter.ACTIVE,
				tkinter.NORMAL,
				tkinter.NORMAL,
			)
			
			# 描画図形オブジェクトをTextクラスのオブジェクトに設定
			self.figure = self.text

	def line_mode(self):
		'''図形を線に設定するメソッド'''

		# 一旦イベントを全てアンバインドする
		self.canvas.unbind("<ButtonPress>")
		self.canvas.unbind("<Motion>")
		self.canvas.unbind("<ButtonRelease>")
		
		if type(self.figure) == Line:
			# 現在選択されている図形が線の場合

			# 全ボタンを通常状態に設定
			self.change_figure_state(
				tkinter.NORMAL,
				tkinter.NORMAL,
				tkinter.NORMAL,
			)

			# 選択されている図形をNoneに設定
			self.figure = None
		else:
			# 現在選択されている図形が線以外の場合

			# イベントを再設定
			self.canvas.bind("<ButtonPress>", self.click)
			self.canvas.bind("<Motion>", self.motion)
			self.canvas.bind("<ButtonRelease>", self.release)
		
			# 線ボタンをアクティブに設定
			self.change_figure_state(
				tkinter.NORMAL,
				tkinter.ACTIVE,
				tkinter.NORMAL,
			)
			
			# 描画図形オブジェクトをLineクラスのオブジェクトに設定
			self.figure = self.line

	def rect_mode(self):
		'''図形を長方形に設定するメソッド'''

		# 一旦イベントを全てアンバインドする
		self.canvas.unbind("<ButtonPress>")
		self.canvas.unbind("<Motion>")
		self.canvas.unbind("<ButtonRelease>")
		
		if type(self.figure) == Rect:
			# 現在選択されている図形が長方形の場合

			# 全ボタンを通常状態に設定
			self.change_figure_state(
				tkinter.NORMAL,
				tkinter.NORMAL,
				tkinter.NORMAL,
			)

			# 選択されている図形をNoneに設定
			self.figure = None
		else:
			# 現在選択されている図形が長方形以外の場合

			# イベントを再設定
			self.canvas.bind("<ButtonPress>", self.click)
			self.canvas.bind("<Motion>", self.motion)
			self.canvas.bind("<ButtonRelease>", self.release)
		
			# 長方形ボタンをアクティブに設定
			self.change_figure_state(
				tkinter.NORMAL,
				tkinter.NORMAL,
				tkinter.ACTIVE,
			)
			
			# 描画図形オブジェクトをRectクラスのオブジェクトに設定
			self.figure = self.rect

	def click(self, event):
		'''
		キャンバスクリック時の処理
		'''

		# クリック中フラグをON
		self.pressing = True

		# 座標を記憶
		self.start = (event.x, event.y)

	def motion(self, event):
		'''
		マウス移動時の処理
		'''

		if self.pressing:
			# 図形のクラスのmotionメソッド実行
			self.figure.motion(self.start[0], self.start[1], event.x, event.y)

	def release(self, event):
		'''
		マウスクリック終了時の処理
		'''

		# マウスクリック中フラグをOFF
		self.pressing = False

		# 座標を記憶
		self.end = (event.x, event.y)

		# 図形のクラスのcreateeメソッド実行
		self.figure.create(
			self.start[0], self.start[1],
			self.end[0], self.end[1],
			self.show_page
		)

	def file_read(self):
		'''
		ファイルの読み込みを行うメソッド
		'''

		# ファイル選択画面を表示してファイルパスを取得
		self.path = tkinter.filedialog.askopenfilename(
			filetypes=[
				("PDFファイル", "*.pdf"),
			],
			title="ファイル選択",
		)

		# PDFクラスのオブジェクト生成
		self.pdf = Pdf()

		# 指定されたPDFファイルをtkinter画像に変換
		size = (self.canvas_width, self.canvas_height)
		self.num_page = self.pdf.read(self.path, size)

		# ページ0の画像をキャンバスに描画
		self.show_page = 0
		self.canvas.create_image(
			0, 0,
			image=self.pdf.get_image(self.show_page),
			anchor=tkinter.NW,
			tag="show_image"
		)

		self.change_prev_next_button_state()
		self.rect_button.config(
			state=tkinter.NORMAL
		)
		self.line_button.config(
			state=tkinter.NORMAL
		)
		self.text_button.config(
			state=tkinter.NORMAL
		)
		self.write_button.config(
			state=tkinter.NORMAL
		)

		# 図形描画するオブジェクトを設定
		self.text = Text(self.canvas, self.num_page)
		self.line = Line(self.canvas, self.num_page)
		self.rect = Rect(self.canvas, self.num_page)

	def write(self):
		'''
		PDF保存を行うメソッド
		'''

		# ファイル選択ダイアログの表示
		output_path = tkinter.filedialog.asksaveasfilename(
			initialfile="output.pdf",
			defaultextension="pdf",
		)
		
		# PDFファイル保存
		self.pdf.write(
			output_path,
			self.text.get_list(),
			self.line.get_list(),
			self.rect.get_list()
		)

class Pdf():
	'''
	PDFに関する処理を行うクラス
	'''

	def __init__(self):

		# 初期設定
		self.dpi = 200
		self.images = None

	def read(self, pdf_path, size):
		'''
		PDFを画像に変換するメソッド
		pdf_path：画像に変換するPDFへのファイルパス
		size:キャンバスのサイズ
		'''

		# PDFへのパスを覚えておく
		self.pdf_path = pdf_path
		
		# 表示用にPDFをPIL画像データのリストに変換
		pil_images = convert_from_path(
			pdf_path,
			poppler_path='/usr/local/bin',
			dpi=self.dpi
		)

		# キャンバス描画用に画像をリサイズする時の拡大率計算
		x_ratio = size[0] / pil_images[0].width
		y_ratio = size[1] / pil_images[0].height
		
		# キャンバス内に収まるように小さいほうの拡大率を採用
		image_ratio = min(x_ratio, y_ratio)

		# リサイズ後のサイズを計算
		resize_size = (
			int(pil_images[0].width * image_ratio),
			int(pil_images[0].height * image_ratio)
		)

		# リサイズ後のtkinter画像をリストに追加
		self.images = []
		for pil_image in pil_images:
			# リサイズ
			resize_image = pil_image.resize(resize_size)

			# tkinter画像に変換
			tk_image = ImageTk.PhotoImage(resize_image)

			# リストに追加
			self.images.append(tk_image)

		# ページ数を返却
		return len(self.images)

	def get_image(self, num):
		'''
		画像を取得するメソッド
		num:取得する画像のページ番号
		'''

		# ページ数を超える場合はNoneを返却
		if num < 0 or num >= len(self.images):
			return None

		# 指定されたページの画像データを返却
		return self.images[num]

	def write(self, out_path, text_info_list, line_info_list, rect_info_list):
		'''
		PDFに図形を描画してファイル保存するメソッド
		out_path:保存するPDFのファイルパス
		text_info_list:テキストの情報を格納したリスト
		line_info_list:線の情報を格納したリスト
		rect_info_list:長方形の情報を格納したリスト
		'''

		# PDFを読み込む
		self.pdf = PdfReader(self.pdf_path, decompress=False)

		# ページリスト取得
		self.pages = self.pdf.pages

		# PDFデータ作成
		cc = canvas.Canvas(out_path)

		# フォントを登録
		font_name = "HeiseiKakuGo-W5"
		pdfmetrics.registerFont(UnicodeCIDFont(font_name))

		i = 0
		for page in self.pages:

			# PDFデータにページを展開
			pp = pagexobj(page)
			rl_obj = makerl(cc, pp)
			cc.doForm(rl_obj)

			for text_info in text_info_list[i]:
				# フォント設定
				cc.setFont(font_name, text_info["size"] * pp.h / self.images[0].height())

				# 座標を取得
				sx, sy, ex, ey = text_info["place"]
				
				x1 = min(sx, ex)
				x2 = max(sx, ex)
				y1 = max(sy, ey)
				y2 = min(sy, ey)

				# PDFの座標に変換
				target_x1 = x1 * pp.w / self.images[0].width()
				target_y1 = pp.h - y1 * pp.h / self.images[0].height()

				# テキストの色を設定
				cc.setFillColor(TEXT_COLOR)

				# テキスト描画
				cc.drawString(target_x1, target_y1, text_info["obj"].get())



			for line_info in line_info_list[i]:

				# 座標を取得
				x1, y1, x2, y2 = line_info["place"]
				
				# PDFの座標に変換
				target_x1 = x1 * pp.w / self.images[0].width()
				target_x2 = x2 * pp.w / self.images[0].width()
				target_y1 = pp.h - y1 * pp.h / self.images[0].height()
				target_y2 = pp.h - y2 * pp.h / self.images[0].height()
				
				# 線の色を設定
				cc.setStrokeColor(LINE_COLOR)

				# 線描画
				cc.line(target_x1, target_y1, target_x2, target_y2)

			
			for rect_info in rect_info_list[i]:

				# 座標取得
				sx, sy, ex, ey = rect_info["place"]
				
				x1 = min(sx, ex)
				x2 = max(sx, ex)
				y1 = max(sy, ey)
				y2 = min(sy, ey)

				# PDFの座標に変換
				target_x1 = x1 * pp.w / self.images[0].width()
				target_y1 = pp.h - y1 * pp.h / self.images[0].height()
				width = (x2 - x1) * pp.w / self.images[0].width()
				height = (y1 - y2) * pp.h / self.images[0].height()
			
				# 線の色を設定
				cc.setStrokeColor(RECT_COLOR)

				# 長方形描画
				cc.rect(target_x1, target_y1, width, height)

			# ページ確定
			cc.showPage()
			i += 1

		# 確定したページをPD保存
		cc.save()


app = tkinter.Tk()

# アプリの起動
pdfEditor = PdfEditor(app)

app.mainloop()

スクリプトの設定

スクリプトの下記部分で簡単な設定を行うことができます。

# キャンバスのサイズ設定
CANVAS_WIDTH = 400
CANVAS_HEIGHT = 600

# 図形の色の設定
TEXT_COLOR = "red"
LINE_COLOR = "green"
RECT_COLOR = "blue"