# -*- coding: utf-8 -*- from datetime import date from dateutil.relativedelta import relativedelta from openerp import models, api, fields, tools, _ from openerp.exceptions import ValidationError, except_orm, Warning, RedirectWarning from openerp.osv import osv class ProjectServiceTask(models.Model): _name = 'project.service.task' name = fields.Char(string='Title', compute='_get_work_name', store=True) project_id = fields.Many2one('project.service', required=True, track_visibility='onchange') priority = fields.Selection([('0', 'Normal'), ('1', 'High')], string='Priority', select=True) partner_id = fields.Many2one('res.partner', string='Customer') date_assign = fields.Datetime(string='Assigning Date', select=True, copy=False, readonly=True) date = fields.Datetime(string='Deadline', select=True, copy=False) color = fields.Integer(string='Color Index') stage_id = fields.Many2one('project.service.stage', string='Stage', track_visibility='onchange', copy=False) activity_ids = fields.One2many('project.service.activity', 'task_id', string='Planned/Ordered Works') materials_used = fields.One2many('project.service.material', 'task_id', string='Materials Used') amount_total = fields.Float(string='Total Amount', compute="get_amount_total", readonly=True) include_materials = fields.Boolean('Incluir') state = fields.Selection([ ('Preparado', 'Preparado'), ('Facturado', 'Facturado'), ('Cancelado', 'Cancelado'), ], string='Status', readonly=True, default='Preparado', track_visibility='onchange', select=True) sequence = fields.Integer(string='Sequence', select=True, help="Gives the sequence order when displaying a list of tasks.") invoice_ids = fields.One2many('account.invoice', 'task_invoice_id') invoice_count = fields.Integer( string='Facturas', compute='_get_invoice_count', ) works_done = fields.One2many('project.service.activity', 'task_id', string='Work Done', domain=[('completed', '=', True)]) reserve_id = fields.Many2one('project.service.reserve', string='Reserva') _defaults = { 'stage_id': 1, 'state': 'Preparado', 'date': fields.datetime.now(), } @api.onchange('reserve_id') def get_reserve(self): self.partner_id = self.reserve_id.partner_id self.project_id = self.reserve_id.place @api.one @api.depends('partner_id','project_id') def _get_work_name(self): self.name = self.partner_id.name + ' / ' + self.project_id.name @api.multi def cancel(self): self.state = 'Cancelado' self.stage_id = 4 @api.depends('activity_ids.work_cost', 'materials_used.price') def get_amount_total(self): for records in self: for hour in records: amount_totall = 0.0 for line in hour.activity_ids: amount_totall += line.work_cost if self.include_materials==True: for line2 in hour.materials_used: amount_totall += line2.price records.amount_total = amount_totall @api.one @api.depends('invoice_ids','state') def _get_invoice_count(self): self.invoice_count = len(self.invoice_ids) @api.multi def button_draft(self): if self.invoice_count > 0: raise Warning('Esta tarea tiene una factura asociada') if self.invoice_count == 0: self.stage_id = 1 for work in self: work.write({'state': 'Preparado'}) return True @api.multi def unlink(self): for task in self: if task.state in ('Facturado'): raise Warning(('No puedes borrar una tarea ya facturada')) if len(task.activity_ids)>0: raise Warning(('No puedes borrar una tarea que tiene actividades')) return super(ProjectServiceTask, self).unlink() @api.multi def Facturado(self): activity = self.activity_ids if not activity: raise osv.except_osv(_('UserError!'), _('No puedes facturas una tarea sin actividades.')) self.state = 'Facturado' inv_obj = self.env['account.invoice'] inv_line_obj = self.env['account.invoice.line'] customer = self.partner_id if not customer.name: raise osv.except_osv(_('UserError!'), _('Please select a Customer.')) company_id = self.env['res.users'].browse(1).company_id self.ensure_one() ir_values = self.env['ir.values'] inv_data = { 'name': customer.name, 'reference': customer.name, 'account_id': customer.property_account_receivable.id, 'partner_id': customer.id, 'origin': self.name, 'task_invoice_id': self.id, } inv_id = inv_obj.create(inv_data) for records in self.activity_ids: if records.product_id.id: income_account = records.product_id.categ_id.property_account_income_categ.id if not income_account: raise osv.except_osv(_('UserError!'), _('There is no income account defined ' 'for this product: "%s".') % (records.product_id.name,)) if records.completed == False: raise osv.except_osv(_('UserError!'), _('Esta actividad aun no ha sido concluida ' '"%s".') % (records.product_id.name,)) inv_line_data = { 'name': records.product_id.name, 'account_id': income_account, 'price_unit': records.work_cost, 'quantity': 1, 'product_id': records.product_id.id, 'invoice_id': inv_id.id, 'invoice_line_tax_id': [(6, 0, [x.id for x in records.product_id.taxes_id])], } inv_line_obj.create(inv_line_data) for records in self.materials_used: if records.product_id.id: income_account = records.product_id.categ_id.property_account_income_categ.id if not income_account: raise osv.except_osv(_('UserError!'), _('There is no income account defined ' 'for this product: "%s".') % (records.product_id.name,)) if self.include_materials==True: inv_line_data = { 'name': records.product_id.name, 'account_id': income_account, 'price_unit': records.price, 'quantity': records.amount, 'product_id': records.product_id.id, 'invoice_id': inv_id.id, 'invoice_line_tax_id': [(6, 0, [x.id for x in records.product_id.taxes_id])], } inv_line_obj.create(inv_line_data) imd = self.env['ir.model.data'] action = imd.xmlid_to_object('account.action_invoice_tree1') list_view_id = imd.xmlid_to_res_id('account.invoice_tree') form_view_id = imd.xmlid_to_res_id('account.invoice_form') result = { 'name': action.name, 'help': action.help, 'type': 'ir.actions.act_window', 'views': [[list_view_id, 'tree'], [form_view_id, 'form'], [False, 'graph'], [False, 'kanban'], [False, 'calendar'], [False, 'pivot']], 'target': action.target, 'context': action.context, 'res_model': 'account.invoice', } if len(inv_id) > 1: result['domain'] = "[('id','in',%s)]" % inv_id.ids elif len(inv_id) == 1: result['views'] = [(form_view_id, 'form')] result['res_id'] = inv_id.ids[0] else: result = {'type': 'ir.actions.act_window_close'} invoiced_records = self.env['project.service.task'] total = 0 self.stage_id = 3 for rows in invoiced_records: invoiced_date = rows.date invoiced_date = invoiced_date[0:10] if invoiced_date == str(date.today()): total = total + rows.price_subtotal inv_id.signal_workflow('invoice_open') return result