Source code for soweego.importer.models.base_link_entity

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""Base `SQLAlchemy <https://www.sqlalchemy.org/>`_ ORM entity for URLs."""

__author__ = 'Marco Fossati'
__email__ = 'fossati@spaziodati.eu'
__version__ = '1.0'
__license__ = 'GPL-3.0'
__copyright__ = 'Copyleft 2018, Hjfocs'

from sqlalchemy import Boolean, Column, Index, Integer, String, Text
from sqlalchemy.ext.declarative import (
    AbstractConcreteBase,
    declarative_base,
    declared_attr,
)

BASE = declarative_base()


class BaseLinkEntity(AbstractConcreteBase, BASE):
    """Minimal ORM structure for a target catalog link/URL.
    Each ORM link entity should inherit this class.

    **Attributes:**

    - **internal_id** (integer) - an internal primary key
    - **catalog_id** (string(50)) - a target catalog identifier
    - **url** (text) - a full URL
    - **is_wiki** (boolean) - whether a URL is a Wiki link or not
    - **url_tokens** (text) - a **url** tokenized through
      :func:`~soweego.commons.text_utils.tokenize`

    """

    __tablename__ = None
    internal_id = Column(
        Integer, unique=True, primary_key=True, autoincrement=True
    )
    # Catalog identifier of the entity having the link, indexed
    catalog_id = Column(String(50), nullable=False, index=True)
    # Full URL
    url = Column(Text)
    # Whether the URL is a Wiki link or not
    is_wiki = Column(Boolean)
    # Tokenized URL
    url_tokens = Column(Text)

    # Full-text index over 'url_tokens'
    @declared_attr
    def __table_args__(cls):
        return (
            Index(
                'ftix_url_tokens_%s' % cls.__tablename__,
                "url_tokens",
                mysql_prefix="FULLTEXT",
            ),
            {'mysql_charset': 'utf8mb4'},
        )

    def __repr__(self) -> str:
        return "<BaseLinkEntity(catalog_id='{0}', url='{1}')>".format(
            self.catalog_id, self.url
        )