184 lines
7.4 KiB
Python
184 lines
7.4 KiB
Python
import datetime
|
|
import logging
|
|
from typing import List
|
|
|
|
from PyQt5.QtCore import Qt, pyqtSlot, pyqtSignal
|
|
from PyQt5.QtGui import QShowEvent, QHideEvent
|
|
from PyQt5.QtWidgets import (
|
|
QHBoxLayout,
|
|
QWidget,
|
|
QSizePolicy,
|
|
QVBoxLayout,
|
|
QSpacerItem,
|
|
QScrollArea,
|
|
QFrame,
|
|
)
|
|
|
|
from rare.components.tabs.store.api.models.response import CatalogOfferModel, WishlistItemModel
|
|
from rare.widgets.flow_layout import FlowLayout
|
|
from rare.widgets.side_tab import SideTabContents
|
|
from rare.widgets.sliding_stack import SlidingStackedWidget
|
|
from .store_api import StoreAPI
|
|
from .widgets.details import DetailsWidget
|
|
from .widgets.groups import StoreGroup
|
|
from .widgets.items import StoreItemWidget
|
|
|
|
logger = logging.getLogger("StoreLanding")
|
|
|
|
|
|
class LandingPage(SlidingStackedWidget, SideTabContents):
|
|
|
|
def __init__(self, store_api: StoreAPI, parent=None):
|
|
super(LandingPage, self).__init__(parent=parent)
|
|
self.implements_scrollarea = True
|
|
|
|
self.landing_widget = LandingWidget(store_api, parent=self)
|
|
self.landing_widget.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
|
|
self.landing_widget.set_title.connect(self.set_title)
|
|
self.landing_widget.show_details.connect(self.show_details)
|
|
|
|
self.landing_scroll = QScrollArea(self)
|
|
self.landing_scroll.setWidgetResizable(True)
|
|
self.landing_scroll.setFrameStyle(QFrame.NoFrame | QFrame.Plain)
|
|
self.landing_scroll.setWidget(self.landing_widget)
|
|
self.landing_scroll.widget().setAutoFillBackground(False)
|
|
self.landing_scroll.viewport().setAutoFillBackground(False)
|
|
|
|
self.details_widget = DetailsWidget([], store_api, parent=self)
|
|
self.details_widget.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
|
|
self.details_widget.set_title.connect(self.set_title)
|
|
self.details_widget.back_clicked.connect(self.show_main)
|
|
|
|
self.setDirection(Qt.Horizontal)
|
|
self.addWidget(self.landing_scroll)
|
|
self.addWidget(self.details_widget)
|
|
|
|
@pyqtSlot()
|
|
def show_main(self):
|
|
self.slideInWidget(self.landing_scroll)
|
|
|
|
@pyqtSlot(object)
|
|
def show_details(self, game: CatalogOfferModel):
|
|
self.details_widget.update_game(game)
|
|
self.slideInWidget(self.details_widget)
|
|
|
|
|
|
class LandingWidget(QWidget, SideTabContents):
|
|
show_details = pyqtSignal(CatalogOfferModel)
|
|
|
|
def __init__(self, api: StoreAPI, parent=None):
|
|
super(LandingWidget, self).__init__(parent=parent)
|
|
self.api = api
|
|
|
|
layout = QVBoxLayout(self)
|
|
layout.setContentsMargins(0, 0, 3, 0)
|
|
self.setLayout(layout)
|
|
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
|
|
|
|
self.free_games_now = StoreGroup(self.tr("Free now"), layout=QHBoxLayout, parent=self)
|
|
self.free_games_now.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
|
|
|
|
self.free_games_next = StoreGroup(self.tr("Free next week"), layout=QHBoxLayout, parent=self)
|
|
self.free_games_next.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
|
|
|
|
self.discounts_group = StoreGroup(self.tr("Wishlist discounts"), layout=FlowLayout, parent=self)
|
|
self.discounts_group.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
|
|
|
|
self.games_group = StoreGroup(self.tr("Free to play"), FlowLayout, self)
|
|
self.games_group.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
|
|
self.games_group.loading(False)
|
|
self.games_group.setVisible(True)
|
|
|
|
layout.addWidget(self.free_games_now, alignment=Qt.AlignTop)
|
|
layout.addWidget(self.free_games_next, alignment=Qt.AlignTop)
|
|
layout.addWidget(self.discounts_group, alignment=Qt.AlignTop)
|
|
layout.addWidget(self.games_group, alignment=Qt.AlignTop)
|
|
layout.addItem(QSpacerItem(0, 0, QSizePolicy.Fixed, QSizePolicy.Expanding))
|
|
|
|
def showEvent(self, a0: QShowEvent) -> None:
|
|
if a0.spontaneous():
|
|
return super().showEvent(a0)
|
|
self.api.get_free(self.__update_free_games)
|
|
self.api.get_wishlist(self.__update_wishlist_discounts)
|
|
return super().showEvent(a0)
|
|
|
|
def hideEvent(self, a0: QHideEvent) -> None:
|
|
if a0.spontaneous():
|
|
return super().hideEvent(a0)
|
|
# TODO: Implement tab unloading
|
|
return super().hideEvent(a0)
|
|
|
|
def __update_wishlist_discounts(self, wishlist: List[WishlistItemModel]):
|
|
for w in self.discounts_group.findChildren(StoreItemWidget, options=Qt.FindDirectChildrenOnly):
|
|
self.discounts_group.layout().removeWidget(w)
|
|
w.deleteLater()
|
|
|
|
for item in wishlist:
|
|
if item.offer.price.totalPrice.discount > 0:
|
|
w = StoreItemWidget(self.api.cached_manager, item.offer)
|
|
w.show_details.connect(self.show_details)
|
|
self.discounts_group.layout().addWidget(w)
|
|
self.discounts_group.setVisible(bool(wishlist))
|
|
self.discounts_group.loading(False)
|
|
|
|
def __update_free_games(self, free_games: List[CatalogOfferModel]):
|
|
for w in self.free_games_now.findChildren(StoreItemWidget, options=Qt.FindDirectChildrenOnly):
|
|
self.free_games_now.layout().removeWidget(w)
|
|
w.deleteLater()
|
|
|
|
for w in self.free_games_next.findChildren(StoreItemWidget, options=Qt.FindDirectChildrenOnly):
|
|
self.free_games_next.layout().removeWidget(w)
|
|
w.deleteLater()
|
|
|
|
date = datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc)
|
|
free_now = []
|
|
free_next = []
|
|
for item in free_games:
|
|
try:
|
|
if item.price.totalPrice.discountPrice == 0:
|
|
free_now.append(item)
|
|
continue
|
|
if item.title == "Mystery Game":
|
|
free_next.append(item)
|
|
continue
|
|
except KeyError as e:
|
|
logger.warning(str(e))
|
|
|
|
if item.promotions is not None:
|
|
if not item.promotions.promotionalOffers:
|
|
start_date = item.promotions.upcomingPromotionalOffers[0].promotionalOffers[0].startDate
|
|
else:
|
|
start_date = item.promotions.promotionalOffers[0].promotionalOffers[0].startDate
|
|
|
|
if start_date > date:
|
|
free_next.append(item)
|
|
|
|
# free games now
|
|
self.free_games_now.setVisible(bool(free_now))
|
|
for item in free_now:
|
|
w = StoreItemWidget(self.api.cached_manager, item)
|
|
w.show_details.connect(self.show_details)
|
|
self.free_games_now.layout().addWidget(w)
|
|
self.free_games_now.loading(False)
|
|
|
|
# free games next week
|
|
self.free_games_next.setVisible(bool(free_next))
|
|
for item in free_next:
|
|
w = StoreItemWidget(self.api.cached_manager, item)
|
|
if item.title != "Mystery Game":
|
|
w.show_details.connect(self.show_details)
|
|
self.free_games_next.layout().addWidget(w)
|
|
self.free_games_next.loading(False)
|
|
|
|
def show_games(self, data):
|
|
for w in self.games_group.findChildren(StoreItemWidget, options=Qt.FindDirectChildrenOnly):
|
|
self.games_group.layout().removeWidget(w)
|
|
w.deleteLater()
|
|
|
|
if data:
|
|
for game in data:
|
|
w = StoreItemWidget(self.api.cached_manager, game)
|
|
w.show_details.connect(self.show_details)
|
|
self.games_group.layout().addWidget(w)
|
|
self.games_group.loading(False)
|