Skip to content

Commit d891394

Browse files
update from_saved_search method to avoid possible srt-key/srt-dir conflicts
1 parent ebb42ce commit d891394

1 file changed

Lines changed: 30 additions & 17 deletions

File tree

src/_incydr_sdk/queries/file_events.py

Lines changed: 30 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -396,23 +396,6 @@ def subquery(self, subgroup_query: EventQuery):
396396
)
397397
return self
398398

399-
@classmethod
400-
def from_saved_search(cls, saved_search: SavedSearch):
401-
"""
402-
Create an `EventQuery` object from a `SavedSearch` response.
403-
"""
404-
query = cls()
405-
if saved_search.group_clause:
406-
query.group_clause = saved_search.group_clause
407-
if saved_search.groups:
408-
for i in saved_search.groups:
409-
query.groups.append(_handle_filter_group_type(i))
410-
if saved_search.srt_dir:
411-
query.sort_dir = saved_search.srt_dir
412-
if saved_search.srt_key:
413-
query.sort_key = saved_search.srt_key
414-
return query
415-
416399

417400
class EventQuery(BaseEventQuery):
418401
"""
@@ -430,6 +413,23 @@ class EventQuery(BaseEventQuery):
430413
sort_dir: str = Field("asc", alias="srtDir")
431414
sort_key: EventSearchTerm = Field("event.id", alias="srtKey")
432415

416+
@classmethod
417+
def from_saved_search(cls, saved_search: SavedSearch):
418+
"""
419+
Create an `EventQuery` object from a `SavedSearch` response.
420+
"""
421+
query = cls()
422+
if saved_search.group_clause:
423+
query.group_clause = saved_search.group_clause
424+
if saved_search.groups:
425+
for i in saved_search.groups:
426+
query.groups.append(_handle_filter_group_type(i))
427+
if saved_search.srt_dir:
428+
query.sort_dir = saved_search.srt_dir
429+
if saved_search.srt_key:
430+
query.sort_key = saved_search.srt_key
431+
return query
432+
433433

434434
class GroupingEventQuery(BaseEventQuery):
435435
"""
@@ -459,6 +459,19 @@ def maximum_size(self, size: int):
459459
self.size = size
460460
return self
461461

462+
@classmethod
463+
def from_saved_search(cls, saved_search: SavedSearch):
464+
"""
465+
Create an `EventQuery` object from a `SavedSearch` response.
466+
"""
467+
query = cls()
468+
if saved_search.group_clause:
469+
query.group_clause = saved_search.group_clause
470+
if saved_search.groups:
471+
for i in saved_search.groups:
472+
query.groups.append(_handle_filter_group_type(i))
473+
return query
474+
462475

463476
def _create_date_range_filter_group(start_date, end_date, term=None):
464477
def _validate_duration_str(iso_duration_str):

0 commit comments

Comments
 (0)