main.py

import sys
from crontab import CronTab

from PyQt6.QtWidgets import (
    QApplication, QWidget, QVBoxLayout, QPushButton, QListWidget,
    QInputDialog, QMessageBox, QHBoxLayout
)


# ---------------------- Constants ----------------------
WINDOW_TITLE = "Linux Cron Manager (PyQt6)"
WINDOW_WIDTH = 600
WINDOW_HEIGHT = 400


# ---------------------- Main Class ----------------------
class CronManager(QWidget):
    def __init__(self):
        super().__init__()
        self.cron = CronTab(user=True)

        self.setup_window()
        self.setup_ui()
        self.refresh_jobs()

    # ---------- Window Setup ----------
    def setup_window(self):
        self.setWindowTitle(WINDOW_TITLE)
        self.setFixedSize(WINDOW_WIDTH, WINDOW_HEIGHT)

    # ---------- UI Setup ----------
    def setup_ui(self):
        main_layout = QVBoxLayout()
        btn_layout = QHBoxLayout()

        # Job List
        self.listWidget = QListWidget()
        main_layout.addWidget(self.listWidget)

        # Buttons
        self.create_button("Refresh", self.refresh_jobs, btn_layout)
        self.create_button("Add Job", self.add_job, btn_layout)
        self.create_button("Delete Job", self.delete_job, btn_layout)

        main_layout.addLayout(btn_layout)
        self.setLayout(main_layout)

    # Helper to create buttons
    def create_button(self, text, action, layout):
        btn = QPushButton(text)
        btn.clicked.connect(action)
        layout.addWidget(btn)

    # ---------- Cron Functions ----------
    def refresh_jobs(self):
        """Reload job list from user crontab."""
        self.listWidget.clear()
        for job in self.cron:
            self.listWidget.addItem(job.sentence)

    def add_job(self):
        """Add a new cron job."""
        schedule = self.get_input("Schedule", "Cron format: */5 * * * *")
        if not schedule:
            return

        command = self.get_input("Command", "Command to execute:")
        if not command:
            return

        try:
            job = self.cron.new(command=command)
            job.setall(schedule)
            self.cron.write()
            self.refresh_jobs()
        except Exception as e:
            self.show_error(str(e))

    def delete_job(self):
        """Delete selected cron job."""
        index = self.listWidget.currentRow()
        if index < 0:
            return self.show_warning("Please select a job to delete.")

        if not self.confirm("Delete this job?"):
            return

        try:
            for i, job in enumerate(self.cron):
                if i == index:
                    self.cron.remove(job)
                    self.cron.write()
                    break
            self.refresh_jobs()
        except Exception as e:
            self.show_error(str(e))

    # ---------- Dialog Helpers ----------
    def get_input(self, title, message):
        value, ok = QInputDialog.getText(self, title, message)
        return value.strip() if ok and value.strip() else None

    def show_error(self, message):
        QMessageBox.critical(self, "Error", message)

    def show_warning(self, message):
        QMessageBox.warning(self, "Warning", message)

    def confirm(self, message):
        result = QMessageBox.question(
            self, "Confirm", message,
            QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No
        )
        return result == QMessageBox.StandardButton.Yes


# ---------------------- App Runner ----------------------
if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = CronManager()
    window.show()
    sys.exit(app.exec())