forked from tanner/qotnews
Integrate with external MeiliSearch server
This commit is contained in:
57
apiserver/search.py
Normal file
57
apiserver/search.py
Normal file
@@ -0,0 +1,57 @@
|
||||
import logging
|
||||
logging.basicConfig(
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
level=logging.DEBUG)
|
||||
|
||||
import requests
|
||||
|
||||
MEILI_URL = 'http://127.0.0.1:7700/'
|
||||
|
||||
def create_index():
|
||||
try:
|
||||
json = dict(name='qotnews', uid='qotnews')
|
||||
r = requests.post(MEILI_URL + 'indexes', json=json, timeout=2)
|
||||
if r.status_code != 201:
|
||||
raise Exception('Bad response code ' + str(r.status_code))
|
||||
return r.json()
|
||||
except KeyboardInterrupt:
|
||||
raise
|
||||
except BaseException as e:
|
||||
logging.error('Problem creating MeiliSearch index: {}'.format(str(e)))
|
||||
return False
|
||||
|
||||
def init():
|
||||
create_index()
|
||||
|
||||
def put_story(story):
|
||||
story = story.copy()
|
||||
story.pop('text', None)
|
||||
story.pop('comments', None)
|
||||
try:
|
||||
r = requests.post(MEILI_URL + 'indexes/qotnews/documents', json=[story], timeout=2)
|
||||
if r.status_code != 202:
|
||||
raise Exception('Bad response code ' + str(r.status_code))
|
||||
return r.json()
|
||||
except KeyboardInterrupt:
|
||||
raise
|
||||
except BaseException as e:
|
||||
logging.error('Problem putting MeiliSearch story: {}'.format(str(e)))
|
||||
return False
|
||||
|
||||
def search(q):
|
||||
try:
|
||||
params = dict(q=q, limit=250)
|
||||
r = requests.get(MEILI_URL + 'indexes/qotnews/search', params=params, timeout=2)
|
||||
if r.status_code != 200:
|
||||
raise Exception('Bad response code ' + str(r.status_code))
|
||||
return r.json()['hits']
|
||||
except KeyboardInterrupt:
|
||||
raise
|
||||
except BaseException as e:
|
||||
logging.error('Problem searching MeiliSearch: {}'.format(str(e)))
|
||||
return False
|
||||
|
||||
if __name__ == '__main__':
|
||||
create_index()
|
||||
|
||||
print(search('the'))
|
Reference in New Issue
Block a user