from lxml import etree

from openerp import tools
from openerp.osv import osv, fields, orm
from openerp import SUPERUSER_ID


class view(osv.osv):

    _inherit = "ir.ui.view"

    _columns = {
        'website_id': fields.many2one('website', ondelete='cascade', string="Website", copy=False),
        'key': fields.char('Key')
    }

    _sql_constraints = [(
        'key_website_id_unique',
        'unique(key, website_id)',
        'Key must be unique per website.'
    )]

    def _view_obj(self, cr, uid, view_id, context=None):
        if isinstance(view_id, basestring):
            try:
                return self.pool['ir.model.data'].xmlid_to_object(
                    cr, uid, view_id, raise_if_not_found=True, context=context
                )
            except:
                # Try to fallback on key instead of xml_id
                if self.search(cr, uid, [('key', '=', view_id),('website_id','=',context.get('website_id', False))], context=context):
                    rec_id = self.search(cr, uid, [('key', '=', view_id),('website_id','=',context.get('website_id', False))], context=context)
                else:
                    rec_id = self.search(cr, uid, [('key', '=', view_id)], context=context)
                if rec_id:
                    return self.browse(cr, uid, rec_id, context=context)[0]
                else:
                    raise
        elif isinstance(view_id, (int, long)):
            return self.browse(cr, uid, view_id, context=context)

        # assume it's already a view object (WTF?)
        return view_id

    @tools.ormcache_context(accepted_keys=('website_id',))
    def get_view_id(self, cr, uid, xml_id, context=None):
        if context and 'website_id' in context and not isinstance(xml_id, (int, long)):
            domain = [
                ('key', '=', xml_id),
                '|',
                ('website_id', '=', context['website_id']),
                ('website_id', '=', False)
            ]
            xml_ids = self.search(cr, uid, domain, order='website_id', limit=1, context=context)
            if not xml_ids:
                xml_id = self.pool['ir.model.data'].xmlid_to_res_id(cr, uid, xml_id, raise_if_not_found=True)
                if self.read(cr, uid, xml_id, ['page'], context=context)['page']:
                    raise ValueError('Invalid template id: %r' % (xml_id,))
            else:
                xml_id = xml_ids[0]
        else:
            xml_id = self.pool['ir.model.data'].xmlid_to_res_id(cr, uid, xml_id, raise_if_not_found=True)
        return xml_id

    _read_template_cache = dict(accepted_keys=('lang', 'inherit_branding', 'editable', 'translatable', 'website_id'))

    @tools.ormcache_context(**_read_template_cache)
    def _read_template(self, cr, uid, view_id, context=None):
        arch = self.read_combined(cr, uid, view_id, fields=['arch'], context=context)['arch']
        arch_tree = etree.fromstring(arch)

        if 'lang' in context:
            arch_tree = self.translate_qweb(cr, uid, view_id, arch_tree, context['lang'], context)

        self.distribute_branding(arch_tree)
        root = etree.Element('templates')
        root.append(arch_tree)
        arch = etree.tostring(root, encoding='utf-8', xml_declaration=True)
        return arch

    @tools.ormcache(size=0)
    def read_template(self, cr, uid, xml_id, context=None):
        if isinstance(xml_id, (int, long)):
            view_id = xml_id
        else:
            if '.' not in xml_id:
                raise ValueError('Invalid template id: %r' % (xml_id,))
            view_id = self.get_view_id(cr, uid, xml_id, context=context)
        return self._read_template(cr, uid, view_id, context=context)

    def clear_cache(self):
        self._read_template.clear_cache(self)
        self.get_view_id.clear_cache(self)
        
    def get_inheriting_views_arch(self, cr, uid, view_id, model, context=None):
        arch = super(view, self).get_inheriting_views_arch(cr, uid, view_id, model, context=context)
        if not context or not 'website_id' in context:
            return arch
        
        view_ids = [v for _, v in arch]
        view_arch_to_add_per_key = {}
        keep_view_ids = []
        for view_rec in self.browse(cr, SUPERUSER_ID, view_ids, context):
            #case 1: there is no key, always keep the view
            if not view_rec.key:
                keep_view_ids.append(view_rec.id)
                
            #case 2: Correct website
            elif view_rec.website_id and view_rec.website_id.id == context['website_id']:
                view_arch_to_add_per_key[view_rec.key] = (view_rec.website_id.id, view_rec.id)
            #case 3: no website add it if no website
            if not view_rec.website_id:
                view_web_id, view_id = view_arch_to_add_per_key.get(view_rec.key, (False, False))
                if not view_web_id:
                    view_arch_to_add_per_key[view_rec.key] = (False, view_rec.id)
                #else: do nothing, you already have the right view
            #case 4: website is wrong: do nothing
        #Put all the view_id we keep together
        keep_view_ids.extend([view_id for _, view_id in view_arch_to_add_per_key.values()])
        return [(arch, view_id) for arch, view_id  in arch if view_id in keep_view_ids]