app.py

import platform
from pathlib import Path
from subprocess import call

import dearpygui.dearpygui as dpg

from model import model

# import dearpygui.demo as demo

# NOTE Build with: pyinstaller --onefile --windowed app.py


# def macos_settings():
#     if platform.system() == "Darwin":
#         font = Path("fonts/Menlo-Regular.ttf")
#         font_size = 13 * 2
#         scale = 0.5

# with dpg.font_registry():
#     with dpg.font(font, font_size) as myfont:
#         dpg.add_font_range_hint(dpg.mvFontRangeHint_Default)
#         dpg.add_font_range(0x370, 0x3FF)
# dpg.set_global_font_scale(scale)
# dpg.bind_font(myfont)


def _log(sender, app_data, user_data):
    print(f"sender: {sender}, \t app_data: {app_data}, \t user_data: {user_data}")


# ------------------------------------------------------------------------------

dpg.create_context()
# macos_settings()

# ------------------------------------------------------------------------------

primary_window = dpg.generate_uuid()
#
source_folder = dpg.generate_uuid()
source_folder_status = dpg.generate_uuid()
source_folder_search_pdfs = dpg.generate_uuid()
source_folder_show_pdfs = dpg.generate_uuid()
source_folder_loading_indicator = dpg.generate_uuid()
#
check_add = dpg.generate_uuid()
group_existing_pdf = dpg.generate_uuid()
existing_pdf = dpg.generate_uuid()
existing_pdf_status = dpg.generate_uuid()
#
check_custom_output_folder = dpg.generate_uuid()
output_folder = dpg.generate_uuid()
output_folder_status = dpg.generate_uuid()
#
check_custom_output_filename = dpg.generate_uuid()
output_filename = dpg.generate_uuid()
timestamp_info = dpg.generate_uuid()
#
check_flatten = dpg.generate_uuid()
#
create_pdf = dpg.generate_uuid()
create_pdf_loading_indicator = dpg.generate_uuid()
create_pdf_feedback = dpg.generate_uuid()
#
goto_output_folder = dpg.generate_uuid()
open_pdf = dpg.generate_uuid()

# ------------------------------------------------------------------------------


def changed_source_folder(sender, folder, user_data):
    model.reset_pdfs()
    dpg.hide_item(source_folder_show_pdfs)
    if model.set_source_folder(folder):
        dpg.set_value(source_folder_status, "ok")
        dpg.show_item(source_folder_search_pdfs)
        dpg.set_item_label(source_folder_search_pdfs, label="search")
    else:
        dpg.set_value(source_folder_status, "no valid path")
        dpg.hide_item(source_folder_search_pdfs)


def clicked_source_folder_search_pdfs(sender, app_data, user_data):
    dpg.show_item(source_folder_loading_indicator)
    pdfs = model.find_pdfs()
    if pdfs:
        dpg.set_item_label(source_folder_search_pdfs, label=f"PDFs ({len(pdfs)})")
        if len(pdfs) > 0:
            dpg.show_item(source_folder_show_pdfs)
        else:
            dpg.hide_item(source_folder_show_pdfs)
    dpg.hide_item(source_folder_loading_indicator)


def clicked_source_show_pdfs():
    # TODO popup window
    for p in model.pdfs:
        print(p)


def changed_check_add(sender, checked, user_data):
    if checked:
        model.add = True
        dpg.show_item(group_existing_pdf)
    else:
        model.add = False
        dpg.hide_item(group_existing_pdf)

    update_output_folder()
    update_output_filename()


def changed_existing_pdf(sender, file, user_data):
    if model.set_existing_pdf(file):
        dpg.set_value(existing_pdf_status, value="ok")
    else:
        dpg.set_value(existing_pdf_status, value="no valid PDF file")

    update_output_folder()
    update_output_filename()


def changed_check_custom_output_folder(sender, checked, user_data):
    if checked:
        model.custom_output_folder = True
        dpg.enable_item(output_folder)
    else:
        model.custom_output_folder = False
        dpg.disable_item(output_folder)

    update_output_folder()
    update_output_filename()


def changed_output_folder(sender, folder, user_data):
    if model.set_custom_output_folder(folder):
        dpg.set_value(output_folder_status, value="ok")
    else:
        dpg.set_value(output_folder_status, value="no valid path")


def changed_check_custom_output_filename(sender, checked, user_data):
    if checked:
        model.custom_output_filename = True
        dpg.enable_item(output_filename)
    else:
        model.custom_output_filename = False
        dpg.disable_item(output_filename)

    update_output_folder()
    update_output_filename()


def changed_output_filename(sender, filename, user_data):
    model.set_custom_output_filename(filename)

    update_output_folder()
    update_output_filename()


def update_output_filename():
    dpg.set_value(output_filename, value=model.output_filename)


def update_output_folder():
    dpg.set_value(output_folder, value=model.output_folder)
    if model.output_folder:
        dpg.set_value(output_folder_status, value="ok")
    else:
        dpg.set_value(output_folder_status, value="no valid path")


def clicked_create_pdf():
    from pdfmb import add_from_folder, merge_from_folder

    dpg.show_item(create_pdf_loading_indicator)

    if model.add:
        print(model.source_folder)
        print(model.output_folder)
        print(model.output_filename)

        if (
            not model.source_folder
            or not model.output_folder
            or not model.output_filename
        ):
            print("input fehlt")
            dpg.hide_item(create_pdf_loading_indicator)
            return

        add_from_folder(
            source_folder=model.source_folder,
            existing_pdf=model.existing_pdf,
            output_folder=model.output_folder,
            filename=model.output_filename,
            add_flat_hierachy=model.flatten,
        )

    else:
        print(model.source_folder)
        print(model.output_folder)
        print(model.output_filename)

        if (
            not model.source_folder
            or not model.output_folder
            or not model.output_filename
        ):
            print("input fehlt")
            dpg.hide_item(create_pdf_loading_indicator)
            return

        merge_from_folder(
            source_folder=model.source_folder,
            output_folder=model.output_folder,
            filename=model.output_filename,
            add_flat_hierachy=model.flatten,
        )
    dpg.hide_item(create_pdf_loading_indicator)


