service_manager_pyqt.py

import sys
import subprocess
from PyQt5.QtWidgets import (
    QApplication, QWidget, QVBoxLayout, QTableWidget, QTableWidgetItem, 
    QPushButton, QMessageBox, QHeaderView, QLineEdit, QLabel, QHBoxLayout, QCheckBox
)
from PyQt5.QtCore import Qt, QSize, QSortFilterProxyModel
from PyQt5.QtGui import QPalette, QColor
import qdarkstyle

class ServiceManagerApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setWindowTitle('Service Manager')
        self.resize(1200, 800)

        main_layout = QVBoxLayout()

        # Search bar
        search_layout = QHBoxLayout()
        search_label = QLabel("Search Service:")
        self.search_input = QLineEdit()
        self.search_input.textChanged.connect(self.filter_services)
        search_layout.addWidget(search_label)
        search_layout.addWidget(self.search_input)

        # Dark mode toggle
        self.dark_mode_check = QCheckBox("Dark Mode")
        self.dark_mode_check.stateChanged.connect(self.toggle_dark_mode)
        search_layout.addWidget(self.dark_mode_check)

        main_layout.addLayout(search_layout)

        # Table for services
        self.table = QTableWidget()
        main_layout.addWidget(self.table)

        self.setLayout(main_layout)

        self.populate_table()
        self.table.horizontalHeader().sectionClicked.connect(self.sort_table)
        self.sort_order = Qt.AscendingOrder

    def populate_table(self):
        services = self.get_services()

        self.table.setRowCount(len(services))
        self.table.setColumnCount(5)
        self.table.setHorizontalHeaderLabels(['Service', 'Description', 'Status', 'Start/Stop', 'Enable/Disable'])

        for row, service in enumerate(services):
            service_name, description, status = service

            self.table.setItem(row, 0, QTableWidgetItem(service_name))
            self.table.setItem(row, 1, QTableWidgetItem(description))
            self.table.setItem(row, 2, QTableWidgetItem(status))

            start_stop_btn = QPushButton('Stop' if 'running' in status else 'Start')
            start_stop_btn.setCheckable(True)
            start_stop_btn.setChecked('running' in status)
            start_stop_btn.setMinimumSize(QSize(100, 30))
            start_stop_btn.clicked.connect(lambda checked, s=service_name: self.toggle_service(checked, s, 'start', 'stop'))
            self.table.setCellWidget(row, 3, start_stop_btn)

            enable_disable_btn = QPushButton('Disable' if 'enabled' in status else 'Enable')
            enable_disable_btn.setCheckable(True)
            enable_disable_btn.setChecked('enabled' in status)
            enable_disable_btn.setMinimumSize(QSize(100, 30))
            enable_disable_btn.clicked.connect(lambda checked, s=service_name: self.toggle_service(checked, s, 'enable', 'disable'))
            self.table.setCellWidget(row, 4, enable_disable_btn)

        self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.table.resizeColumnsToContents()
        self.table.resizeRowsToContents()
        
        # Apply style for better appearance
        self.apply_table_style()

    def get_services(self):
        command = ['systemctl', 'list-units', '--type=service', '--all', '--no-pager']
        result = subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)

        services = []
        lines = result.stdout.splitlines()
        for line in lines[1:]:
            parts = line.split(maxsplit=4)
            if len(parts) < 5:
                continue
            service_name = parts[0]
            status = parts[3]
            description = parts[4] if len(parts) > 4 else ""
            services.append((service_name, description, status))
        return services

    def toggle_service(self, checked, service_name, action_enable, action_disable):
        action = action_enable if checked else action_disable
        result = self.run_command(['sudo', 'systemctl', action, service_name])
        if result:
            self.populate_table()

    def run_command(self, command):
        try:
            result = subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
            self.show_message(result.stdout)
            return True
        except subprocess.CalledProcessError as e:
            self.show_message(e.stderr)
            return False

    def show_message(self, message):
        msg_box = QMessageBox(self)
        msg_box.setText(message)
        msg_box.exec_()

    def filter_services(self):
        filter_text = self.search_input.text().lower()
        for row in range(self.table.rowCount()):
            service_name = self.table.item(row, 0).text().lower()
            self.table.setRowHidden(row, filter_text not in service_name)

    def sort_table(self, index):
        self.table.sortItems(index, self.sort_order)
        self.sort_order = Qt.DescendingOrder if self.sort_order == Qt.AscendingOrder else Qt.AscendingOrder

    def toggle_dark_mode(self, state):
        if state == Qt.Checked:
            self.apply_dark_mode()
        else:
            self.apply_light_mode()

    def apply_dark_mode(self):
        app.setStyleSheet(qdarkstyle.load_stylesheet_pyqt5())

    def apply_light_mode(self):
        app.setStyleSheet("")

    def apply_table_style(self):
        # Set alternating row colors
        self.table.setAlternatingRowColors(True)
        self.table.setStyleSheet(
            "QTableWidget::item { padding: 10px; }"
            "QHeaderView::section { background-color: #404040; color: white; }"
            "QTableWidget { gridline-color: #404040; }"
            "QTableWidget::item:selected { background-color: #565656; }"
        )
        self.table.setShowGrid(True)
        self.table.setGridStyle(Qt.SolidLine)

if __name__ == '__main__':
    app = QApplication(sys.argv)
    
    ex = ServiceManagerApp()
    
    # Save original palette for light mode
    ex.original_palette = app.palette()
    
    ex.show()
    sys.exit(app.exec_())