kingfisher_pro_original/models/products.py

# -*- coding: utf-8 -*-
# Part of Biztech Consultancy. See LICENSE file for full copyright and licensing details.

from openerp import api, fields, models, _
from openerp.exceptions import Warning
from openerp.addons.website_sale.controllers import main


class Brands(models.Model):
    _name = 'product.brands'

    name = fields.Char(string='Brand Name', required=True, translate=True)
    brand_description = fields.Html(string='Description', translate=True)
    brand_logo = fields.Binary(string='Brand Logo')
    brand_cover = fields.Binary(string='Brand Cover')
    product_ids = fields.One2many(
        'product.template',
        'product_brand_id',
        string='Product Brands',
    )
    products_count = fields.Integer(
        string='Number of products',
        compute='_get_products_count',
    )

    @api.one
    @api.depends('product_ids')
    def _get_products_count(self):
        self.products_count = len(self.product_ids)


class ProductStyleTags(models.Model):
    _name = 'biztech.product.style.tag'

    name = fields.Char(string='Tag Name', required=True, translate=True)
    color = fields.Selection(
        [('red', 'Red'), ('new', 'Green'), ('sale', 'Orange')], string="Color")
    product_ids = fields.One2many(
        'product.template',
        'product_style_tag_id',
        string='Product Tags',
    )


class KingProductImages(models.Model):
    _name = 'biztech.product.images'
    _description = "Add Multiple Image in Product"

    name = fields.Char(string='Title', translate=True)
    alt = fields.Char(string='Alt', translate=True)
    attach_type = fields.Selection([('image', 'Image'), ('video', 'Video')],
                                   default='image',
                                   string="Type")
    image = fields.Binary(string='Image')
    video_type = fields.Selection([('youtube', 'Youtube'),
                                   ('vimeo', 'Vimeo'),
                                   ('html5video', 'Html5 Video')],
                                  default='youtube',
                                  string="Video media player")
    cover_image = fields.Binary(string='Cover image',
                                # required=True,
                                help="Cover Image will be show untill video is loaded.")
    video_id = fields.Char(string='Video ID')
    video_ogv = fields.Char(string='Video OGV', help="Link for ogv format video")
    video_webm = fields.Char(string='Video WEBM', help="Link for webm format video")
    video_mp4 = fields.Char(string='Video MP4', help="Link for mp4 format video")
    sequence = fields.Integer(string='Sort Order')
    product_tmpl_id = fields.Many2one('product.template', string='Product')
    more_view_exclude = fields.Boolean(string="More View Exclude")


class ProductTemplate(models.Model):
    _inherit = 'product.template'

    product_brand_id = fields.Many2one(
        'product.brands',
        string='Brand',
        help='Select a brand for this product'
    )
    images = fields.One2many('biztech.product.images', 'product_tmpl_id',
                             string='Images')
    multi_image = fields.Boolean(string="Add Multiple Images?")
    tag_ids = fields.Many2many('biztech.product.tags', string="Tags")
    is_flip_image = fields. Boolean(
        string="Add flip image", help="Enable this checkbox for adding flip image on product website.")
    flip_image = fields.Binary(string='Flip image',
                               help="Flip image will be shown on mouse hover in website on specific product.")
    product_style_tag_id = fields.Many2one(
        'biztech.product.style.tag',
        string='Tags',
        help='Select a tag for this product'
    )


class ProductTags(models.Model):
    _name = 'biztech.product.tags'
    _order = "sequence"

    name = fields.Char(string="Tag Name", help="Tag Name", required=True, translate=True)
    active = fields.Boolean(
        string="Active", help="Enable or Disable tag from website", default=True)
    sequence = fields.Integer(
        string='Sequence', help="You can define sequence of tags you want to show tags")
    product_ids = fields.Many2many('product.template', string='Products', required=True)

    _sql_constraints = [('unique_tag_name', 'unique(name)', 'Tag name should be unique..!'), ]


class ProductSortBy(models.Model):
    _name = 'biztech.product.sortby'

    name = fields.Char(string="Name", help='Name for sorting option', translate=True,
                       required=True)
    sort_type = fields.Selection(
        [('asc', 'Ascending'), ('desc', 'Descending')], string="Type", default='asc')
    sort_on = fields.Many2one('ir.model.fields', string='Sort On',
                              help='Select field on which you want to apply sorting',
                              domain=[('model', '=', 'product.template'), ('ttype', 'in', ('char', 'float', 'integer', 'datetime', 'date'))])


class ProductCategory(models.Model):
    _inherit = 'product.public.category'

    include_in_megamenu = fields.Boolean(
        string="Include in mega menu", help="Include in mega menu")
    menu_id = fields.Many2one('website.menu', string="Main menu")
    description = fields.Text(string="Description",
                              help="Short description which will be visible below category slider.")


class ProductPerPageNo(models.Model):
    _name = "product.per.page.no"
    _order = 'name asc'

    name = fields.Integer(string='Product per page')
    set_default_check = fields.Boolean(string="Set default")
    prod_page_id = fields.Many2one('product.per.page')

    @api.model
    def create(self, vals):
        res = super(ProductPerPageNo, self).create(vals)
        if vals.get('name') == 0:
            raise Warning(_("Warning! You cannot set 'zero' for product page."))
        if vals.get('set_default_check'):
            true_records = self.search(
                [('set_default_check', '=', True), ('id', '!=', res.id)])
            true_records.write({'set_default_check': False})
        return res

    @api.multi
    def write(self, vals):
        res = super(ProductPerPageNo, self).write(vals)
        if vals.get('name') == 0:
            raise Warning(_("Warning! You cannot set 'zero' for product page."))
        if vals.get('set_default_check'):
            true_records = self.search(
                [('set_default_check', '=', True), ('id', '!=', self.id)])
            true_records.write({'set_default_check': False})
        return res


class ProductPerPage(models.Model):
    _name = "product.per.page"

    name = fields.Char(string="Label Name", translate=True)
    no_ids = fields.One2many(
        'product.per.page.no', 'prod_page_id', string="No of product to display")

    @api.multi
    def write(self, vals):
        res = super(ProductPerPage, self).write(vals)
        default_pg = self.env['product.per.page.no'].search([('set_default_check', '=', True)])
        if default_pg.name:
            main.PPG = int(default_pg.name)
        else:
            raise Warning(_("Warning! You have to set atleast one default value."))
        return res