def clicked_goto_output_folder():
    if platform.system() == "Darwin":
        call(["open", model.output_folder.absolute()])
    if platform.system() == "Windows":
        import subprocess

        cmd = rf"""explorer {model.output_folder}"""
        subprocess.Popen(cmd)


def clicked_open_pdf():
    if platform.system() == "Darwin":
        pass


def clicked_debug():
    print("")
    print(model)
    print("")


# ------------------------------------------------------------------------------

with dpg.window(tag=primary_window):
    dpg.add_text("Merge PDF files with bookmarks")
    dpg.add_text("")

    # ------------------------------------------------------------------------------

    dpg.add_text("PDF files")
    with dpg.group(horizontal=True):
        dpg.add_input_text(
            tag=source_folder,
            hint="Path to folder",
            callback=changed_source_folder,
        )
        dpg.add_text(
            tag=source_folder_status,
            default_value="no valid path",
        )
        dpg.add_button(
            tag=source_folder_search_pdfs,
            label="search for pdfs",
            width=130,
            show=False,
            callback=clicked_source_folder_search_pdfs,
        )
        dpg.add_button(
            tag=source_folder_show_pdfs,
            label="show pdfs",
            width=130,
            show=False,
            callback=clicked_source_show_pdfs,
        )
        # with dpg.popup(
        #     source_folder_show_pdfs,
        #     modal=True,
        #     mousebutton=dpg.mvMouseButton_Left,
        #     tag="show pdfs popup",
        # ):
        #     with dpg.table(header_row=True, row_background=True):
        #         dpg.add_table_column(label="PDFs")
        #         print(model.pdfs)
        #         for p in model.pdfs:
        #             with dpg.table_row():
        #                 print(p)
        #                 dpg.add_text(p)

        dpg.add_loading_indicator(
            tag=source_folder_loading_indicator,
            radius=1.4,
            style=1,
            color=(150, 255, 150, 255),
            show=False,
        )
    dpg.add_text("")

    # ------------------------------------------------------------------------------

    dpg.add_checkbox(
        tag=check_add,
        label="Add to existing PDF",
        callback=changed_check_add,
    )
    with dpg.group(tag=group_existing_pdf, show=False):
        with dpg.group(horizontal=True):
            dpg.add_input_text(
                tag=existing_pdf,
                hint="Path to existing PDF",
                callback=changed_existing_pdf,
            )
            dpg.add_text(
                tag=existing_pdf_status,
                default_value="no valid pdf file",
            )
    dpg.add_text("")

    # ------------------------------------------------------------------------------

    dpg.add_text("Output")
    dpg.add_checkbox(
        tag=check_custom_output_folder,
        label="Custom folder",
        callback=changed_check_custom_output_folder,
    )
    with dpg.group(horizontal=True):
        dpg.add_input_text(
            tag=output_folder,
            hint="Path to output folder",
            default_value=model.output_folder,
            enabled=False,
            callback=changed_output_folder,
        )
        dpg.add_text(
            "",
            tag=output_folder_status,
        )

    dpg.add_text("")

    dpg.add_checkbox(
        tag=check_custom_output_filename,
        label="Custom filename",
        callback=changed_check_custom_output_filename,
    )
    with dpg.group(horizontal=True):
        dpg.add_input_text(
            tag=output_filename,
            hint="File name",
            default_value=model.output_filename,
            enabled=False,
            callback=changed_output_filename,
        )
        dpg.add_text("- YYYY-MM-DD HHMMSS.pdf")

    dpg.add_text(
        "A unique timestamp is automatically appended to the file name",
        tag=timestamp_info,
        show=False,
        color=(130, 130, 130),
    )
    dpg.add_text("")

    # ------------------------------------------------------------------------------

    dpg.add_checkbox(
        tag=check_flatten,
        label="Flatten folder structure in bookmark outline",
    )
    dpg.add_text("")

    # ------------------------------------------------------------------------------

    with dpg.group(horizontal=True):
        dpg.add_button(
            tag=create_pdf,
            label="Create PDF",
            width=150,
            callback=clicked_create_pdf,
        )
        dpg.add_loading_indicator(
            tag=create_pdf_loading_indicator,
            radius=1.4,
            style=1,
            color=(150, 255, 150, 255),
            show=False,
        )
        # dpg.add_text(
        #     "PDF file created with 123 pages and 12 bookmakrs",
        #     tag=create_pdf_feedback,
        # )

    dpg.add_text("")
    dpg.add_separator()
    dpg.add_button(
        tag=goto_output_folder,
        label="Open output",
        width=150,
        callback=clicked_goto_output_folder,
    )
    dpg.add_button(
        tag=open_pdf,
        label="Open PDF",
        width=150,
        callback=clicked_open_pdf,
    )

    # ------------------------------------------------------------------------------

    # dpg.add_text("")
    # dpg.add_button(
    #     tag="debug",
    #     label="debug",
    #     width=300,
    #     callback=clicked_debug,
    # )

    # demo.show_demo()

dpg.create_viewport(title="pdfmb", width=900, height=550)
dpg.setup_dearpygui()
dpg.show_viewport()
dpg.set_primary_window(primary_window, True)
dpg.start_dearpygui()
dpg.destroy_context()