org-editor/main.py

#!/usr/bin/env python3

import logging
import os
import sys
import time
import webbrowser

from PySide2.QtCore import QObject, QThread, Signal, Slot

from PySide2.QtGui import QPalette, QColor

from PySide2.QtWidgets import (
    QApplication,
    QDialog,
    QFrame,
    QGroupBox,
    QHBoxLayout,
    QLabel,
    QLineEdit,
    QProgressBar,
    QPushButton,
    QScrollArea,
    QScroller,
    QTabBar,
    QVBoxLayout,
)

import doc_manager

DOCS_PATH = os.environ["ORG_PATH"]
MAX_SEARCH_NODES = 100


class LoadDoneSignal(QObject):
    sig = Signal(doc_manager.DocumentManager)


class DocumentLoader(QThread):
    def __init__(self, manager):
        QThread.__init__(self, None)
        self.manager = manager
        self.signal = LoadDoneSignal()

    def run(self):
        self.manager.load()
        self.signal.sig.emit(self.manager)


class Dialog(QDialog):
    def __init__(self):
        super(Dialog, self).__init__()

        palette = self.palette()
        palette.setColor(QPalette.Window, QColor(255, 255, 255))
        self.setPalette(palette)
        self.setAutoFillBackground(True)

        self.setWindowTitle("OrgEditor")
        scrSize = self.screen().size()
        self.resize(scrSize.width() / 1.5, scrSize.height() / 1.5)
        self.loader = None
        self.manager = doc_manager.DocumentManager(DOCS_PATH)

        layout = QVBoxLayout()

        # Edit box
        self.progressBar = QProgressBar()
        self.progressBar.setRange(0, 0)  # Make undetermined
        layout.addWidget(self.progressBar)

        self.edit = QLineEdit("", placeholderText="Search for notes")
        self.edit.textEdited.connect(self.on_text_edited)
        layout.addWidget(self.edit)

        layout.setSpacing(0)

        self.results = QScrollArea(widgetResizable=True)
        layout.addWidget(self.results)
        QScroller.grabGesture(
            self.results.viewport(),
            QScroller.LeftMouseButtonGesture,
        )

        # Options
        self.tabBar = QTabBar(shape=QTabBar.RoundedSouth)

        self.tabBar.addTab("Agenda")
        self.tabBar.addTab("Notes")
        self.tabBar.addTab("Tasks")
        self.tabBar.currentChanged.connect(self.update_tab)

        layout.addWidget(self.tabBar)

        self.setLayout(layout)
        self.startLoad()

    @Slot()
    def on_text_edited(self):
        if self.tabBar.currentIndex() != 1:
            self.tabBar.setCurrentIndex(1)
        else:
            self.loadNotes()

    @Slot()
    def update_tab(self):
        tabIndex = self.tabBar.currentIndex()
        if tabIndex == 0:
            self.loadAgenda()
        elif tabIndex == 1:
            self.loadNotes()
        elif tabIndex == 2:
            self.loadTasks()

    def startLoad(self):
        self.edit.setDisabled(True)
        self.edit.setVisible(False)
        self.tabBar.setDisabled(True)
        self.progressBar.setVisible(True)

        self.loader = DocumentLoader(self.manager)
        self.loader.signal.sig.connect(self.longoperationcomplete)
        self.loading_start_time = time.time()
        self.loader.start()

    def endLoad(self):
        self.edit.setDisabled(False)
        self.edit.setVisible(True)
        self.tabBar.setDisabled(False)
        self.progressBar.setVisible(False)

        self.update_tab()

    def longoperationcomplete(self, data):
        logging.info(
            "Loading complete in {:.3f}s".format(time.time() - self.loading_start_time)
        )
        self.endLoad()

    def loadAgenda(self):
        agenda = self.manager.get_agenda()
        old = self.results.layout()

        if old:
            print("Deleting old")
            old.deleteLater()

        layout = QVBoxLayout()

        for item in agenda.with_hour:
            layout.addWidget(self.build_agenda_task_widget(item))

        # if len(agenda.with_hour) > 0 and len(agenda.no_hour) > 0:
        #     layout.addWidget(QSplitter())

        for item in agenda.no_hour:
            layout.addWidget(self.build_agenda_task_widget(item))

        layout.addStretch()

        frame = QFrame(self.results)
        frame.setLayout(layout)
        self.results.setWidget(frame)

    def build_agenda_task_widget(self, item):
        box = QHBoxLayout()
        frame = QFrame()
        frame.setLayout(box)

        state_button = QPushButton(text=f"{item.state or '-'}", maximumWidth=60)
        if item.is_done:
            state_button.setFlat(True)
        box.addWidget(state_button)

        box.addWidget(QLabel(text=f"{item.scheduled.time}", maximumWidth=200))
        box.addWidget(QLabel(text=f"{item.title}"))

        def on_clicked():
            state_button.setText("DONE")
            # state_button.setFlat(True)
            # item.state = 'DONE'

        if not item.is_done:
            state_button.clicked.connect(on_clicked)

        return frame

    def build_note_task_widget(self, item):
        box = QHBoxLayout()
        frame = QGroupBox()
        frame.setLayout(box)

        titleButton = QPushButton(text=f"{item.title}")
        box.addWidget(titleButton)

        def on_clicked():
            webbrowser.open("org-protocol://org-id?id=" + item.id)

        titleButton.clicked.connect(on_clicked)

        return frame

    def loadNotes(self):
        query = self.edit.text()
        notes = self.manager.get_notes(query.split())
        old = self.results.layout()

        if old:
            print("Deleting old")
            old.deleteLater()

        layout = QVBoxLayout()

        for note in notes[:MAX_SEARCH_NODES]:
            layout.addWidget(self.build_note_task_widget(note))

        layout.addStretch()

        frame = QFrame(self.results)
        frame.setLayout(layout)
        self.results.setWidget(frame)

    def loadTasks(self):
        logging.warning("loadTasks not yet implemented")


# Create the Qt Application
if __name__ == "__main__":
    logging.basicConfig(level=logging.INFO, format="%(levelname)-8s %(message)s")

    app = QApplication(sys.argv)

    dialog = Dialog()
    sys.exit(dialog.exec_())