How to chain queries?

[yashrathi-git]

Is there any way, to pass the result of the previous query (list of documents) to db.search so it don't search globally but only search in the provided input?

[msiemens]

The way TinyDB Queries work is that they are callables which return True if they match a document. This means that you can take a query and apply it to a list of objects like this:

docs = db.search(query1)
docs = filter(query2, docs)
# or:
docs = [doc for doc in docs if query2(doc)]

[yashrathi-git]

@msiemens That is really amazing. Tiny-db is really well designed and I love working with it, thank you.
For my use case I ended up doing this:

def _chain_queries(base_query: Optional[Query], new_query: Query):
    if base_query:
        return base_query & new_query
    return new_query


def query(projects: Optional[List[str]] = None,
          labels: Optional[List[str]] = None,
          timestamp_from: Optional[datetime] = None,
          timestamp_to: Optional[datetime] = None,
          priority: Optional[int] = None):
    generated_query = None

    projects_label_query = (_TaskQuery.projects.any(projects)) | (_TaskQuery.labels.any(labels))
    # `timestamp_from<=_TaskQuery.time.started_at<=timestamp_to` didn't worked, I tried it.
    # lower limit wasn't considered.
    time_stamp_query = (_TaskQuery.time.started_at.exists() & _TaskQuery.time.completed_at.exists()) & (
            (timestamp_from <= _TaskQuery.time.started_at) & (_TaskQuery.time.started_at <= timestamp_to))
    priority_query = _TaskQuery.priority == priority

    if projects or labels:
        generated_query = _chain_queries(generated_query, projects_label_query)

    if timestamp_from and timestamp_to:
        generated_query = _chain_queries(generated_query, time_stamp_query)

    if priority:
        generated_query = _chain_queries(generated_query, priority_query)
    if generated_query:
       return db.search(generated_query)

I haven't written tests for it yet, but in my brief manual testing, this seemed to work. It is so boring, and hectic to generate test data for it and then write tests.

Thank you!