pip-manager/pmanager.py at main · mixelpixx/pip-manager · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import sys
import os
import subprocess
import json
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QTableWidget, QTableWidgetItem, QPushButton, QLabel, QInputDialog, QMessageBox
from PyQt5.QtCore import Qt
from package_icons import get_package_icon

class PackageManagerTab(QWidget):
    def __init__(self, venv_manager):
        super().__init__()
        self.venv_manager = venv_manager
        layout = QVBoxLayout()
        self.setLayout(layout)

        self.package_table = QTableWidget()
        self.package_table.setColumnCount(3)
        self.package_table.setHorizontalHeaderLabels(["Package", "Version", "Description"])
        self.package_table.horizontalHeader().setStretchLastSection(True)
        layout.addWidget(self.package_table)

        button_layout = QHBoxLayout()
        layout.addLayout(button_layout)

        self.install_button = QPushButton("Install Package")
        self.install_button.clicked.connect(self.install_package)
        button_layout.addWidget(self.install_button)

        self.uninstall_button = QPushButton("Uninstall Package")
        self.uninstall_button.clicked.connect(self.uninstall_package)
        button_layout.addWidget(self.uninstall_button)

        self.update_button = QPushButton("Update Package") #Added Update button
        self.update_button.clicked.connect(self.update_package) # Connected update button to function
        button_layout.addWidget(self.update_button) #Added update button to layout

        self.refresh_button = QPushButton("Refresh List")
        self.refresh_button.clicked.connect(self.refresh_package_list)
        button_layout.addWidget(self.refresh_button)

        self.status_label = QLabel()
        layout.addWidget(self.status_label)

        self.refresh_package_list()

    def refresh_package_list(self):
        self.package_table.setRowCount(0)
        try:
            venv_path = self.venv_manager.get_active_venv_path()
            if venv_path:
                pip_path = os.path.join(venv_path, "Scripts" if sys.platform == "win32" else "bin", "pip")
                result = subprocess.run([pip_path, "list", "--format=json"], capture_output=True, text=True, check=True)
            else:
                result = subprocess.run([sys.executable, "-m", "pip", "list", "--format=json"], capture_output=True, text=True, check=True)
            packages = json.loads(result.stdout)
            self.package_table.setRowCount(len(packages))
            for row, package in enumerate(packages):
                name_item = QTableWidgetItem(package['name'])
                name_item.setIcon(get_package_icon(package['name']))
                self.package_table.setItem(row, 0, name_item)
                self.package_table.setItem(row, 1, QTableWidgetItem(package['version']))
                # You might want to add a description column here if available
            self.package_table.resizeColumnsToContents()
            self.status_label.setText("Package list refreshed")
        except subprocess.CalledProcessError as e:
            self.status_label.setText(f"Error refreshing package list: {e}")

    def uninstall_package(self):
        selected_rows = self.package_table.selectionModel().selectedRows()
        if not selected_rows:
            self.status_label.setText("No package selected")
            return
        package = self.package_table.item(selected_rows[0].row(), 0).text()

        reply = QMessageBox.question(self, 'Uninstall Package',
                                     f"Are you sure you want to uninstall '{package}'?",
                                     QMessageBox.Yes | QMessageBox.No, QMessageBox.No)

        if reply == QMessageBox.Yes:
            try:
                venv_path = self.venv_manager.get_active_venv_path()
                if venv_path:
                    pip_path = os.path.join(venv_path, "Scripts" if sys.platform == "win32" else "bin", "pip")
                    subprocess.run([pip_path, "uninstall", "-y", package], check=True)
                else:
                    subprocess.run([sys.executable, "-m", "pip", "uninstall", "-y", package], check=True)
                self.status_label.setText(f"Uninstalled {package}")
                self.refresh_package_list()
            except subprocess.CalledProcessError as e:
                self.status_label.setText(f"Error uninstalling {package}: {e}")

    def install_package(self):
        package, ok = QInputDialog.getText(self, "Install Package", "Enter package name:")
        if ok and package:
            try:
                venv_path = self.venv_manager.get_active_venv_path()
                if venv_path:
                    pip_path = os.path.join(venv_path, "Scripts" if sys.platform == "win32" else "bin", "pip")
                    subprocess.run([pip_path, "install", package], check=True)
                else:
                    subprocess.run([sys.executable, "-m", "pip", "install", package], check=True)
                self.status_label.setText(f"Installed {package}")
                self.refresh_package_list()
            except subprocess.CalledProcessError as e:
                self.status_label.setText(f"Error installing {package}: {e}")

    def update_package(self):
        selected_rows = self.package_table.selectionModel().selectedRows()
        if not selected_rows:
            self.status_label.setText("No package selected")
            return
        package = self.package_table.item(selected_rows[0].row(), 0).text()

        try:
            venv_path = self.venv_manager.get_active_venv_path()
            if venv_path:
                pip_path = os.path.join(venv_path, "Scripts" if sys.platform == "win32" else "bin", "pip")
                subprocess.run([pip_path, "install", "--upgrade", package], check=True)
            else:
                subprocess.run([sys.executable, "-m", "pip", "install", "--upgrade", package], check=True)
            self.status_label.setText(f"Updated {package}")
            self.refresh_package_list()
        except subprocess.CalledProcessError as e:
            self.status_label.setText(f"Error updating {package}: {e}")