social/mail_tracking/hooks.py

# Copyright 2016 Antonio Espinosa - <antonio.espinosa@tecnativa.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

import logging

from psycopg2.extensions import AsIs

_logger = logging.getLogger(__name__)


def column_exists(cr, table, column):
    cr.execute(
        """
        SELECT column_name
        FROM information_schema.columns
        WHERE table_name = %s AND column_name = %s""",
        (table, column),
    )
    return bool(cr.fetchall())


def column_add_with_value(cr, table, column, field_type, value):
    if not column_exists(cr, table, column):
        cr.execute(
            """
            ALTER TABLE %s
            ADD COLUMN %s %s""",
            (AsIs(table), AsIs(column), AsIs(field_type)),
        )
        cr.execute(
            """
            UPDATE %s SET %s = %s""",
            (AsIs(table), AsIs(column), value),
        )


def pre_init_hook(cr):
    _logger.info("Creating res.partner.tracking_emails_count column " "with value 0")
    column_add_with_value(cr, "res_partner", "tracking_emails_count", "integer", 0)
    _logger.info("Creating res.partner.email_score column " "with value 50.0")
    column_add_with_value(cr, "res_partner", "email_score", "double precision", 50.0)