¿Cómo crear un índice pg_trgm usando SQLAlchemy for Scrapy?
Estoy usando Scrapy para extraer datos de un foro web. Estoy almacenando estos datos en una base de datos PostgreSQL usando SQLAlchemy. La tabla y las columnas crean bien, sin embargo, no puedo hacer que SQLAlchemy cree un índice en una de las columnas. Estoy tratando de crear un índice de trigrama (pg_trgm) usando gin.
El código Postgresql que crearía este índice es:
CREATE INDEX description_idx ON table USING gin (description gin_trgm_ops);
El código SQLAlchemy que he agregado a mi archivo models.py es:
desc_idx = Index('description_idx', text("description gin_trgm_ops"), postgresql_using='gin')
He agregado esta línea a mi models.py pero cuando reviso postgresql, el índice nunca se creó.
A continuación se encuentran mis archivos completos models.py y pipelines.py. ¿Voy sobre todo esto mal?
¡Cualquier ayuda sería muy apreciada!
modelos.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