svg.py

import glob
from os import getcwd
from os.path import basename, join

import cairosvg
import PyPDF2


def rotate(file_path, angle):
    file = PyPDF2.PdfFileReader(open(file_path, 'rb'))
    print(file_path + "を回転します。")
    file_output = PyPDF2.PdfFileWriter()
    for page_num in range(file.numPages):
        page = file.getPage(page_num)
        page.rotateClockwise(angle)
        print("回転！！")
        file_output.addPage(page)
        print(file_path + "の書き込み準備完了")
    with open(file_path, "wb") as f:
        file_output.write(f)
        print(file_path + "の書き込み完了！")


def convert(dir):
    path_list = []

    path_file = join(dir, "*.svg")
    filename = glob.glob(path_file)
    path_list.extend(filename)

    for file in path_list:
        converted_file_name = join(dir, (basename(file).split('.', 1)[0] + ".pdf"))
        print(file + "を" + converted_file_name + "に変換します。")
        cairosvg.svg2pdf(url=file, write_to=converted_file_name)
        print(converted_file_name + "を出力しました。")
        # rotate(converted_file_name, 270)
        # print(converted_file_name + "を回転しました。")
    return


def main():
    image_path = getcwd() + '/images'
    print(image_path + 'のファイルが対象です。')
    convert(image_path)


if __name__ == "__main__":
    main()