def slot_change_notifications_state(self):
if (self.__alertWindow.notificationsEnabled):
self.__alertWindow.notificationsEnabled = False
self.__alertWindow.quit()
else:
self.__alertWindow.notificationsEnabled = True
self.__alertWindow = Afficher(
self.__favouriteSeries, self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
def slot_clear_favourites(self):
self.__favouriteSeries = []
self.__favouritesIDList = []
self.__favouritesWidget.clear()
self.__alertWindow.quit()
self.__alertWindow = Afficher(self.__favouriteSeries,
self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
# Clear favourites file
with open(self.__fileName, "wb") as favFile:
pickler = pickle.Pickler(favFile)
pickler.dump(
self.__favouriteSeries) # Saving refreshed favourites list
def slot_magic_add_to_favourites(self):
serie = findForthcomingSerie(self.__favouritesIDList)
id = serie.id
self.__favouritesIDList += [id]
nm = serie.name
self.__favouriteSeries += [serie]
self.__favouritesWidget.addItem(nm)
self.__alertWindow.quit()
self.__alertWindow = Afficher(self.__favouriteSeries,
self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
with open(self.__fileName, "wb") as favFile:
pickler = pickle.Pickler(favFile)
pickler.dump(
self.__favouriteSeries) # Saving refreshed favourites list
def slot_add_to_favourites(self,
id): #id = id of the serie to add to favourites
if (id not in self.__favouritesIDList
): #Check if the serie is already in the favourite
self.__favouritesIDList += [id]
serie = searchSerie(id)
nm = serie.name
self.__favouriteSeries += [serie]
self.__favouritesWidget.addItem(nm)
self.__alertWindow.quit()
self.__alertWindow = Afficher(
self.__favouriteSeries, self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
with open(self.__fileName, "wb") as favFile:
pickler = pickle.Pickler(favFile)
pickler.dump(
self.__favouriteSeries) # Saving refreshed favourites list
else: # If the serie is already in the favourites, displaying an error message
error_dialog = QMessageBox.information(None, "Error",
"Favourite already added.",
QMessageBox.Cancel)
def slot_remove_favourite(self):
idx = self.__favouritesWidget.currentRow(
) # The index of the selected row
if (
idx == -1
): # Exception if the user didn't select a favourite before clicking "remove favourite" button
QMessageBox.information(None, "Error",
"You didn't select a favourite.",
QMessageBox.Ok)
else:
del self.__favouriteSeries[
idx] # The favourites widget list and the inner user's favourites list are sorted in the same order
del self.__favouritesIDList[idx]
self.__favouritesWidget.takeItem(idx)
self.__alertWindow.quit()
self.__alertWindow = Afficher(
self.__favouriteSeries, self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
with open(self.__fileName, "wb") as favFile:
pickler = pickle.Pickler(favFile)
pickler.dump(
self.__favouriteSeries) # Saving refreshed favourites list
def __init__(self, series_list, nameWindow):
# series_list = list of series to be displayed and retrieved from the API,
# nameWindow = name to be displayed on top of the indow
super().__init__()
self.__nDisplay = len(series_list) # Number of series displayed
self.__seriesList = series_list # List of series displayed at first
# Load .ui designed on Qt
self.__UI = uic.loadUi('main.ui', self)
# Show window on desktop
self.showMaximized()
# Name of the window
self.__textLabel = QLabel(nameWindow)
self.__textLabel.setText(nameWindow)
self.__textLabel.setTextFormat(QtCore.Qt.RichText)
self.__textLabel.setText(
"<span style=' font-size:16pt; font-weight:600; color:#aa0000;'>" +
nameWindow + "</span>")
self.__UI.horizontalLayout.addWidget(self.__textLabel)
# Define a Scroll area for serie display
self.__serieWind = QWidget() # Create a widget for the scroll area
self.__scrollArea = QScrollArea() # I Create a Scroll Area
self.__scrollArea.setWidgetResizable(True)
self.__scrollArea.setWidget(
self.__serieWind) # Insert the scroll area in the widget
self.__gridLayout = QGridLayout(
) # Create a grid layout for the scroll area
self.__serieWind.setLayout(
self.__gridLayout) # Insert the grid layout in the scroll area
self.__UI.horizontalLayout_2.addWidget(
self.__scrollArea
) # Insert the scroll area in the main horizontal layout from .ui
self.__serieWind.setObjectName("serieWind")
self.setStyleSheet("#serieWind{background-color: black;}"
) # Define color of the scroll area
# Add research bar
self.__searchWidget = QLineEdit()
self.__searchWidget.setFixedSize(150, 40)
self.__searchWidget.returnPressed.connect(
self.slot_research
) # Connect the signal return pressed to slot_research
self.__UI.horizontalLayout.addWidget(
self.__searchWidget) # Insert research bar in layout from .ui
# Add research button
self.__researchButton = QPushButton("Search")
self.__researchButton.setFixedSize(100, 40)
self.__researchButton.pressed.connect(
self.slot_research) # Connect the signal pressed to slot_research
self.__UI.horizontalLayout.addWidget(
self.__researchButton
) # Insert research button in layout from .ui # Add favourites title
# Favourites Layout
self.__favLayout = QVBoxLayout()
self.__UI.horizontalLayout_2.addLayout(self.__favLayout)
# Favourites Title
self.__favouritesTitle = QLabel("Favourites")
self.__favouritesTitle.setText(
"<span style=' font-size:16pt; font-weight:600; color:#aa0000;'> Favourites </span>"
)
self.__favLayout.addWidget(self.__favouritesTitle)
# Add favourites list to the layout with a QListWidget
self.__favouritesWidget = QListWidget()
self.__favouritesWidget.setMaximumWidth(250)
self.__favLayout.addWidget(self.__favouritesWidget)
self.__favButtonsTopLayout = QHBoxLayout()
self.__favButtonsBottomLayout = QHBoxLayout()
self.__favLayout.addLayout(self.__favButtonsTopLayout)
self.__favLayout.addLayout(self.__favButtonsBottomLayout)
# Add More Info button for favourites list
self.__favMoreInfoButton = QPushButton("More Info")
self.__favButtonsTopLayout.addWidget(self.__favMoreInfoButton)
self.__favMoreInfoButton.clicked.connect(
self.slot_open_serie_window
) # Connect clicked signal of MoreInfo button to slot_open_serie_window
# Add Remove favourite button for favourites list
self.__removeFavButton = QPushButton("Remove Favourite")
self.__favButtonsTopLayout.addWidget(self.__removeFavButton)
self.__removeFavButton.clicked.connect(
self.slot_remove_favourite
) # Connect clicked signal to Remove button to slot_remove_favourites
# Magic Button : finds a serie that has an episode that is going to be aired soon
self.__magicButton = QPushButton("MAGIC BUTTON")
self.__magicButton.clicked.connect(self.slot_magic_add_to_favourites)
self.__favButtonsBottomLayout.addWidget(self.__magicButton)
# Clear Favourites Button
self.__clearFavButton = QPushButton("Clear Favourites")
self.__favButtonsBottomLayout.addWidget(self.__clearFavButton)
self.__clearFavButton.clicked.connect(self.slot_clear_favourites)
# Notifications checkbox
self.__enableNotifications = QCheckBox()
self.__enableNotifications.setText("Enable Notifications")
self.__enableNotifications.setChecked(True)
self.__enableNotifications.stateChanged.connect(
self.slot_change_notifications_state)
self.__favLayout.addWidget(self.__enableNotifications)
#Create and load favourites list
self.__favouritesIDList = [] # Creation of a ID list of favourites
self.__favouriteSeries = [
] # Creation of list of favourites of class Serie
self.__fileName = "favourites" #Creation of a file for pickler
if (os.path.exists(self.__fileName)) and (os.path.getsize(
self.__fileName) > 0): #Check if the file exists
with open(self.__fileName, "rb") as favFile:
depickler = pickle.Unpickler(favFile)
self.__favouriteSeries = depickler.load(
) #Loading previously saved favourites from file
for i in range(
len(self.__favouriteSeries)
): #Loop to add favourites series to favourite QWidgetList and create favouritesIDList
favItem = self.__favouriteSeries[i].name
self.__favouritesWidget.addItem(favItem)
self.__favouritesIDList += [self.__favouriteSeries[i].id]
#Alert
self.__alertWindow = Afficher(self.__favouriteSeries,
self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
# Signal Mapper to connect slot_add_to_favourites to class MainWidget
self.__sigMapper = QSignalMapper(self)
self.__sigMapper.mapped.connect(self.slot_add_to_favourites)
# Display series MainWidgets on MainWindow
self.__numberSeriesWidgetLines = ceil(self.__nDisplay /
5) # 5 widgets per line maximum
self.__positions = [(i + 1, j)
for i in range(self.__numberSeriesWidgetLines)
for j in range(5)
] # Define positions for the grid layout
self.__seriesWidgetList = []
for i in range(len(self.__seriesList)
): #Loop for creation and display of serie MainWidgets
currentWidget = MainWidget(i, self.__seriesList[i])
self.__seriesWidgetList += [currentWidget]
self.__gridLayout.addWidget(currentWidget, *self.__positions[i])
i += 1
self.__sigMapper.setMapping(currentWidget.favButton,
currentWidget.id)
currentWidget.favButton.clicked.connect(
self.__sigMapper.map
) #Connect add to favourite button of MainWidget to signal mapper
class MainWindow(QMainWindow): #Main window of the Serie Browser
def __init__(self, series_list, nameWindow):
# series_list = list of series to be displayed and retrieved from the API,
# nameWindow = name to be displayed on top of the indow
super().__init__()
self.__nDisplay = len(series_list) # Number of series displayed
self.__seriesList = series_list # List of series displayed at first
# Load .ui designed on Qt
self.__UI = uic.loadUi('main.ui', self)
# Show window on desktop
self.showMaximized()
# Name of the window
self.__textLabel = QLabel(nameWindow)
self.__textLabel.setText(nameWindow)
self.__textLabel.setTextFormat(QtCore.Qt.RichText)
self.__textLabel.setText(
"<span style=' font-size:16pt; font-weight:600; color:#aa0000;'>" +
nameWindow + "</span>")
self.__UI.horizontalLayout.addWidget(self.__textLabel)
# Define a Scroll area for serie display
self.__serieWind = QWidget() # Create a widget for the scroll area
self.__scrollArea = QScrollArea() # I Create a Scroll Area
self.__scrollArea.setWidgetResizable(True)
self.__scrollArea.setWidget(
self.__serieWind) # Insert the scroll area in the widget
self.__gridLayout = QGridLayout(
) # Create a grid layout for the scroll area
self.__serieWind.setLayout(
self.__gridLayout) # Insert the grid layout in the scroll area
self.__UI.horizontalLayout_2.addWidget(
self.__scrollArea
) # Insert the scroll area in the main horizontal layout from .ui
self.__serieWind.setObjectName("serieWind")
self.setStyleSheet("#serieWind{background-color: black;}"
) # Define color of the scroll area
# Add research bar
self.__searchWidget = QLineEdit()
self.__searchWidget.setFixedSize(150, 40)
self.__searchWidget.returnPressed.connect(
self.slot_research
) # Connect the signal return pressed to slot_research
self.__UI.horizontalLayout.addWidget(
self.__searchWidget) # Insert research bar in layout from .ui
# Add research button
self.__researchButton = QPushButton("Search")
self.__researchButton.setFixedSize(100, 40)
self.__researchButton.pressed.connect(
self.slot_research) # Connect the signal pressed to slot_research
self.__UI.horizontalLayout.addWidget(
self.__researchButton
) # Insert research button in layout from .ui # Add favourites title
# Favourites Layout
self.__favLayout = QVBoxLayout()
self.__UI.horizontalLayout_2.addLayout(self.__favLayout)
# Favourites Title
self.__favouritesTitle = QLabel("Favourites")
self.__favouritesTitle.setText(
"<span style=' font-size:16pt; font-weight:600; color:#aa0000;'> Favourites </span>"
)
self.__favLayout.addWidget(self.__favouritesTitle)
# Add favourites list to the layout with a QListWidget
self.__favouritesWidget = QListWidget()
self.__favouritesWidget.setMaximumWidth(250)
self.__favLayout.addWidget(self.__favouritesWidget)
self.__favButtonsTopLayout = QHBoxLayout()
self.__favButtonsBottomLayout = QHBoxLayout()
self.__favLayout.addLayout(self.__favButtonsTopLayout)
self.__favLayout.addLayout(self.__favButtonsBottomLayout)
# Add More Info button for favourites list
self.__favMoreInfoButton = QPushButton("More Info")
self.__favButtonsTopLayout.addWidget(self.__favMoreInfoButton)
self.__favMoreInfoButton.clicked.connect(
self.slot_open_serie_window
) # Connect clicked signal of MoreInfo button to slot_open_serie_window
# Add Remove favourite button for favourites list
self.__removeFavButton = QPushButton("Remove Favourite")
self.__favButtonsTopLayout.addWidget(self.__removeFavButton)
self.__removeFavButton.clicked.connect(
self.slot_remove_favourite
) # Connect clicked signal to Remove button to slot_remove_favourites
# Magic Button : finds a serie that has an episode that is going to be aired soon
self.__magicButton = QPushButton("MAGIC BUTTON")
self.__magicButton.clicked.connect(self.slot_magic_add_to_favourites)
self.__favButtonsBottomLayout.addWidget(self.__magicButton)
# Clear Favourites Button
self.__clearFavButton = QPushButton("Clear Favourites")
self.__favButtonsBottomLayout.addWidget(self.__clearFavButton)
self.__clearFavButton.clicked.connect(self.slot_clear_favourites)
# Notifications checkbox
self.__enableNotifications = QCheckBox()
self.__enableNotifications.setText("Enable Notifications")
self.__enableNotifications.setChecked(True)
self.__enableNotifications.stateChanged.connect(
self.slot_change_notifications_state)
self.__favLayout.addWidget(self.__enableNotifications)
#Create and load favourites list
self.__favouritesIDList = [] # Creation of a ID list of favourites
self.__favouriteSeries = [
] # Creation of list of favourites of class Serie
self.__fileName = "favourites" #Creation of a file for pickler
if (os.path.exists(self.__fileName)) and (os.path.getsize(
self.__fileName) > 0): #Check if the file exists
with open(self.__fileName, "rb") as favFile:
depickler = pickle.Unpickler(favFile)
self.__favouriteSeries = depickler.load(
) #Loading previously saved favourites from file
for i in range(
len(self.__favouriteSeries)
): #Loop to add favourites series to favourite QWidgetList and create favouritesIDList
favItem = self.__favouriteSeries[i].name
self.__favouritesWidget.addItem(favItem)
self.__favouritesIDList += [self.__favouriteSeries[i].id]
#Alert
self.__alertWindow = Afficher(self.__favouriteSeries,
self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
# Signal Mapper to connect slot_add_to_favourites to class MainWidget
self.__sigMapper = QSignalMapper(self)
self.__sigMapper.mapped.connect(self.slot_add_to_favourites)
# Display series MainWidgets on MainWindow
self.__numberSeriesWidgetLines = ceil(self.__nDisplay /
5) # 5 widgets per line maximum
self.__positions = [(i + 1, j)
for i in range(self.__numberSeriesWidgetLines)
for j in range(5)
] # Define positions for the grid layout
self.__seriesWidgetList = []
for i in range(len(self.__seriesList)
): #Loop for creation and display of serie MainWidgets
currentWidget = MainWidget(i, self.__seriesList[i])
self.__seriesWidgetList += [currentWidget]
self.__gridLayout.addWidget(currentWidget, *self.__positions[i])
i += 1
self.__sigMapper.setMapping(currentWidget.favButton,
currentWidget.id)
currentWidget.favButton.clicked.connect(
self.__sigMapper.map
) #Connect add to favourite button of MainWidget to signal mapper
# Getters and setters for seriesList
@property
def seriesList(self):
return self.__seriesList
@seriesList.setter
def seriesList(self, newSeriesList):
self.__seriesList = newSeriesList
# Methods
# Slot to add a favourite to the QListWidget
def slot_add_to_favourites(self,
id): #id = id of the serie to add to favourites
if (id not in self.__favouritesIDList
): #Check if the serie is already in the favourite
self.__favouritesIDList += [id]
serie = searchSerie(id)
nm = serie.name
self.__favouriteSeries += [serie]
self.__favouritesWidget.addItem(nm)
self.__alertWindow.quit()
self.__alertWindow = Afficher(
self.__favouriteSeries, self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
with open(self.__fileName, "wb") as favFile:
pickler = pickle.Pickler(favFile)
pickler.dump(
self.__favouriteSeries) # Saving refreshed favourites list
else: # If the serie is already in the favourites, displaying an error message
error_dialog = QMessageBox.information(None, "Error",
"Favourite already added.",
QMessageBox.Cancel)
# The slot that adds a serie that has a soon-to-be-aired episode to favourites
def slot_magic_add_to_favourites(self):
serie = findForthcomingSerie(self.__favouritesIDList)
id = serie.id
self.__favouritesIDList += [id]
nm = serie.name
self.__favouriteSeries += [serie]
self.__favouritesWidget.addItem(nm)
self.__alertWindow.quit()
self.__alertWindow = Afficher(self.__favouriteSeries,
self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
with open(self.__fileName, "wb") as favFile:
pickler = pickle.Pickler(favFile)
pickler.dump(
self.__favouriteSeries) # Saving refreshed favourites list
# Slot to remove a serie from user's favourites list
def slot_remove_favourite(self):
idx = self.__favouritesWidget.currentRow(
) # The index of the selected row
if (
idx == -1
): # Exception if the user didn't select a favourite before clicking "remove favourite" button
QMessageBox.information(None, "Error",
"You didn't select a favourite.",
QMessageBox.Ok)
else:
del self.__favouriteSeries[
idx] # The favourites widget list and the inner user's favourites list are sorted in the same order
del self.__favouritesIDList[idx]
self.__favouritesWidget.takeItem(idx)
self.__alertWindow.quit()
self.__alertWindow = Afficher(
self.__favouriteSeries, self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
with open(self.__fileName, "wb") as favFile:
pickler = pickle.Pickler(favFile)
pickler.dump(
self.__favouriteSeries) # Saving refreshed favourites list
# Slot that clears user's favourites list
def slot_clear_favourites(self):
self.__favouriteSeries = []
self.__favouritesIDList = []
self.__favouritesWidget.clear()
self.__alertWindow.quit()
self.__alertWindow = Afficher(self.__favouriteSeries,
self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()
# Clear favourites file
with open(self.__fileName, "wb") as favFile:
pickler = pickle.Pickler(favFile)
pickler.dump(
self.__favouriteSeries) # Saving refreshed favourites list
# Slot to open window with more information for favourites
def slot_open_serie_window(self):
idx = self.__favouritesWidget.currentRow()
try:
if idx == -1:
raise ValueError("No row is selected in favourites widget.")
else:
ser = self.__favouriteSeries[idx]
self.__newWindow = NewWindow(ser, self)
self.__newWindow.exec_()
except ValueError: # If the user hasn't selected an element from the list, currentRow returns -1 and an error dialog is displayed
QMessageBox.information(None, "Error",
"You didn't select a serie in your list.",
QMessageBox.Cancel)
# Slot to do the research
def slot_research(self):
self.__searchText = self.__searchWidget.text()
# Delete all the widgets displayed in scroll area
for i in reversed(range(self.__gridLayout.count())):
self.__gridLayout.itemAt(i).widget().setParent(None)
# Research on the API
self.__seriesList = searchSeries(self.__searchText)
# Add results of the research to the scroll area
for i in range(len(self.__seriesList)):
currentWidget = MainWidget(i, self.__seriesList[i])
self.__gridLayout.addWidget(currentWidget, *self.__positions[i])
self.__sigMapper.setMapping(currentWidget.favButton,
currentWidget.id)
currentWidget.favButton.clicked.connect(self.__sigMapper.map)
# A slot that is activated when the notifications checkbox is checked or unchecked. It starts and stops the notifications thread
def slot_change_notifications_state(self):
if (self.__alertWindow.notificationsEnabled):
self.__alertWindow.notificationsEnabled = False
self.__alertWindow.quit()
else:
self.__alertWindow.notificationsEnabled = True
self.__alertWindow = Afficher(
self.__favouriteSeries, self.__enableNotifications.isChecked(),
self)
self.__alertWindow.start()