[Python] PDF 파일을 이미지로 변환하여 그누보드에 자동 글 작성

안녕하세요?? ^-^

SIR 회원님들도 주말 잘 보내고 계시는가요??

PDF 파일의 각 페이지를 PNG 이미지 파일로 변환한 후,

이를 그누보드에 업로드하여 자동으로 글 작성을 하는 Python 스크립트를 작성해봤어요! :)

가끔은 PDF 뷰어 형식으로 보는 것보다 이미지로 스크롤해서 보는게 더 편한 경우도 있더군요~ ^^

보시다시피 PDF2image라는 모듈을 사용하여 PDF -> PNG 변환을 했구요~

만약 PDF2image 모듈의 설치에 어려움을 겪으신다면, 제가 작성한 아래 웹 문서를 참고하세요 :)

https://studyforus.com/tipnknowhow/826049

실행과정은 다음과 같이 진행됩니다!

1. 임시폴더 생성

2. 임시폴더에서 PDF -> PNG 변환

3. DB 연결 후 파일명으로 게시글 작성

4. FTP로 PNG 파일 업로드

5. DB에 파일 업로드 정보를 저장

6. 임시폴더 삭제

[code]

import pymysql, ftputil, hashlib, os, sys, shutil

from pdf2image import convert_from_path

from datetime import datetime

from PIL import Image

def pdf2png(file_name):

    temp = os.path.splitext(file_name)[0]

    pages = convert_from_path(file_name, poppler_path='poppler설치폴더를입력하세요')

    folder = '.\\temp_pdf2png\\'

    os.makedirs(folder)

    png_file_list = []

    for i, page in enumerate(pages, 1):

        png_file_name = f'{folder}{temp}{i}.png'

        page.save(f'{folder}{temp}{i}.png', 'PNG')

        png_file_list.append(png_file_name)

    return png_file_list

def file_upload(filename, bf_file): # FTP를 이용하여 파일을 업로드하는 함수입니다.

    with ftputil.FTPHost('URL을입력하세요', 'FTP계정ID를입력하세요', 'FTP계정비번을입력하세요') as fh:

        fh.chdir('/web/data/file/board') # 업로드할 디렉토리를 입력하세요

        fh.upload(filename, bf_file, callback = None)

    return

def get_filename(filename): # 파일명을 변환하는 함수입니다.

    ms = datetime.now().microsecond

    encoded_name = filename.encode('utf-8')

    result = f'{ms}_{hashlib.sha1(encoded_name).hexdigest()}'

    return result

