Skip to content

Commit 266a6d8

Browse files
mihowDebian
authored andcommitted
Support filtering collection by deployment in interval method
1 parent dba9bba commit 266a6d8

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

ami/main/models.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2046,10 +2046,14 @@ def sample_manual(self, image_ids: list[int]):
20462046
qs = self.get_queryset()
20472047
return qs.filter(id__in=image_ids)
20482048

2049-
def sample_interval(self, minute_interval: int = 10, exclude_events: list[int] = []):
2049+
def sample_interval(
2050+
self, minute_interval: int = 10, exclude_events: list[int] = [], deployment_id: int | None = None
2051+
):
20502052
"""Create a sample of source images based on a time interval"""
20512053

20522054
qs = self.get_queryset()
2055+
if deployment_id:
2056+
qs = qs.filter(deployment=deployment_id)
20532057
if exclude_events:
20542058
qs = qs.exclude(event__in=exclude_events)
20552059
qs.exclude(event__in=exclude_events)

0 commit comments

Comments
 (0)