Better active drop search
This commit is contained in:
1 parent
bceb49b23a
commit
62a3bc715c
2 files changed
+6
-16
No files matched your search
@@ -155,12 +155,3 @@ class DropsCampaign:
|
|||||||
|
|
||||||
def get_drop(self, drop_id: str) -> Optional[TimedDrop]:
|
def get_drop(self, drop_id: str) -> Optional[TimedDrop]:
|
||||||
return self.timed_drops.get(drop_id)
|
return self.timed_drops.get(drop_id)
|
||||||
|
|
||||||
def get_active_drop(self) -> TimedDrop:
|
|
||||||
if not self.active:
|
|
||||||
raise RuntimeError("You can't call this for an unactive campaign")
|
|
||||||
drops = sorted(
|
|
||||||
(drop for drop in self.timed_drops.values() if drop.can_earn),
|
|
||||||
key=lambda d: d.remaining_minutes,
|
|
||||||
)
|
|
||||||
return drops[0]
|
|
||||||
@@ -191,12 +191,9 @@ class Twitch:
|
|||||||
self.gui.games.set_games(games)
|
self.gui.games.set_games(games)
|
||||||
selected_game = self.gui.games.get_selection()
|
selected_game = self.gui.games.get_selection()
|
||||||
# pre-display the active drop without a countdown
|
# pre-display the active drop without a countdown
|
||||||
for campaign in self.inventory:
|
active_drop = self.get_active_drop(selected_game)
|
||||||
if campaign.active and campaign.game == selected_game:
|
if active_drop is not None:
|
||||||
active_drop = campaign.get_active_drop()
|
active_drop.display(countdown=False)
|
||||||
if active_drop is not None:
|
|
||||||
active_drop.display(countdown=False)
|
|
||||||
break
|
|
||||||
self.change_state(State.CHANNEL_CLEANUP)
|
self.change_state(State.CHANNEL_CLEANUP)
|
||||||
elif self._state is State.CHANNEL_FETCH:
|
elif self._state is State.CHANNEL_FETCH:
|
||||||
if selected_game is None:
|
if selected_game is None:
|
||||||
@@ -693,9 +690,11 @@ class Twitch:
|
|||||||
def get_active_drop(self, game: Game) -> Optional[TimedDrop]:
|
def get_active_drop(self, game: Game) -> Optional[TimedDrop]:
|
||||||
drops = sorted(
|
drops = sorted(
|
||||||
(
|
(
|
||||||
campaign.get_active_drop()
|
drop
|
||||||
for campaign in self.inventory
|
for campaign in self.inventory
|
||||||
if campaign.active and campaign.game == game
|
if campaign.active and campaign.game == game
|
||||||
|
for drop in campaign.timed_drops.values()
|
||||||
|
if drop.can_earn
|
||||||
),
|
),
|
||||||
key=lambda d: d.remaining_minutes,
|
key=lambda d: d.remaining_minutes,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in new issue
Block a user