Skip to content

Support query language #37

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
vbraun wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Support query language #37

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 27 additions & 11 deletions prismic/api.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,8 @@ def form(self, name):
raise Exception("Bad form name %s, valid form names are: %s" % (name, ', '.join(self.forms)))
return SearchForm(self.forms.get(name), self.access_token, self.cache, self.request_handler)

def query(self, q, ref=None, page_size=None, page=None, orderings=None, after=None, fetch_links=None):
def query(self, q, ref=None, page_size=None, page=None, orderings=None,
after=None, fetch_links=None, lang=None):
if ref is None:
ref = self.get_master()
form = self.form('everything').ref(ref)
Expand All @@ -131,32 +132,39 @@ def query(self, q, ref=None, page_size=None, page=None, orderings=None, after=No
form.after(after)
if fetch_links is not None:
form.fetch_links(fetch_links)
if lang is not None:
form.lang(lang)
return form.query(q).submit()

def query_first(self, q, ref=None):
documents = self.query(q, ref, page_size=1, page=1).documents
def query_first(self, q, ref=None, lang=None):
documents = self.query(q, ref, page_size=1, page=1, lang=lang).documents
if len(documents) > 0:
return documents[0]

def get_by_uid(self, type, uid, ref=None):
return self.query_first(predicates.at('my.' + type + '.uid', uid), ref)
def get_by_uid(self, type, uid, ref=None, lang=None):
return self.query_first(
predicates.at('my.' + type + '.uid', uid), ref, lang=lang)

def get_by_id(self, id, ref=None):
return self.query_first(predicates.at('document.id', id), ref)
def get_by_id(self, id, ref=None, lang=None):
return self.query_first(predicates.at('document.id', id), ref, lang=lang)

def get_by_ids(self, ids, ref=None, page_size=None, page=None, orderings=None, after=None, fetch_links=None):
def get_by_ids(self, ids, ref=None, page_size=None,
page=None, orderings=None, after=None,
fetch_links=None, lang=None):
return self.query(
predicates.in_('document.id', ids),
ref,
page_size=page_size,
page=page,
orderings=orderings,
after=after,
fetch_links=fetch_links
fetch_links=fetch_links,
lang=lang,
)

def get_single(self, type, ref=None):
return self.query_first(predicates.at('document.type', type), ref)
def get_single(self, type, ref=None, lang=None):
return self.query_first(predicates.at('document.type', type), ref, lang=lang)


class Ref(object):
"""
Expand Down Expand Up @@ -316,6 +324,14 @@ def pageSize(self, nb_results):
"""
return self.page_size(nb_results)

def lang(self, lang):
"""
Set query language

:param lang: string
"""
return self.set("lang", lang)

def count(self):
"""Count the total number of results
"""
Expand Down