Пример #1
0
def tag_attributes(conn, tag_links):
    """
    Update attribute.attribute_tag_link table

    :param conn: Minerva database connection
    :param tag_links: list of tuples like (attribute_id, tag_name)
    """
    tmp_table_name = store_in_temp_table(conn, tag_links)

    query = (
        "INSERT INTO attribute_directory.attribute_tag_link "
            "(attribute_id, tag_id) "
        "(SELECT tmp.attribute_id, tag.id "
        "FROM {} tmp "
        "JOIN directory.tag tag ON lower(tag.name) = lower(tmp.tag) "
        "LEFT JOIN attribute_directory.attribute_tag_link ttl ON "
        "ttl.attribute_id = tmp.attribute_id AND ttl.tag_id = tag.id "
        "WHERE ttl.attribute_id IS NULL)").format(tmp_table_name)

    with closing(conn.cursor()) as cursor:
        cursor.execute(query)

    drop_table(conn, tmp_table_name)

    conn.commit()
Пример #2
0
def tag_trends(conn, tag_links):
    """
    Update trend.trend_tag_link table

    :param conn: Minerva database connection
    :param tag_links: list of tuples like (trend_id, tag_name)
    """
    tmp_table_name = store_in_temp_table(conn, tag_links)

    query = (
        "INSERT INTO {0}.trend_tag_link (trend_id, tag_id) "
        "("
        "   SELECT tmp.trend_id, tag.id "
        "   FROM {1} tmp "
        "   JOIN directory.tag tag ON lower(tag.name) = lower(tmp.tag) "
        "   LEFT JOIN {0}.trend_tag_link ttl ON "
        "       ttl.trend_id = tmp.trend_id AND ttl.tag_id = tag.id "
        "   WHERE ttl.trend_id IS NULL"
        ")").format(schema.name, tmp_table_name)

    with closing(conn.cursor()) as cursor:
        cursor.execute(query)

    drop_table(conn, tmp_table_name)

    conn.commit()