discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

pgtrgm-add
All checks were successful
Deploy on push / deploy (push) Successful in 1m49s
Details

Browse Source
This commit is contained in:
Untone 2024-02-25 16:12:08 +03:00
parent a4745df71b
commit 857a3648a3
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
services/db.py
View File

@ -115,7 +115,18 @@ def cache_method(cache_key: str):
inspector = inspect(engine) inspector = inspect(engine)
def add_pg_trgm_extension_if_not_exists():
with local_session() as session:
result = session.execute("SELECT 1 FROM pg_extension WHERE extname = 'pg_trgm';")
if not result.scalar():
session.execute("CREATE EXTENSION IF NOT EXISTS pg_trgm;")
print("pg_trgm extension added successfully.")
else:
print("pg_trgm extension already exists.")
def create_fts_index(table_name, fts_index_name): def create_fts_index(table_name, fts_index_name):
add_pg_trgm_extension_if_not_exists()
logger.info(f'Full text index for {table_name}...') logger.info(f'Full text index for {table_name}...')
authors_indexes = inspector.get_indexes(table_name) authors_indexes = inspector.get_indexes(table_name)
author_fts_index_exists = any( author_fts_index_exists = any(