PyCapture/app.py at main · abdullahashfaqvirk/PyCapture · 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
125
126
127
128
129
130
import os
import sys
from PyQt5 import QtWidgets, uic
from PyQt5.QtCore import QTimer


class Screenshots(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        uic.loadUi('app.ui', self)

        self.screenshot_timer = None
        self.stop_timer = None
        self.button1.clicked.connect(self.start)
        self.button2.clicked.connect(self.stop)
        self.counter = 1
        self.setFixedSize(self.size())

    def take_screenshot(self):
        try:
            pixmap = QtWidgets.QApplication.primaryScreen().grabWindow(
                QtWidgets.QApplication.desktop().winId()
            )
            save_path = self.input3.text()
            if not save_path:
                raise ValueError('Save path is empty')
            if not os.path.exists(save_path):
                os.makedirs(save_path)
            save_path += f'/screenshot_{self.counter}.png'
            pixmap.save(save_path)
            self.counter += 1
        except Exception as e:
            self.handle_error('Error taking screenshot', str(e))

    def start(self):
        try:
            save_path = self.input3.text()
            if not save_path:
                raise ValueError('Save path is empty')
            time_interval = self.input1.text()
            if not time_interval.isdigit() or int(time_interval) <= 0:
                raise ValueError('Invalid time interval')
            total_interval = self.input2.text()
            if total_interval.lower() == 'true':
                total_interval = -1
            elif not total_interval.isdigit() or int(total_interval) <= 0:
                raise ValueError('Invalid total interval')
            else:
                total_interval = int(total_interval)
            if not os.path.exists(save_path):
                os.makedirs(save_path)

            self.input1.setEnabled(False)
            self.input2.setEnabled(False)
            self.input3.setEnabled(False)
            self.button1.setEnabled(False)

            start_message = QtWidgets.QMessageBox(self)
            start_message.setStyleSheet('color: white; font-size: 12px;')
            start_message.setIcon(QtWidgets.QMessageBox.Icon.Information)
            start_message.setText('Program is starting...')
            start_message.setWindowTitle('Starting')
            start_message.setStandardButtons(
                QtWidgets.QMessageBox.StandardButton.Ok
            )
            start_message.show()

            self.screenshot_timer = QTimer()
            self.screenshot_timer.timeout.connect(self.take_screenshot)
            self.screenshot_timer.start(int(time_interval) * 1000)

            if total_interval != -1:
                self.stop_timer = QTimer()
                self.stop_timer.setSingleShot(True)
                self.stop_timer.timeout.connect(self.stop)
                self.stop_timer.start(total_interval * 60000)

        except Exception as e:
            error_message = QtWidgets.QMessageBox(self)
            error_message.setStyleSheet('color: white; font-size: 12px;')
            error_message.setIcon(QtWidgets.QMessageBox.Icon.Critical)
            error_message.setText(f'Error starting application: {str(e)}')
            error_message.setWindowTitle('Error')
            error_message.setStandardButtons(
                QtWidgets.QMessageBox.StandardButton.Ok
            )
            error_message.show()

    def stop(self):
        try:
            if self.screenshot_timer is not None:
                self.screenshot_timer.stop()

            if self.stop_timer is not None:
                self.stop_timer.stop()

            self.input1.setEnabled(True)
            self.input2.setEnabled(True)
            self.input3.setEnabled(True)
            self.button1.setEnabled(True)

            end_message = QtWidgets.QMessageBox(self)
            end_message.setStyleSheet('color: white; font-size: 12px;')
            end_message.setIcon(QtWidgets.QMessageBox.Icon.Information)
            end_message.setText('Program has ended.')
            end_message.setWindowTitle('Ending')
            end_message.setStandardButtons(
                QtWidgets.QMessageBox.StandardButton.Ok
            )
            end_message.show()

        except Exception as e:
            error_message = QtWidgets.QMessageBox(self)
            error_message.setStyleSheet('color: white; font-size: 12px;')
            error_message.setIcon(QtWidgets.QMessageBox.Icon.Critical)
            error_message.setText(
                'An error occurred while stopping the application. Please try again or contact support.'
            )
            error_message.setWindowTitle('Error')
            error_message.setStandardButtons(
                QtWidgets.QMessageBox.StandardButton.Ok
            )
            error_message.show()


if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)
    window = Screenshots()
    window.show()
    sys.exit(app.exec())