webScrappingPython/main.py at main · Caderonas/webScrappingPython · 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
import sys
import webScrapping as WS
from PySide6.QtCore import Qt, Slot
from PySide6.QtGui import QAction, QPainter
from PySide6.QtWidgets import (QLabel, QLineEdit, QPushButton, QApplication,
    QGridLayout, QWidget, QMainWindow, QTableWidgetItem, QTableWidget, QHeaderView)

class Widget(QWidget):
    def __init__(self):
        QWidget.__init__(self)
        self.items = 0

        # Create widgets
        self.title = QLabel("WebScrapping App")
        self.title.setStyleSheet("color: black; font-size: 30pt; text-align: right")
        self.label = QLabel("Search :")
        self.research = QLineEdit()
        self.button = QPushButton("Enter")
        self.button.setShortcut("Return")

        self.table = QTableWidget()
        self.table.setColumnCount(5)
        self.table.setHorizontalHeaderLabels(["Date", "Name", "Size", "Ratio", "Link"])
        self.table.horizontalHeader().setSectionResizeMode(QHeaderView.ResizeToContents)

        # Create layout and add widgets
        layout = QGridLayout()
        layout.addWidget(self.title, 0 ,0)
        layout.addWidget(self.label, 1, 0)
        layout.addWidget(self.research, 1, 1)
        layout.addWidget(self.button, 1, 2)
        layout.addWidget(self.table, 2, 0, 1, 3)
        # Set dialog layout
        self.setLayout(layout)
        # Add button signal to greetings slot
        self.button.clicked.connect(self.run_search)

    @Slot()
    def run_search(self):
        print ("Running search")
        self.add_element()
        print ("Table print")

    def add_element(self):
        self.items = 0
        self.torrents = WS.WebScrapping(self.research.text()).get_result()
        try:
            for torrent in self.torrents:
                date_item = QTableWidgetItem(torrent.date)
                date_item.setTextAlignment(Qt.AlignLeft)

                name_item = QTableWidgetItem(torrent.name)
                name_item.setTextAlignment(Qt.AlignRight)

                size_item = QTableWidgetItem(str(torrent.size)+ "Gb")
                size_item.setTextAlignment(Qt.AlignLeft)

                ratio_item = QTableWidgetItem(str(torrent.seed) +"/"+ str(torrent.leech))
                ratio_item.setTextAlignment(Qt.AlignLeft)

                button_item = QPushButton("Stream")


                self.table.insertRow(self.items)

                self.table.setItem(self.items, 0, date_item)
                self.table.setItem(self.items, 1, name_item)
                self.table.setItem(self.items, 2, size_item)
                self.table.setItem(self.items, 3, ratio_item)
                self.table.setCellWidget(self.items, 4, button_item)
                self.table.horizontalHeader().setSectionResizeMode(QHeaderView.ResizeToContents)
                button_item.clicked.connect(SimpleTorrentStreaming  (torrent.get_magnet))
                self.items += 1
        except ValueError:
            print("Wrong search ", self.research)

class MainWindow(QMainWindow):
    def __init__(self, widget):
        QMainWindow.__init__(self)
        self.setWindowTitle("WebScrapping App")

        # Menu
        self.menu = self.menuBar()
        self.file_menu = self.menu.addMenu("File")

        # Exit QAction
        exit_action = QAction("Exit", self)
        exit_action.setShortcut("Ctrl+Q")
        exit_action.triggered.connect(self.exit_app)

        self.file_menu.addAction(exit_action)
        self.setCentralWidget(widget)

    @Slot()
    def exit_app(self):
            QApplication.quit()


if __name__ == "__main__":
    # Qt Application
    app = QApplication(sys.argv)
    # QWidget
    widget = Widget()
    # QMainWindow using QWidget as central widget
    window = MainWindow(widget)
    window.resize(1200, 800)
    window.show()

    # Execute application
    sys.exit(app.exec())