def board_write(board, subject, content, mb_id, nickname, file_list = None):

    # MySQL connection 및 cursor 생성

    conn = pymysql.connect(host = 'URL을입력하세요', 

                           user = 'DB계정ID를입력하세요', 

                           password = 'DB계정비번을입력하세요',

                           db = 'DB명을입력하세요', 

                           charset = 'utf8')

    curs = conn.cursor()

    # 작성글 INSERT

    sql = f"select wr_num from g5_write_{board}"

    curs.execute(sql)

    wr_num = str(int(curs.fetchone()[0]) - 1)

    now = datetime.today().strftime('%Y-%m-%d %H:%M:%S') # 그누보드의 날짜 형식 준수 (ex: 2021-04-05 23:45:15)

    sql = f"insert into g5_write_{board} set wr_num = {wr_num}, \

          wr_reply = '', wr_comment = 0, ca_name = '', wr_option = 'html1', wr_subject = '{subject}', \

          wr_content = '{content}', wr_link1 = '', wr_link2 = '', \

          wr_link1_hit = 0, wr_link2_hit = 0, wr_hit = 1, wr_good = 0, wr_nogood = 0, \

          mb_id = '{mb_id}', wr_password = '', wr_name = '{nickname}', wr_email = '', wr_homepage = '', \

          wr_datetime = '{now}', wr_last = '{now}', wr_ip = '111.111.111.111', \

          wr_1 = '', wr_2 = '', wr_3 = '', wr_4 = '', wr_5 = '', \

          wr_6 = '', wr_7 = '', wr_8 = '', wr_9 = '', wr_10 = '', \

          wr_comment_reply = '', wr_facebook_user = '', wr_twitter_user = '', \

          as_re_name = '', as_tag = '', as_map = '', as_icon = '', as_thumb = '', as_video = ''"

    curs.execute(sql)

    # 부모 아이디에 UPDATE

    sql = f"select wr_id from g5_write_{board}"

    curs.execute(sql)

    wr_id = str(curs.fetchall()[-1][0])

    sql = f"update g5_write_{board} set wr_parent = {wr_id} where wr_id = {wr_id}"

    curs.execute(sql)

    # 새글 INSERT

    sql = f"insert into g5_board_new ( bo_table, wr_id, wr_parent, bn_datetime, mb_id ) values \

          ( '{board}', '{wr_id}', '{wr_id}', '{now}', '{mb_id}' )"

    curs.execute(sql)

    # 게시글 1 증가

    sql = f"select bo_count_write from g5_board where bo_table = '{board}'"

    curs.execute(sql)

    bo_count_write = str(int(curs.fetchone()[0]))

    sql = f"update g5_board set bo_count_write = {bo_count_write} + 1 where bo_table = '{board}'"

    curs.execute(sql)

    # 파일 업로드 및 관련 정보를 테이블에 저장

    if not file_list or file_list == [''] or file_list == []: # 첨부파일이 없는 경우에는 스크립트를 중단합니다.

        conn.close()

        sys.exit()

    file_count = len(file_list)

    for cnt, file in enumerate(file_list):

        ext = os.path.splitext(file)[1].lstrip('.')

        bf_file = f'{get_filename(file)}.{ext}'

        file_upload(file, bf_file)

        im = Image.open(file)

        width, height = im.size

        size = os.path.getsize(file)

        sql = f"insert into g5_board_file set bo_table = '{board}', wr_id = '{wr_id}', \

              bf_no = '{cnt}', bf_source = '{file}', bf_file = '{bf_file}', \

              bf_content = '', bf_download = 0, bf_filesize = '{size}', \

              bf_width = '{width}', bf_height = '{height}', bf_type = '3', bf_datetime = '{now}'" # PNG 파일은 type 3입니다.

        curs.execute(sql)

    # 파일의 개수를 게시물에 업데이트

    sql = f"update g5_write_board set wr_file = '{file_count}' where wr_id = '{wr_id}'"

    curs.execute(sql)

    # MySQL connection 닫기 및 로그 출력

    conn.close()

    print(f'총 {file_count}장의 이미지가 "{subject}"라는 제목으로 업로드되었습니다.')

    return

def main():

    filename = '업로드할PDF파일명을입력하세요'

    board = '게시판명을입력하세요'

    subject = os.path.splitext(filename)[0] # 글 제목을 파일명으로 합니다.

    content = ''

    mb_id = '아이디를입력하세요'

    nickname = '닉네임을입력하세요'

    file_list = pdf2png(filename) 

    board_write(board, subject, content, mb_id, nickname, file_list)

    shutil.rmtree('.\\temp_pdf2png\\') # 임시폴더를 삭제합니다.

    return

if __name__ == "__main__":

    main()

[/code]

위 스크립트를 실행하면 아래와 같이 파일명을 제목으로 한 글이 작성된 것을 확인할 수 있어요~ ^^

해당 게시글을 클릭하면 아래와 같이 PDF 파일의 총 4페이지가 4개의 이미지로 변환되어 업로드되어 있습니다 :)

저작권에 저촉되지 않는 PDF 파일로 테스트했어요!

여러모로 부족한 점이 많은 스크립트이지만, 아무쪼록 잘 사용하시면 좋겠네요 :)

다음에는 HWP 파일의 내용을 그누보드에 게시글로 올리는 스크립트를 작성해볼게요~

그럼 남은 주말 즐겁고 뜻깊게 보내시고, 날씨가 무더운데 항강 건강하세요!

SIR 회원님들께 항상 감사드립니다~ ^-^