From ceec3ec123c5303cef2e1b6182eea8d182e8359b Mon Sep 17 00:00:00 2001
From: "Denis Ledoux dle@openerp.com" <>
Date: Tue, 23 Oct 2012 15:46:54 +0200
Subject: [PATCH] [ADD]Cron for generated costs

bzr revid: dle@openerp.com-20121023134654-xbvm5egnc44ro8k8
---
 addons/fleet/data.xml | 15 ++++++++++++++-
 addons/fleet/demo.xml |  6 +++---
 addons/fleet/fleet.py | 38 +++++++++++++++++++++++++++++++++++---
 3 files changed, 52 insertions(+), 7 deletions(-)

diff --git a/addons/fleet/data.xml b/addons/fleet/data.xml
index 07cf8062cde..f6da343abf0 100644
--- a/addons/fleet/data.xml
+++ b/addons/fleet/data.xml
@@ -2,7 +2,7 @@
 <openerp>
     <data noupdate="0">
       <record forcecreate="True" id="ir_cron_service_services_reminders" model="ir.cron">
-        <field name="name">Creation ofVehicle services and Services renewals reminders</field>
+        <field name="name">Creation of Vehicle services and Services renewals reminders</field>
         <field eval="True" name="active" />
         <field name="user_id" ref="base.user_root" />
         <field name="interval_number">1</field>
@@ -14,6 +14,19 @@
         <field eval="'()'" name="args" />
       </record>
 
+      <record forcecreate="True" id="ir_cron_contract_costs_generator" model="ir.cron">
+        <field name="name">Generation of contracts costs based on the costs frequency</field>
+        <field eval="True" name="active" />
+        <field name="user_id" ref="base.user_root" />
+        <field name="interval_number">1</field>
+        <field name="interval_type">minutes</field>
+        <field name="numbercall">-1</field>
+        <field eval="False" name="doall" />
+        <field eval="'fleet.vehicle.log.contract'" name="model" />
+        <field eval="'run_scheduler'" name="function" />
+        <field eval="'()'" name="args" />
+      </record>
+
       <record id="type_service_service_1" model="fleet.service.type">
           <field name="name">Calculation Benefit In Kind</field>
           <field name="category">service</field>
diff --git a/addons/fleet/demo.xml b/addons/fleet/demo.xml
index 1603c2ee654..f58bc5a9159 100644
--- a/addons/fleet/demo.xml
+++ b/addons/fleet/demo.xml
@@ -769,7 +769,7 @@
       <record id="log_contract_2" model="fleet.vehicle.log.contract" >
         <field name="vehicle_id" ref="vehicle_1" />
         <field name="amount">400</field>
-        <field name="cost_frequency">monthly</field>
+        <field name="cost_frequency">weekly</field>
         <field name="cost_type" ref="type_service_56"/>
         <field name="date">2012-01-01</field>
         <field name="start_date" eval="time.strftime('%Y-01-01')" />
@@ -806,7 +806,7 @@
       <record id="log_contract_3" model="fleet.vehicle.log.contract" >
         <field name="vehicle_id" ref="vehicle_3" />
         <field name="amount">400</field>
-        <field name="cost_frequency">monthly</field>
+        <field name="cost_frequency">daily</field>
         <field name="cost_type" ref="type_service_56"/>
         <field name="date">2012-01-01</field>
         <field name="start_date" eval="time.strftime('%Y-01-01')" />
@@ -843,7 +843,7 @@
       <record id="log_contract_4" model="fleet.vehicle.log.contract" >
         <field name="vehicle_id" ref="vehicle_4" />
         <field name="amount">400</field>
-        <field name="cost_frequency">monthly</field>
+        <field name="cost_frequency">yearly</field>
         <field name="cost_type" ref="type_service_56"/>
         <field name="date">2012-01-01</field>
         <field name="start_date" eval="time.strftime('%Y-01-01')" />
diff --git a/addons/fleet/fleet.py b/addons/fleet/fleet.py
index 3cc7d5f6284..7feda666814 100644
--- a/addons/fleet/fleet.py
+++ b/addons/fleet/fleet.py
@@ -1,8 +1,9 @@
+# -*- coding: utf-8 -*-
 from itertools import chain
 from osv import osv, fields
 import time
-import tools
 import datetime
+import tools
 from osv.orm import except_orm
 from tools.translate import _
 ############################
@@ -63,10 +64,8 @@ class fleet_vehicle_cost(osv.Model):
             data['vehicle_id'] = self.browse(cr, uid, data['parent_id'], context=context).vehicle_id.id
             data['date'] = self.browse(cr, uid, data['parent_id'], context=context).date
         if 'contract_id' in data and data['contract_id']:
-            print str(data)
             data['vehicle_id'] = self.pool.get('fleet.vehicle.log.contract').browse(cr, uid, data['contract_id'], context=context).vehicle_id.id
             data['cost_type'] = self.pool.get('fleet.vehicle.log.contract').browse(cr, uid, data['contract_id'], context=context).cost_type.id
-            print 'Contract : '+ str(data['contract_id']) +  ' Add generated cost for vehicle ' + str(data['vehicle_id']) + ' with cost type : ' + str(data['cost_type']) 
         cost_id = super(fleet_vehicle_cost, self).create(cr, uid, data, context=context)
         return cost_id
 
@@ -849,6 +848,39 @@ class fleet_service_type(osv.Model):
 class fleet_vehicle_log_contract(osv.Model):
 
     _inherits = {'fleet.vehicle.cost': 'cost_id'}
+
+    def run_scheduler(self,cr,uid,context=None):
+
+        d = datetime.datetime.now()
+        #d = datetime.date(2001, 01, 01)
+
+        frequencies = []
+        if d.day == 1 and d.month == 1:
+            frequencies.append(('cost_frequency','=','yearly'))
+        if d.day == 1:
+            frequencies.append(('cost_frequency','=','monthly'))
+        if d.isoweekday() == 1:
+            frequencies.append(('cost_frequency','=','weekly'))
+        frequencies.append(('cost_frequency','=','daily')) 
+
+        frequencies_size = len(frequencies)
+        for i in range(frequencies_size-1):
+            frequencies.insert(0,'|')
+
+        condition = ['&','&','&',('state','=','open',),('start_date','<=',d),('expiration_date','>=',d)]
+        condition.extend(frequencies)
+
+        print str(condition)
+        contract_ids = self.pool.get('fleet.vehicle.log.contract').search(cr, uid, condition, offset=0, limit=None, order=None,context=None, count=False)
+        for contract_id in contract_ids:
+
+            nbr = self.pool.get('fleet.vehicle.cost').search(cr,uid,['&',('contract_id','=',contract_id),('date','=',d)],context=context,count=True)
+
+            if not nbr:
+                contract = self.pool.get('fleet.vehicle.log.contract').browse(cr,uid,contract_id,context=context)
+                data = {'amount' : contract.amount,'date' : d,'vehicle_id' : contract.vehicle_id.id,'cost_type' : contract.cost_type.id,'contract_id' : contract_id}
+                cost_id = self.pool.get('fleet.vehicle.cost').create(cr, uid, data, context=context) 
+        return True
     
     def name_get(self, cr, uid, ids, context=None):
         if context is None: