mirror of
https://github.com/Tautulli/Tautulli.git
synced 2025-07-06 05:01:14 -07:00
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import json
|
|
from copy import deepcopy
|
|
|
|
from cloudinary.api_client.call_api import call_json_api
|
|
|
|
|
|
class Search:
|
|
"""Build and execute a search query."""
|
|
def __init__(self):
|
|
self.query = {}
|
|
|
|
def expression(self, value):
|
|
"""Specify the search query expression."""
|
|
self.query["expression"] = value
|
|
return self
|
|
|
|
def max_results(self, value):
|
|
"""Set the max results to return"""
|
|
self.query["max_results"] = value
|
|
return self
|
|
|
|
def next_cursor(self, value):
|
|
"""Get next page in the query using the ``next_cursor`` value from a previous invocation."""
|
|
self.query["next_cursor"] = value
|
|
return self
|
|
|
|
def sort_by(self, field_name, direction=None):
|
|
"""Add a field to sort results by. If not provided, direction is ``desc``."""
|
|
if direction is None:
|
|
direction = 'desc'
|
|
self._add("sort_by", {field_name: direction})
|
|
return self
|
|
|
|
def aggregate(self, value):
|
|
"""Aggregate field."""
|
|
self._add("aggregate", value)
|
|
return self
|
|
|
|
def with_field(self, value):
|
|
"""Request an additional field in the result set."""
|
|
self._add("with_field", value)
|
|
return self
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.query)
|
|
|
|
def execute(self, **options):
|
|
"""Execute the search and return results."""
|
|
options["content_type"] = 'application/json'
|
|
uri = ['resources', 'search']
|
|
return call_json_api('post', uri, self.as_dict(), **options)
|
|
|
|
def _add(self, name, value):
|
|
if name not in self.query:
|
|
self.query[name] = []
|
|
self.query[name].append(value)
|
|
return self
|
|
|
|
def as_dict(self):
|
|
return deepcopy(self.query)
|