# Copyright 2018 Ivan Yelizariev <https://it-projects.info/team/yelizariev>
# License MIT (https://opensource.org/licenses/MIT).
from . import models
from . import wizard
from .tests import test_mail_model


def post_init_hook(cr, registry):
    from odoo import api, SUPERUSER_ID

    env = api.Environment(cr, SUPERUSER_ID, {})

    env.cr.execute(
        "ALTER TABLE res_users ADD COLUMN IF NOT EXISTS email_multi_website VARCHAR"
    )

    # fill new email column with values from partner
    for user in env["res.users"].with_context(active_test=False).search([]):
        email = user.partner_id.email
        if email:
            user._force_default("email_multi_website", email)


def uninstall_hook(cr, registry):
    from odoo import api, SUPERUSER_ID

    env = api.Environment(cr, SUPERUSER_ID, {})

    # remove properties
    field_ids = [
        env.ref("base.field_res_users_email").id,
        env.ref("base.field_res_users_signature").id,
        env.ref("mail.field_mail_template_body_html").id,
        env.ref("mail.field_mail_template_mail_server_id").id,
        env.ref("mail.field_mail_template_report_template").id,
    ]
    env["ir.property"].search([("fields_id", "in", field_ids)]).unlink()

    # copy emails from partner to user
    cr.execute("SELECT partner_id,email_multi_website FROM res_users")
    for partner_id, default_email in cr.fetchall():
        env["res.partner"].browse(partner_id).email = default_email