from odoo import api, SUPERUSER_ID


def post_init_hook(env_or_cr, registry=None):
    if registry is not None:
        env = api.Environment(env_or_cr, SUPERUSER_ID, {})
    else:
        env = env_or_cr
    CorporateWebsiteSetup(env).run()


class CorporateWebsiteSetup:
    def __init__(self, env):
        self.env = env

    def run(self):
        self.env["mcs.corporate.website.setup"].apply_corporate_branding()