You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
2.1 KiB
65 lines
2.1 KiB
import logging |
|
logging.basicConfig( |
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', |
|
level=logging.DEBUG) |
|
|
|
import requests |
|
import settings |
|
|
|
SEARCH_ENABLED = bool(settings.MEILI_URL) |
|
|
|
def meili_api(method, route, json=None, params=None, parse_json=True): |
|
try: |
|
r = method(settings.MEILI_URL + route, json=json, params=params, timeout=4) |
|
if r.status_code > 299: |
|
raise Exception('Bad response code ' + str(r.status_code)) |
|
if parse_json: |
|
return r.json() |
|
else: |
|
r.encoding = 'utf-8' |
|
return r.text |
|
except KeyboardInterrupt: |
|
raise |
|
except BaseException as e: |
|
logging.error('Problem with MeiliSearch api route: %s: %s', route, str(e)) |
|
return False |
|
|
|
def create_index(): |
|
json = dict(uid='qotnews', primaryKey='id') |
|
return meili_api(requests.post, 'indexes', json=json) |
|
|
|
def update_rankings(): |
|
json = ['typo', 'words', 'proximity', 'date:desc', 'exactness'] |
|
return meili_api(requests.post, 'indexes/qotnews/settings/ranking-rules', json=json) |
|
|
|
def update_attributes(): |
|
json = ['title', 'url', 'author'] |
|
r = meili_api(requests.post, 'indexes/qotnews/settings/searchable-attributes', json=json) |
|
json = ['id', 'ref', 'source', 'author', 'author_link', 'score', 'date', 'title', 'link', 'url', 'num_comments'] |
|
r = meili_api(requests.post, 'indexes/qotnews/settings/displayed-attributes', json=json) |
|
return r |
|
|
|
def init(): |
|
if not SEARCH_ENABLED: |
|
logging.info('Search is not enabled, skipping init.') |
|
return |
|
print(create_index()) |
|
update_rankings() |
|
update_attributes() |
|
|
|
def put_story(story): |
|
if not SEARCH_ENABLED: return |
|
return meili_api(requests.post, 'indexes/qotnews/documents', [story]) |
|
|
|
def search(q): |
|
if not SEARCH_ENABLED: return [] |
|
params = dict(q=q, limit=settings.FEED_LENGTH) |
|
r = meili_api(requests.get, 'indexes/qotnews/search', params=params, parse_json=False) |
|
return r |
|
|
|
if __name__ == '__main__': |
|
init() |
|
|
|
print(update_rankings()) |
|
|
|
print(search('facebook'))
|
|
|