Como criar um índice pg_trgm usando SQLAlchemy for Scrapy?
Estou usando o Scrapy para coletar dados de um fórum da web. Estou armazenando esses dados em um banco de dados PostgreSQL usando SQLAlchemy. A tabela e as colunas criam bem, no entanto, não é possível que o SQLAlchemy crie um índice em uma das colunas. Estou tentando criar um índice trigrama (pg_trgm) usando gin.
O código do Postgresql que criaria esse índice é:
CREATE INDEX description_idx ON table USING gin (description gin_trgm_ops);
O código SQLAlchemy que adicionei ao meu arquivo models.py é:
desc_idx = Index('description_idx', text("description gin_trgm_ops"), postgresql_using='gin')
Eu adicionei esta linha ao meu models.py, mas quando eu faço check-in no postgresql, o índice nunca foi criado.
Abaixo estão meus arquivos models.py e pipelines.py completos. Eu estou indo sobre isso tudo errado?
Qualquer ajuda seria muito apreciada!!
models.py:from sqlalchemy import create_engine, Column, Integer, String, DateTime, Index, text
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.engine.url import URL
import settings
DeclarativeBase = declarative_base()
def db_connect():
return create_engine(URL(**settings.DATABASE))
def create_forum_table(engine):
DeclarativeBase.metadata.create_all(engine)
class forumDB(DeclarativeBase):
__tablename__ = "table"
id = Column(Integer, primary_key=True)
title = Column('title', String)
desc = Column('description', String, nullable=True)
desc_idx = Index('description_idx', text("description gin_trgm_ops"), postgresql_using='gin')
pipelines.pyfrom scrapy.exceptions import DropItem
from sqlalchemy.orm import sessionmaker
from models import forumDB, db_connect, create_forum_table
class ScrapeforumToDB(object):
def __init__(self):
engine = db_connect()
create_forum_table(engine)
self.Session = sessionmaker(bind=engine)
def process_item(self, item, spider):
session = self.Session()
forumitem = forumDB(**item)
try:
session.add(forumitem)
session.commit()
except:
session.rollback()
raise
finally:
session.close()
return item