66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
from meilisearch import Client
|
|
from meilisearch.errors import MeilisearchApiError
|
|
import json
|
|
import os
|
|
|
|
MEILI_URL=os.getenv('MEILI_URL')
|
|
MEILI_MASTER_KEY=os.getenv('MEILI_MASTER_KEY')
|
|
|
|
client = Client(MEILI_URL, MEILI_MASTER_KEY)
|
|
|
|
def index_exists(index_name: str) -> bool:
|
|
try:
|
|
client.get_index(index_name)
|
|
return True
|
|
except MeilisearchApiError:
|
|
return False
|
|
|
|
def init_indices():
|
|
if not index_exists('cities500'):
|
|
print("Creating cities index...")
|
|
client.create_index('cities500', {'primaryKey': 'id'})
|
|
|
|
client.index('cities500').update_settings({
|
|
'sortableAttributes': ['_geo',],
|
|
'filterableAttributes': ['_geo']
|
|
})
|
|
|
|
print("Starting data import...")
|
|
json_file = open('cities500.json', encoding='utf-8')
|
|
cities = json.load(json_file)
|
|
|
|
client.index('cities500').add_documents(cities)
|
|
print("Data import completed!")
|
|
|
|
if not index_exists('trails'):
|
|
print("Creating trails index...")
|
|
client.create_index('trails', {'primaryKey': 'id'})
|
|
|
|
client.index('trails').update_settings({
|
|
'sortableAttributes': ['name', 'distance', 'elevation_gain', 'created',],
|
|
'filterableAttributes': ['category', 'distance', 'elevation_gain', 'completed', '_geo', 'public', 'author']
|
|
})
|
|
|
|
|
|
def generate_public_token():
|
|
search_key = client.get_keys().results[0]
|
|
|
|
search_rules = {
|
|
'trails': {
|
|
'filter': 'public=true'
|
|
},
|
|
'cities500': {}
|
|
}
|
|
token = client.generate_tenant_token(api_key_uid=search_key.uid, search_rules=search_rules, api_key=search_key.key)
|
|
|
|
return token
|
|
|
|
print("Initializing indices...")
|
|
init_indices()
|
|
print("Indices initialized!")
|
|
token = generate_public_token()
|
|
print("Generating public token...")
|
|
print(f"PUBLIC_MEILISEARCH_API_TOKEN:{token}")
|
|
|
|
print("Bootstrapping complete!")
|