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.
57 lines
1.7 KiB
57 lines
1.7 KiB
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'))
|
|
|