addons/fleet/fleet.py

   1 # -*- coding: utf-8 -*-
   2 from itertools import chain
   3 from osv import osv, fields
   4 import time
   5 import datetime
   6 import tools
   7 from osv.orm import except_orm
   8 from tools.translate import _
   9 from dateutil.relativedelta import relativedelta
  10 ############################
  11 ############################
  12 #Vehicle.cost class
  13 ############################
  14 ############################
  15
  16 class fleet_vehicle_cost(osv.Model):
  17     _name = 'fleet.vehicle.cost'
  18     _description = 'Cost of vehicle'
  19     _order = 'date desc, vehicle_id asc'
  20
  21     def name_get(self, cr, uid, ids, context=None):
  22         if context is None:
  23             context = {}
  24         if not ids:
  25             return []
  26         reads = self.browse(cr, uid, ids, context=context)
  27         res = []
  28         for record in reads:
  29             if record.vehicle_id.license_plate:
  30                 name = record.vehicle_id.license_plate
  31             if record.cost_subtype.name:
  32                 name = name + ' / '+ record.cost_subtype.name
  33             if record.date:
  34                 name = name + ' / '+ record.date
  35             res.append((record.id, name))
  36         return res
  37
  38     def _cost_name_get_fnc(self, cr, uid, ids, name, unknow_none, context=None):
  39         res = self.name_get(cr, uid, ids, context=context)
  40         return dict(res)
  41
  42     _columns = {
  43         'name' : fields.function(_cost_name_get_fnc, type="char", string='Name', store=True),
  44         #'name' : fields.char('Name',size=32),
  45         'vehicle_id': fields.many2one('fleet.vehicle', 'Vehicle', required=True, help='Vehicle concerned by this fuel log'),
  46         'cost_subtype': fields.many2one('fleet.service.type', 'Service type', required=False, help='Service type purchased with this cost'),
  47         'amount': fields.float('Total Price'),
  48         'cost_type' : fields.selection([('contract', 'Contract'),('services','Services'),('fuel','Fuel'),('other','Other')], 'Category of the cost', help='For internal purpose only',required=True),
  49         'parent_id': fields.many2one('fleet.vehicle.cost', 'Parent', required=False, help='Parent cost to this current cost'),
  50         'cost_ids' : fields.one2many('fleet.vehicle.cost', 'parent_id', 'Included Services'),
  51
  52         'date' :fields.date('Date',help='Date when the cost has been executed'),
  53
  54         'contract_id' : fields.many2one('fleet.vehicle.log.contract', 'Contract', required=False, help='Contract attached to this cost'),
  55         'auto_generated' : fields.boolean('automatically generated',readonly=True,required=True)
  56     }
  57
  58     _default ={
  59         'parent_id':None,
  60         'auto_generated' : False,
  61         'cost_type' : 'other',
  62     }
  63
  64
  65
  66     def create(self, cr, uid, data, context=None):
  67         if 'parent_id' in data and data['parent_id']:
  68             parent = self.browse(cr, uid, data['parent_id'], context=context)
  69             data['vehicle_id'] = parent.vehicle_id.id
  70             data['date'] = parent.date
  71             data['cost_type'] = parent.cost_type
  72         if 'contract_id' in data and data['contract_id']:
  73             contract = self.pool.get('fleet.vehicle.log.contract').browse(cr, uid, data['contract_id'], context=context)
  74             data['vehicle_id'] = contract.vehicle_id.id
  75             data['cost_subtype'] = contract.cost_subtype.id
  76             data['cost_type'] = contract.cost_type
  77         if not('cost_type' in data and data['cost_type']):
  78             data['cost_type'] = 'other'
  79         cost_id = super(fleet_vehicle_cost, self).create(cr, uid, data, context=context)
  80         return cost_id
  81
  82 ############################
  83 ############################
  84 #Vehicle.tag class
  85 ############################
  86 ############################
  87
  88 class fleet_vehicle_tag(osv.Model):
  89     _name = 'fleet.vehicle.tag'
  90     _columns = {
  91         'name': fields.char('Name', required=True, translate=True),
  92     }
  93
  94 ############################
  95 ############################
  96 #Vehicle.state class
  97 ############################
  98 ############################
  99
 100 class fleet_vehicle_state(osv.Model):
 101     _name = 'fleet.vehicle.state'
 102     _columns = {
 103         'name': fields.char('Name', required=True),
 104         'sequence': fields.integer('Order',help="Used to order the note stages")
 105     }
 106     _order = 'sequence asc'
 107     _sql_constraints = [('fleet_state_name_unique','unique(name)','State name already exists')]
 108
 109
 110 ############################
 111 ############################
 112 #Vehicle.model class
 113 ############################
 114 ############################
 115
 116 class fleet_vehicle_model(osv.Model):
 117
 118     def name_get(self, cr, uid, ids, context=None):
 119         if context is None:
 120             context = {}
 121         if not ids:
 122             return []
 123         reads = self.browse(cr, uid, ids, context=context)
 124         res = []
 125         for record in reads:
 126             name = record.modelname
 127             if record.brand.name:
 128                 name = record.brand.name+' / '+name
 129             res.append((record.id, name))
 130         return res
 131
 132     def _model_name_get_fnc(self, cr, uid, ids, prop, unknow_none, context=None):
 133         res = self.name_get(cr, uid, ids, context=context)
 134         return dict(res)
 135
 136     def on_change_brand(self, cr, uid, ids, model_id, context=None):
 137
 138         if not model_id:
 139             return {}
 140
 141         brand = self.pool.get('fleet.vehicle.model.brand').browse(cr, uid, model_id, context=context)
 142
 143         return {
 144             'value' : {
 145                 'image' : brand.image,
 146             }
 147         }
 148
 149     _name = 'fleet.vehicle.model'
 150     _description = 'Model of a vehicle'
 151
 152     _columns = {
 153         'name' : fields.function(_model_name_get_fnc, type="char", string='Name', store=True),
 154         'modelname' : fields.char('Model name', size=32, required=True),
 155         'brand' : fields.many2one('fleet.vehicle.model.brand', 'Model Brand', required=True, help='Brand of the vehicle'),
 156         'vendors': fields.many2many('res.partner','fleet_vehicle_model_vendors','model_id', 'partner_id',string='Vendors',required=False),
 157         'image': fields.related('brand','image',type="binary",string="Logo",store=False),
 158         'image_medium': fields.related('brand','image_medium',type="binary",string="Logo",store=False),
 159         'image_small': fields.related('brand','image_small',type="binary",string="Logo",store=False),
 160     }
 161
 162 ############################
 163 ############################
 164 #Vehicle.brand class
 165 ############################
 166 ############################
 167
 168 class fleet_vehicle_model_brand(osv.Model):
 169     _name = 'fleet.vehicle.model.brand'
 170     _description = 'Brand model of the vehicle'
 171
 172     _order = 'name asc'
 173
 174     def _get_image(self, cr, uid, ids, name, args, context=None):
 175         result = dict.fromkeys(ids, False)
 176         for obj in self.browse(cr, uid, ids, context=context):
 177             result[obj.id] = tools.image_get_resized_images(obj.image)
 178         return result
 179
 180     def _set_image(self, cr, uid, id, name, value, args, context=None):
 181         return self.write(cr, uid, [id], {'image': tools.image_resize_image_big(value)}, context=context)
 182
 183     _columns = {
 184         'name' : fields.char('Brand Name',size=32, required=True),
 185
 186         'image': fields.binary("Logo",
 187             help="This field holds the image used as logo for the brand, limited to 1024x1024px."),
 188         'image_medium': fields.function(_get_image, fnct_inv=_set_image,
 189             string="Medium-sized photo", type="binary", multi="_get_image",
 190             store = {
 191                 'fleet.vehicle.model.brand': (lambda self, cr, uid, ids, c={}: ids, ['image'], 10),
 192             },
 193             help="Medium-sized logo of the brand. It is automatically "\
 194                  "resized as a 128x128px image, with aspect ratio preserved. "\
 195                  "Use this field in form views or some kanban views."),
 196         'image_small': fields.function(_get_image, fnct_inv=_set_image,
 197             string="Smal-sized photo", type="binary", multi="_get_image",
 198             store = {
 199                 'fleet.vehicle.model.brand': (lambda self, cr, uid, ids, c={}: ids, ['image'], 10),
 200             },
 201             help="Small-sized photo of the brand. It is automatically "\
 202                  "resized as a 64x64px image, with aspect ratio preserved. "\
 203                  "Use this field anywhere a small image is required."),
 204     }
 205
 206 ############################
 207 ############################
 208 #Vehicle class
 209 ############################
 210 ############################
 211
 212
 213 class fleet_vehicle(osv.Model):
 214
 215     _inherit = 'mail.thread'
 216
 217     def name_get(self, cr, uid, ids, context=None):
 218         if context is None:
 219             context = {}
 220         if not ids:
 221             return []
 222         reads = self.browse(cr, uid, ids, context=context)
 223         res = []
 224         for record in reads:
 225             if record.license_plate:
 226                 name = record.license_plate
 227             if record.model_id.modelname:
 228                 name = record.model_id.modelname + ' / ' + name
 229             if record.model_id.brand.name:
 230                 name = record.model_id.brand.name+' / '+ name
 231             res.append((record.id, name))
 232         return res
 233
 234     def _vehicle_name_get_fnc(self, cr, uid, ids, prop, unknow_none, context=None):
 235         res = self.name_get(cr, uid, ids, context=context)
 236         return dict(res)
 237
 238     def act_show_log_services(self, cr, uid, ids, context=None):
 239         """ This opens log view to view and add new log for this vehicle
 240             @return: the service log view
 241         """
 242         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid ,'fleet','fleet_vehicle_log_services_act', context)
 243         res['context'] = {
 244             'default_vehicle_id': ids[0]
 245         }
 246         res['domain']=[('vehicle_id','=', ids[0])]
 247         return res
 248
 249     def act_show_log_contract(self, cr, uid, ids, context=None):
 250         """ This opens log view to view and add new log for this vehicle
 251             @return: the contract log view
 252         """
 253         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid ,'fleet','fleet_vehicle_log_contract_act', context)
 254         res['context'] = {
 255             'default_vehicle_id': ids[0]
 256         }
 257         res['domain']=[('vehicle_id','=', ids[0])]
 258         return res
 259
 260     def act_show_log_fuel(self, cr, uid, ids, context=None):
 261         """ This opens log view to view and add new log for this vehicle
 262             @return: the fuel log view
 263         """
 264         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid ,'fleet','fleet_vehicle_log_fuel_act', context)
 265         res['context'] = {
 266             'default_vehicle_id': ids[0]
 267         }
 268         res['domain']=[('vehicle_id','=', ids[0])]
 269         return res
 270
 271     def act_show_log_cost(self, cr, uid, ids, context=None):
 272         """ This opens log view to view and add new log for this vehicle
 273             @return: the costs log view
 274         """
 275         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid ,'fleet','fleet_vehicle_costs_act', context)
 276         res['context'] = {
 277             'default_vehicle_id': ids[0],
 278             'search_default_parent_false' : True
 279         }
 280         res['domain']=[('vehicle_id','=', ids[0])]
 281         return res
 282
 283     def act_show_log_odometer(self, cr, uid, ids, context=None):
 284         """ This opens log view to view and add new log for this vehicle
 285             @return: the odometer log view
 286         """
 287         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid ,'fleet','fleet_vehicle_odometer_act', context)
 288         res['context'] = {
 289             'default_vehicle_id': ids[0]
 290         }
 291         res['domain']=[('vehicle_id','=', ids[0])]
 292         return res
 293
 294     def _get_odometer(self, cr, uid, ids, odometer_id, arg, context):
 295         res = dict.fromkeys(ids, False)
 296         for record in self.browse(cr,uid,ids,context=context):
 297             ids = self.pool.get('fleet.vehicle.odometer').search(cr,uid,[('vehicle_id','=',record.id)],limit=1, order='value desc')
 298             if len(ids) > 0:
 299                 res[record.id] = str(self.pool.get('fleet.vehicle.odometer').browse(cr,uid,ids[0],context=context).value)
 300             else:
 301                 res[record.id] = str(0)
 302         return res
 303
 304     def _set_odometer(self, cr, uid, id, name, value, args=None, context=None):
 305         if value:
 306             try:
 307                 value = float(value)
 308             except ValueError:
 309                 #_logger.exception(value+' is not a correct odometer value. Please, fill a float for this field')
 310                 raise except_orm(_('Error!'), value+' is not a correct odometer value. Please, fill a float for this field')
 311
 312             date = time.strftime('%Y-%m-%d')
 313             data = {'value' : value,'date' : date,'vehicle_id' : id}
 314             odometer_id = self.pool.get('fleet.vehicle.odometer').create(cr, uid, data, context=context)
 315             return value
 316         self.write(cr, uid, id, {'odometer_id': ''})
 317         return False
 318
 319     def str_to_date(self,strdate):
 320         return datetime.datetime(int(strdate[:4]),int(strdate[5:7]),int(strdate[8:]))
 321
 322     def get_overdue_contract_reminder_fnc(self,cr,uid,ids,context=None):
 323         if context is None:
 324             context={}
 325         if not ids:
 326             return dict([])
 327         reads = self.browse(cr,uid,ids,context=context)
 328         res=[]
 329
 330         for record in reads:
 331             overdue=0
 332             if (record.log_contracts):
 333                 for element in record.log_contracts:
 334                     if (element.state=='open' and element.expiration_date):
 335                         current_date_str=time.strftime('%Y-%m-%d')
 336                         due_time_str=element.expiration_date
 337                             #due_time_str=element.browse()
 338                         current_date=self.str_to_date(current_date_str)
 339                         due_time=self.str_to_date(due_time_str)
 340
 341                         diff_time=int((due_time-current_date).days)
 342                         if diff_time<0:
 343                             overdue = overdue +1;
 344                 res.append((record.id,overdue))
 345             else:
 346                 res.append((record.id,0))
 347
 348         return dict(res)
 349
 350     def get_overdue_contract_reminder(self,cr,uid,ids,prop,unknow_none,context=None):
 351         res = self.get_overdue_contract_reminder_fnc(cr, uid, ids, context=context)
 352         return res
 353
 354     def get_next_contract_reminder_fnc(self,cr,uid,ids,context=None):
 355         if context is None:
 356             context={}
 357         if not ids:
 358             return dict([])
 359         reads = self.browse(cr,uid,ids,context=context)
 360         res=[]
 361
 362         for record in reads:
 363             due_soon=0
 364             if (record.log_contracts):
 365                 for element in record.log_contracts:
 366                     if (element.state=='open' and element.expiration_date):
 367                         current_date_str=time.strftime('%Y-%m-%d')
 368                         due_time_str=element.expiration_date
 369                             #due_time_str=element.browse()
 370                         current_date=self.str_to_date(current_date_str)
 371                         due_time=self.str_to_date(due_time_str)
 372
 373                         diff_time=int((due_time-current_date).days)
 374                         if diff_time<15 and diff_time>=0:
 375                             due_soon = due_soon +1;
 376                 res.append((record.id,due_soon))
 377             else:
 378                 res.append((record.id,0))
 379
 380         return dict(res)
 381
 382     def _search_get_overdue_contract_reminder(self, cr, uid, obj, name, args, context):
 383         res = []
 384         for field, operator, value in args:
 385             #assert field == name
 386             vehicle_ids = self.search(cr, uid, [])
 387             renew_ids = self.get_overdue_contract_reminder_fnc(cr,uid,vehicle_ids,context=context)
 388             res_ids = []
 389             for renew_key,renew_value in renew_ids.items():
 390                 if eval(str(renew_value) + " " + str(operator) + " " + str(value)):
 391                     res_ids.append(renew_key)
 392             res.append(('id', 'in', res_ids))
 393         return res
 394
 395     def _search_contract_renewal_due_soon(self, cr, uid, obj, name, args, context):
 396         res = []
 397         for field, operator, value in args:
 398             #assert field == name
 399             vehicle_ids = self.search(cr, uid, [])
 400             renew_ids = self.get_next_contract_reminder_fnc(cr,uid,vehicle_ids,context=context)
 401             res_ids = []
 402             for renew_key,renew_value in renew_ids.items():
 403                 if eval(str(renew_value) + " " + str(operator) + " " + str(value)):
 404                     res_ids.append(renew_key)
 405             res.append(('id', 'in', res_ids))
 406         return res
 407
 408     def get_next_contract_reminder(self, cr, uid, ids, prop, unknow_none, context=None):
 409         res = self.get_next_contract_reminder_fnc(cr, uid, ids, context=context)
 410         return res
 411
 412     def get_contract_renewal_names(self,cr,uid,ids,function_name,args,context=None):
 413         if not ids:
 414             return dict([])
 415         reads = self.browse(cr,uid,ids,context=context)
 416         res=[]
 417         for record in reads:
 418             if (record.log_contracts):
 419                 ids = self.pool.get('fleet.vehicle.log.contract').search(cr,uid,[('vehicle_id','=',record.id),('state','=','open')],limit=1,order='expiration_date asc')
 420                 if len(ids) > 0:
 421                     res.append((record.id,self.pool.get('fleet.vehicle.log.contract').browse(cr,uid,ids[0],context=context).cost_subtype.name))
 422             else:
 423                 res.append((record.id,''))
 424         return dict(res)
 425
 426     def get_total_contract_reminder(self,cr,uid,ids,prop,unknow_none,context=None):
 427         if context is None:
 428             context={}
 429         if not ids:
 430             return dict([])
 431         reads = self.browse(cr,uid,ids,context=context)
 432         res=[]
 433
 434         for record in reads:
 435             due_soon=0
 436             if (record.log_contracts):
 437                 for element in record.log_contracts:
 438                     if (element.state=='open' and element.expiration_date):
 439                         current_date_str=time.strftime('%Y-%m-%d')
 440                         due_time_str=element.expiration_date
 441
 442                         current_date=self.str_to_date(current_date_str)
 443                         due_time=self.str_to_date(due_time_str)
 444
 445                         diff_time=int((due_time-current_date).days)
 446                         if diff_time<15:
 447                             due_soon = due_soon +1;
 448                 if due_soon>0:
 449                     due_soon=due_soon-1
 450                 res.append((record.id,due_soon))
 451             else:
 452                 res.append((record.id,0))
 453
 454         return dict(res)
 455
 456     def run_scheduler(self,cr,uid,context=None):
 457         ids = self.pool.get('fleet.vehicle').search(cr, uid, [], offset=0, limit=None, order=None,context=None, count=False)
 458         nexts = self.get_next_contract_reminder_fnc(cr,uid,ids,context=context)
 459         overdues = self.get_overdue_contract_reminder_fnc(cr,uid,ids,context=context)
 460         for key,value in nexts.items():
 461             if value > 0 and overdues[key] > 0:
 462                 self.message_post(cr, uid, [key], body=str(value) + ' contract(s) has to be renewed soon and '+str(overdues[key])+' contract(s) is (are) overdued', context=context)
 463             elif value > 0:
 464                 self.message_post(cr, uid, [key], body=str(value) + ' contract(s) has to be renewed soon!', context=context)
 465             elif overdues[key] > 0 :
 466                 self.message_post(cr, uid, [key], body=str(overdues[key]) + ' contract(s) is(are) overdued!', context=context)
 467         return True
 468
 469     _name = 'fleet.vehicle'
 470     _description = 'Fleet Vehicle'
 471     #_order = 'contract_renewal_overdue desc, contract_renewal_due_soon desc'
 472     _order= 'license_plate asc'
 473     _columns = {
 474         'name' : fields.function(_vehicle_name_get_fnc, type="char", string='Name', store=True),
 475
 476         'company_id': fields.many2one('res.company', 'Company'),
 477         'license_plate' : fields.char('License Plate', size=32, required=True, help='License plate number of the vehicle (ie: plate number for a car)'),
 478         'vin_sn' : fields.char('Chassis Number', size=32, required=False, help='Unique number written on the vehicle motor (VIN/SN number)'),
 479         'driver' : fields.many2one('res.partner', 'Driver',required=False, help='Driver of the vehicle'),
 480         'model_id' : fields.many2one('fleet.vehicle.model', 'Model', required=True, help='Model of the vehicle'),
 481         'log_fuel' : fields.one2many('fleet.vehicle.log.fuel','vehicle_id', 'Fuel Logs'),
 482         'log_services' : fields.one2many('fleet.vehicle.log.services','vehicle_id', 'Services Logs'),
 483         'log_contracts' : fields.one2many('fleet.vehicle.log.contract','vehicle_id', 'Contracts'),
 484         'acquisition_date' : fields.date('Acquisition Date', required=False, help='Date when the vehicle has been bought'),
 485         'color' : fields.char('Color',size=32, help='Color of the vehicle'),
 486         'state': fields.many2one('fleet.vehicle.state', 'State', help='Current state of the vehicle',ondelete="set null"),
 487         'location' : fields.char('Location',size=32, help='Location of the vehicle (garage, ...)'),
 488         'seats' : fields.integer('Seats Number', help='Number of seats of the vehicle'),
 489         'doors' : fields.integer('Doors Number', help='Number of doors of the vehicle'),
 490         'tag_ids' :fields.many2many('fleet.vehicle.tag','fleet_vehicle_vehicle_tag_rel','vehicle_tag_id','tag_id','Tags'),
 491
 492         'odometer' : fields.function(_get_odometer,fnct_inv=_set_odometer,type='char',string='Odometer Value',store=False,help='Odometer measure of the vehicle at the moment of this log'),
 493         'odometer_unit': fields.selection([('kilometers', 'Kilometers'),('miles','Miles')], 'Odometer Unit', help='Unit of the odometer ',required=True),
 494
 495         'transmission' : fields.selection([('manual', 'Manual'),('automatic','Automatic')], 'Transmission', help='Transmission Used by the vehicle',required=False),
 496         'fuel_type' : fields.selection([('gasoline', 'Gasoline'),('diesel','Diesel'),('electric','Electric'),('hybrid','Hybrid')], 'Fuel Type', help='Fuel Used by the vehicle',required=False),
 497         'horsepower' : fields.integer('Horsepower',required=False),
 498         'horsepower_tax': fields.float('Horsepower Taxation'),
 499         'power' : fields.integer('Power (kW)',required=False,help='Power in kW of the vehicle'),
 500         'co2' : fields.float('CO2 Emissions',required=False,help='CO2 emissions of the vehicle'),
 501
 502         'image': fields.related('model_id','image',type="binary",string="Logo",store=False),
 503         'image_medium': fields.related('model_id','image_medium',type="binary",string="Logo",store=False),
 504         'image_small': fields.related('model_id','image_small',type="binary",string="Logo",store=False),
 505
 506         'contract_renewal_due_soon' : fields.function(get_next_contract_reminder,fnct_search=_search_contract_renewal_due_soon,type="integer",string='Contracts to renew',store=False),
 507         'contract_renewal_overdue' : fields.function(get_overdue_contract_reminder,fnct_search=_search_get_overdue_contract_reminder,type="integer",string='Contracts Overdued',store=False),
 508         'contract_renewal_name' : fields.function(get_contract_renewal_names,type="text",string='Name of contract to renew soon',store=False),
 509         'contract_renewal_total' : fields.function(get_total_contract_reminder,type="integer",string='Total of contracts due or overdue minus one',store=False),
 510
 511         'car_value': fields.float('Car Value', help='Value of the bought vehicle'),
 512         #'leasing_value': fields.float('Leasing value',help='Value of the leasing(Monthly, usually'),
 513         }
 514
 515     _defaults = {
 516         'doors' : 5,
 517         'odometer_unit' : 'kilometers',
 518         'state' : lambda s,cr,uid,c:s.get_state(cr,uid,'Active',context=c),
 519     }
 520
 521     def get_state(self,cr,uid,state_name,context=None):
 522         states=self.pool.get('fleet.vehicle.state').search(cr,uid,[('name','=',state_name)],context=context,limit=1)
 523         return states
 524
 525     def copy(self, cr, uid, id, default=None, context=None):
 526         if not default:
 527             default = {}
 528
 529         default.update({
 530         #    'name': self.pool.get('ir.sequence').get(cr, uid, 'stock.orderpoint') or '',
 531             #'log_ids':[],
 532             'log_fuel':[],
 533             'log_contracts':[],
 534             'log_services':[],
 535             'tag_ids':[],
 536             'vin_sn':'',
 537         })
 538         return super(fleet_vehicle, self).copy(cr, uid, id, default, context=context)
 539
 540     def on_change_model(self, cr, uid, ids, model_id, context=None):
 541
 542         if not model_id:
 543             return {}
 544
 545         model = self.pool.get('fleet.vehicle.model').browse(cr, uid, model_id, context=context)
 546
 547         return {
 548             'value' : {
 549                 'image' : model.image,
 550             }
 551         }
 552     def create(self, cr, uid, data, context=None):
 553         vehicle_id = super(fleet_vehicle, self).create(cr, uid, data, context=context)
 554         try:
 555             vehicle = self.browse(cr, uid, vehicle_id, context=context)
 556             self.message_post(cr, uid, [vehicle_id], body='Vehicle %s has been added to the fleet!' % (vehicle.license_plate), context=context)
 557         except:
 558             pass # group deleted: do not push a message
 559         return vehicle_id
 560
 561     def write(self, cr, uid, ids, vals, context=None):
 562         changes = []
 563         if 'driver' in vals:
 564             value = self.pool.get('res.partner').browse(cr,uid,vals['driver'],context=context).name
 565             olddriver = self.browse(cr, uid, ids, context)[0].driver
 566             if olddriver:
 567                 olddriver = olddriver.name
 568             else:
 569                 olddriver = 'None'
 570             changes.append('Driver: from \'' + olddriver + '\' to \'' + value+'\'')
 571         if 'state' in vals:
 572             value = self.pool.get('fleet.vehicle.state').browse(cr,uid,vals['state'],context=context).name
 573             oldstate = self.browse(cr, uid, ids, context)[0].state
 574             if oldstate:
 575                 oldstate=oldstate.name
 576             else:
 577                 oldstate = 'None'
 578             changes.append('State: from \'' + oldstate + '\' to \'' + value+'\'')
 579         if 'license_plate' in vals:
 580             old_license_plate = self.browse(cr, uid, ids, context)[0].license_plate
 581             if not old_license_plate:
 582                 old_license_plate = 'None'
 583             changes.append('License Plate: from \'' + old_license_plate + '\' to \'' + vals['license_plate']+'\'')
 584
 585         vehicle_id = super(fleet_vehicle,self).write(cr, uid, ids, vals, context)
 586
 587         try:
 588             if len(changes) > 0:
 589                 self.message_post(cr, uid, [self.browse(cr, uid, ids, context)[0].id], body=", ".join(changes), context=context)
 590         except Exception as e:
 591             print e
 592             pass
 593         return vehicle_id
 594
 595 ############################
 596 ############################
 597 #Vehicle.odometer class
 598 ############################
 599 ############################
 600
 601 class fleet_vehicle_odometer(osv.Model):
 602     _name='fleet.vehicle.odometer'
 603     _description='Odometer log for a vehicle'
 604
 605     _order='date desc'
 606
 607     def name_get(self, cr, uid, ids, context=None):
 608         if context is None:
 609             context = {}
 610         if not ids:
 611             return []
 612         reads = self.browse(cr, uid, ids, context=context)
 613         res = []
 614         for record in reads:
 615             if record.vehicle_id.name:
 616                 name = str(record.vehicle_id.name)
 617             if record.date:
 618                 name = name+ ' / '+ str(record.date)
 619             res.append((record.id, name))
 620         return res
 621
 622     def _vehicle_log_name_get_fnc(self, cr, uid, ids, prop, unknow_none, context=None):
 623         res = self.name_get(cr, uid, ids, context=context)
 624         return dict(res)
 625     def on_change_vehicle(self, cr, uid, ids, vehicle_id, context=None):
 626
 627         if not vehicle_id:
 628             return {}
 629
 630         odometer_unit = self.pool.get('fleet.vehicle').browse(cr, uid, vehicle_id, context=context).odometer_unit
 631
 632         return {
 633             'value' : {
 634                 'unit' : odometer_unit,
 635             }
 636         }
 637
 638     _columns = {
 639         'name' : fields.function(_vehicle_log_name_get_fnc, type="char", string='Name', store=True),
 640
 641         'date' : fields.date('Purchase Date'),
 642         'value' : fields.float('Odometer Value',group_operator="max"),
 643         'vehicle_id' : fields.many2one('fleet.vehicle', 'Vehicle', required=True),
 644         'unit': fields.related('vehicle_id','odometer_unit',type="char",string="Unit",store=False, readonly=True),
 645
 646     }
 647     _defaults = {
 648         'date' : time.strftime('%Y-%m-%d')
 649     }
 650
 651 ############################
 652 ############################
 653 #Vehicle.log classes
 654 ############################
 655 ############################
 656
 657
 658 ############################
 659 ############################
 660 #Vehicle.log.fuel class
 661 ############################
 662 ############################
 663
 664
 665 class fleet_vehicle_log_fuel(osv.Model):
 666
 667     #_inherits = {'fleet.vehicle.odometer': 'odometer_id'}
 668     _inherits = {'fleet.vehicle.cost': 'cost_id'}
 669
 670     def on_change_vehicle(self, cr, uid, ids, vehicle_id, context=None):
 671
 672         if not vehicle_id:
 673             return {}
 674
 675         odometer_unit = self.pool.get('fleet.vehicle').browse(cr, uid, vehicle_id, context=context).odometer_unit
 676
 677         return {
 678             'value' : {
 679                 'odometer_unit' : odometer_unit,
 680             }
 681         }
 682
 683     def on_change_liter(self, cr, uid, ids, liter, price_per_liter, amount, context=None):
 684
 685         if liter > 0 and price_per_liter > 0:
 686             return {'value' : {'amount' : float(liter) * float(price_per_liter),}}
 687         elif liter > 0 and amount > 0:
 688             return {'value' : {'price_per_liter' : float(amount) / float(liter),}}
 689         elif price_per_liter > 0 and amount > 0:
 690             return {'value' : {'liter' : float(amount) / float(price_per_liter),}}
 691         else :
 692             return {}
 693
 694     def on_change_price_per_liter(self, cr, uid, ids, liter, price_per_liter, amount, context=None):
 695
 696         liter = float(liter);
 697         price_per_liter = float(price_per_liter);
 698         if price_per_liter > 0 and liter > 0:
 699             return {'value' : {'amount' : float(liter) * float(price_per_liter),}}
 700         elif price_per_liter > 0 and amount > 0:
 701             return {'value' : {'liter' : float(amount) / float(price_per_liter),}}
 702         elif liter > 0 and amount > 0:
 703             return {'value' : {'price_per_liter' : float(amount) / float(liter),}}
 704         else :
 705             return {}
 706
 707     def on_change_amount(self, cr, uid, ids, liter, price_per_liter, amount, context=None):
 708
 709         if amount > 0 and liter > 0:
 710             return {'value' : {'price_per_liter' : float(amount) / float(liter),}}
 711         elif amount > 0 and price_per_liter > 0:
 712             return {'value' : {'liter' : float(amount) / float(price_per_liter),}}
 713         elif liter > 0 and price_per_liter > 0:
 714             return {'value' : {'amount' : float(liter) * float(price_per_liter),}}
 715         else :
 716             return {}
 717
 718     def _get_odometer(self, cr, uid, ids, odometer_id, arg, context):
 719         res = dict.fromkeys(ids, False)
 720         for record in self.browse(cr,uid,ids,context=context):
 721             if record.odometer_id:
 722                 res[record.id] = record.odometer_id.value
 723         return res
 724
 725     def _set_odometer(self, cr, uid, id, name, value, args=None, context=None):
 726         if value:
 727             try:
 728                 value = float(value)
 729             except ValueError:
 730                 #_logger.exception(value+' is not a correct odometer value. Please, fill a float for this field')
 731                 raise except_orm(_('Error!'), value+' is not a correct odometer value. Please, fill a float for this field')
 732
 733             date = self.browse(cr, uid, id, context=context).date
 734             if not(date):
 735                 date = time.strftime('%Y-%m-%d')
 736             vehicle_id = self.browse(cr, uid, id, context=context).vehicle_id
 737             data = {'value' : value,'date' : date,'vehicle_id' : vehicle_id.id}
 738             odometer_id = self.pool.get('fleet.vehicle.odometer').create(cr, uid, data, context=context)
 739             self.write(cr, uid, id, {'odometer_id': odometer_id})
 740             return value
 741         self.write(cr, uid, id, {'odometer_id': ''})
 742         return False
 743
 744     def _get_default_service_type(self, cr, uid, context):
 745         model, model_id = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'fleet', 'type_service_refueling')
 746         return model_id
 747
 748     _name = 'fleet.vehicle.log.fuel'
 749
 750     _columns = {
 751         #'name' : fields.char('Name',size=64),
 752         'liter' : fields.float('Liter'),
 753         'price_per_liter' : fields.float('Price Per Liter'),
 754         'purchaser_id' : fields.many2one('res.partner', 'Purchaser',domain="['|',('customer','=',True),('employee','=',True)]"),
 755         'inv_ref' : fields.char('Invoice Reference', size=64),
 756         'vendor_id' : fields.many2one('res.partner', 'Supplier', domain="[('supplier','=',True)]"),
 757         'notes' : fields.text('Notes'),
 758         'odometer_id' : fields.many2one('fleet.vehicle.odometer', 'Odometer', required=False, help='Odometer measure of the vehicle at the moment of this log'),
 759         'odometer' : fields.function(_get_odometer,fnct_inv=_set_odometer,type='char',string='Odometer',store=False),
 760         'odometer_unit': fields.related('vehicle_id','odometer_unit',type="char",string="Unit",store=False, readonly=True),
 761         'cost_amount': fields.related('cost_id','amount',type="float",string="Amount",store=True, readonly=True),
 762     }
 763     _defaults = {
 764         'purchaser_id': lambda self, cr, uid, ctx: uid,
 765         'date' : time.strftime('%Y-%m-%d'),
 766         'cost_subtype': _get_default_service_type,
 767     }
 768     def create(self, cr, uid, data, context=None):
 769         data['cost_type'] = 'fuel'
 770         cost_id = super(fleet_vehicle_log_fuel, self).create(cr, uid, data, context=context)
 771         return cost_id
 772
 773 ############################
 774 ############################
 775 #Vehicle.log.service class
 776 ############################
 777 ############################
 778
 779
 780 class fleet_vehicle_log_services(osv.Model):
 781
 782     _inherits = {'fleet.vehicle.cost': 'cost_id'}
 783
 784     def on_change_vehicle(self, cr, uid, ids, vehicle_id, context=None):
 785
 786         if not vehicle_id:
 787             return {}
 788
 789         odometer_unit = self.pool.get('fleet.vehicle').browse(cr, uid, vehicle_id, context=context).odometer_unit
 790
 791         return {
 792             'value' : {
 793                 'odometer_unit' : odometer_unit,
 794             }
 795         }
 796
 797     def _get_odometer(self, cr, uid, ids, odometer_id, arg, context):
 798         res = dict.fromkeys(ids, False)
 799         for record in self.browse(cr,uid,ids,context=context):
 800             if record.odometer_id:
 801                 res[record.id] = record.odometer_id.value
 802         return res
 803
 804     def _set_odometer(self, cr, uid, id, name, value, args=None, context=None):
 805         if value:
 806             try:
 807                 value = float(value)
 808             except ValueError:
 809                 #_logger.exception(value+' is not a correct odometer value. Please, fill a float for this field')
 810                 raise except_orm(_('Error!'), value+' is not a correct odometer value. Please, fill a float for this field')
 811
 812             date = self.browse(cr, uid, id, context=context).date
 813             if not(date):
 814                 date = time.strftime('%Y-%m-%d')
 815             vehicle_id = self.browse(cr, uid, id, context=context).vehicle_id
 816             data = {'value' : value,'date' : date,'vehicle_id' : vehicle_id.id}
 817             odometer_id = self.pool.get('fleet.vehicle.odometer').create(cr, uid, data, context=context)
 818             self.write(cr, uid, id, {'odometer_id': odometer_id})
 819             return value
 820         self.write(cr, uid, id, {'odometer_id': ''})
 821         return False
 822
 823     _name = 'fleet.vehicle.log.services'
 824     _columns = {
 825
 826         #'name' : fields.char('Name',size=64),
 827
 828         'purchaser_id' : fields.many2one('res.partner', 'Purchaser',domain="['|',('customer','=',True),('employee','=',True)]"),
 829         'inv_ref' : fields.char('Invoice Reference', size=64),
 830         'vendor_id' :fields.many2one('res.partner', 'Supplier', domain="[('supplier','=',True)]"),
 831         'notes' : fields.text('Notes'),
 832
 833         'odometer_id' : fields.many2one('fleet.vehicle.odometer', 'Odometer', required=False, help='Odometer measure of the vehicle at the moment of this log'),
 834         'odometer' : fields.function(_get_odometer,fnct_inv=_set_odometer,type='char',string='Odometer Value',store=False),
 835         'odometer_unit': fields.related('vehicle_id','odometer_unit',type="char",string="Unit",store=False, readonly=True),
 836         'cost_amount': fields.related('cost_id','amount',type="float",string="Amount",store=True, readonly=True),
 837     }
 838     _defaults = {
 839         'purchaser_id': lambda self, cr, uid, ctx: uid,
 840         'date' : time.strftime('%Y-%m-%d'),
 841     }
 842     def create(self, cr, uid, data, context=None):
 843         data['cost_type'] = 'services'
 844         cost_id = super(fleet_vehicle_log_services, self).create(cr, uid, data, context=context)
 845         return cost_id
 846
 847 ############################
 848 ############################
 849 #Vehicle.service.type class
 850 ############################
 851 ############################
 852
 853 class fleet_service_type(osv.Model):
 854     _name = 'fleet.service.type'
 855     _columns = {
 856         'name': fields.char('Name', required=True, translate=True),
 857         'category': fields.selection([('contract', 'Contract'), ('service', 'Service'),('both', 'Both')], 'Category',required=True, help='Choose wheter the service refer to contracts, vehicle services or both'),
 858     }
 859     #_defaults = {
 860     #    'category': 'both'
 861     #}
 862
 863 ############################
 864 ############################
 865 #Vehicle.log.contract class
 866 ############################
 867 ############################
 868
 869 class fleet_vehicle_log_contract(osv.Model):
 870
 871     _inherits = {'fleet.vehicle.cost': 'cost_id'}
 872
 873     def run_scheduler(self,cr,uid,context=None):
 874
 875         d = datetime.date.today()
 876         #d = datetime.datetime(2012, 12, 01)
 877
 878         contract_ids = self.pool.get('fleet.vehicle.log.contract').search(cr, uid, [('state','=','open')], offset=0, limit=None, order=None,context=None, count=False)
 879         for contract in self.pool.get('fleet.vehicle.log.contract').browse(cr,uid,contract_ids,context=context):
 880             if not contract.start_date:
 881                 break;
 882             if contract.generated_cost_ids != []:
 883                 last_cost_id = self.pool.get('fleet.vehicle.cost').search(cr, uid, ['&',('contract_id','=',contract.id),('auto_generated','=',True)], offset=0, limit=1, order='date desc',context=None, count=False)
 884                 last_cost_date = self.pool.get('fleet.vehicle.cost').browse(cr,uid,last_cost_id[0],context=None).date
 885                 found = True
 886             else :
 887                 found = False
 888                 last_cost_date = contract.start_date
 889             startdate = datetime.datetime.strptime(last_cost_date,'%Y-%m-%d').date()
 890             if contract.cost_frequency == 'yearly':
 891                 delta = relativedelta(years=+1)
 892             elif contract.cost_frequency == 'monthly':
 893                 delta = relativedelta(months=+1)
 894             elif contract.cost_frequency == 'weekly':
 895                 delta = relativedelta(weeks=+1)
 896             elif contract.cost_frequency == 'daily':
 897                 delta = relativedelta(days=+1)
 898             elif contract.cost_frequency == 'no':
 899                 break;
 900             if found:
 901                 startdate += delta
 902             while startdate < d:
 903                 data = {'amount' : contract.cost_generated,'date' : startdate.strftime('%Y-%m-%d'),'vehicle_id' : contract.vehicle_id.id,'cost_subtype' : contract.cost_subtype.id,'contract_id' : contract.id,'auto_generated' : True}
 904                 print data
 905                 cost_id = self.pool.get('fleet.vehicle.cost').create(cr, uid, data, context=context)
 906                 startdate += delta
 907         return True
 908
 909     def name_get(self, cr, uid, ids, context=None):
 910         if context is None:
 911             context = {}
 912         if not ids:
 913             return []
 914         reads = self.browse(cr, uid, ids, context=context)
 915         res = []
 916         for record in reads:
 917             if record.vehicle_id.name:
 918                 name = str(record.vehicle_id.name)
 919             if record.cost_subtype.name:
 920                 name = name+ ' / '+ str(record.cost_subtype.name)
 921             if record.date:
 922                 name = name+ ' / '+ record.date
 923             res.append((record.id, name))
 924         return res
 925
 926     def _vehicle_contract_name_get_fnc(self, cr, uid, ids, prop, unknow_none, context=None):
 927         res = self.name_get(cr, uid, ids, context=context)
 928         return dict(res)
 929
 930     def _get_odometer(self, cr, uid, ids, odometer_id, arg, context):
 931         res = dict.fromkeys(ids, False)
 932         for record in self.browse(cr,uid,ids,context=context):
 933             if record.odometer_id:
 934                 res[record.id] = record.odometer_id.value
 935         return res
 936
 937     def _set_odometer(self, cr, uid, id, name, value, args=None, context=None):
 938         if value:
 939             try:
 940                 value = float(value)
 941             except ValueError:
 942                 #_logger.exception(value+' is not a correct odometer value. Please, fill a float for this field')
 943                 raise except_orm(_('Error!'), value+' is not a correct odometer value. Please, fill a float for this field')
 944
 945             date = self.browse(cr, uid, id, context=context).date
 946             if not(date):
 947                 date = time.strftime('%Y-%m-%d')
 948             vehicle_id = self.browse(cr, uid, id, context=context).vehicle_id
 949             data = {'value' : value,'date' : date,'vehicle_id' : vehicle_id.id}
 950             odometer_id = self.pool.get('fleet.vehicle.odometer').create(cr, uid, data, context=context)
 951             self.write(cr, uid, id, {'odometer_id': odometer_id})
 952             return value
 953         self.write(cr, uid, id, {'odometer_id': ''})
 954         return False
 955
 956     def on_change_vehicle(self, cr, uid, ids, vehicle_id, context=None):
 957
 958         if not vehicle_id:
 959             return {}
 960
 961         odometer_unit = self.pool.get('fleet.vehicle').browse(cr, uid, vehicle_id, context=context).odometer_unit
 962
 963         return {
 964             'value' : {
 965                 'odometer_unit' : odometer_unit,
 966             }
 967         }
 968
 969     def compute_next_year_date(self, strdate):
 970         oneyear=datetime.timedelta(days=365)
 971         curdate = self.str_to_date(strdate)
 972         nextyear=curdate+oneyear#int(strdate[:4])+1
 973         return str(nextyear)#+strdate[4:]
 974
 975     def on_change_start_date(self, cr, uid, ids, strdate, enddate, context=None):
 976
 977         if (strdate):
 978
 979             return {'value' : {'expiration_date' : self.compute_next_year_date(strdate),}}
 980         else:
 981             return {}
 982
 983     def str_to_date(self,strdate):
 984         return datetime.datetime(int(strdate[:4]),int(strdate[5:7]),int(strdate[8:]))
 985
 986     def get_warning_date(self,cr,uid,ids,prop,unknow_none,context=None):
 987         if context is None:
 988             context={}
 989         if not ids:
 990             return dict([])
 991         reads = self.browse(cr,uid,ids,context=context)
 992         res=[]
 993         for record in reads:
 994             #if (record.reminder==True):
 995             if (record.expiration_date and record.state=='open'):
 996                 today=self.str_to_date(time.strftime('%Y-%m-%d'))
 997                 renew_date = self.str_to_date(record.expiration_date)
 998                 diff_time=int((renew_date-today).days)
 999                 if (diff_time<=0):
1000                     res.append((record.id,0))
1001                 else:
1002                     res.append((record.id,diff_time))
1003             else:
1004                 res.append((record.id,-1))
1005             #else:
1006             #    res.append((record.id,-1))
1007         return dict(res)
1008
1009     def act_renew_contract(self,cr,uid,ids,context=None):
1010         contracts = self.browse(cr,uid,ids,context=context)
1011         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid ,'fleet','act_renew_contract', context)
1012         for element in contracts:
1013             temp = []
1014             temp.append(('default_vehicle_id',element.vehicle_id.id))
1015             temp.append(('default_cost_subtype',element.cost_subtype.id))
1016             temp.append(('default_amount',element.amount))
1017             temp.append(('default_odometer_id',element.odometer_id.id))
1018             temp.append(('default_odometer_unit',element.odometer_unit))
1019             temp.append(('default_insurer_id',element.insurer_id.id))
1020             cost_temp = []
1021             for costs in element.cost_ids:
1022                 cost_temp.append(costs.id)
1023             temp.append(('default_cost_ids',cost_temp))
1024             temp.append(('default_date',time.strftime('%Y-%m-%d')))
1025             temp.append(('default_start_date',str(self.str_to_date(element.expiration_date)+datetime.timedelta(days=1))))
1026             temp.append(('default_purchaser_id',element.purchaser_id.id))
1027             temp.append(('default_ins_ref',element.ins_ref))
1028             temp.append(('default_state','open'))
1029             temp.append(('default_notes',element.notes))
1030             temp.append(('default_cost_frequency',element.cost_frequency))
1031             generated_cost = []
1032             for gen_cost in element.generated_cost_ids:
1033                 generated_cost.append(gen_cost.id)
1034             temp.append(('default_generated_cost_ids',generated_cost))
1035
1036             #compute end date
1037             startdate = self.str_to_date(element.start_date)
1038             enddate = self.str_to_date(element.expiration_date)
1039             diffdate = (enddate-startdate)
1040             newenddate = enddate+diffdate
1041             temp.append(('default_expiration_date',str(newenddate)))
1042
1043             res['context'] = dict(temp)
1044
1045         return res
1046
1047
1048
1049     _name = 'fleet.vehicle.log.contract'
1050     _order='state,expiration_date'
1051     _columns = {
1052         'name' : fields.function(_vehicle_contract_name_get_fnc, type="text", string='Name', store=True),
1053
1054         'start_date' : fields.date('Contract Start Date', required=False, help='Date when the coverage of the contract begins'),
1055         'expiration_date' : fields.date('Contract Expiration Date', required=False, help='Date when the coverage of the contract expirates (by default, one year after begin date)'),
1056         'warning_date' : fields.function(get_warning_date,type='integer',string='Warning Date',store=False),
1057
1058         'insurer_id' :fields.many2one('res.partner', 'Supplier', domain="[('supplier','=',True)]"),
1059         'purchaser_id' : fields.many2one('res.partner', 'Contractor',domain="['|',('customer','=',True),('employee','=',True)]",help='Person to which the contract is signed for'),
1060         'ins_ref' : fields.char('Contract Reference', size=64),
1061         'state' : fields.selection([('open', 'In Progress'), ('closed', 'Terminated')], 'Status', readonly=True, help='Choose wheter the contract is still valid or not'),
1062         #'reminder' : fields.boolean('Renewal Reminder', help="Warn the user a few days before the expiration date of this contract"),
1063         'notes' : fields.text('Terms and Conditions', help='Write here all supplementary informations relative to this contract'),
1064         'odometer_id' : fields.many2one('fleet.vehicle.odometer', 'Odometer', required=False, help='Odometer measure of the vehicle at the moment of this log'),
1065         'odometer' : fields.function(_get_odometer,fnct_inv=_set_odometer,type='char',string='Odometer Value',store=False,help='Odometer measure of the vehicle at the moment of this log'),
1066         'odometer_unit': fields.related('vehicle_id','odometer_unit',type="char",string="Unit",store=False, readonly=True),
1067         'cost_amount': fields.related('cost_id','amount',type="float",string="Amount",store=True, readonly=True),
1068         'cost_generated': fields.float('Recuring Costs Amount',help="Costs paid at regular intervals, depending on the cost frequency. If the cost frequency is set to unique, the cost will be logged at the start date"),
1069         'cost_frequency': fields.selection([('no','No'),('daily', 'Daily'),('weekly','Weekly'),('monthly','Monthly'),('yearly','Yearly')], 'Cost Frequency', help='Frequency of the costs',required=True),
1070         'generated_cost_ids' : fields.one2many('fleet.vehicle.cost', 'contract_id', 'Generated Costs',ondelete='cascade'),
1071     }
1072     _defaults = {
1073         'purchaser_id': lambda self, cr, uid, ctx: uid,
1074         'date' : time.strftime('%Y-%m-%d'),
1075         'start_date' : time.strftime('%Y-%m-%d'),
1076         'state':'open',
1077         'expiration_date' : lambda self,cr,uid,ctx: self.compute_next_year_date(time.strftime('%Y-%m-%d')),
1078
1079     }
1080
1081     def copy(self, cr, uid, id, default=None, context=None):
1082         default = default or {}
1083         current_object = self.browse(cr,uid,id,context)
1084         default['date'] = time.strftime('%Y-%m-%d')
1085         default['start_date'] = time.strftime('%Y-%m-%d')
1086         default['expiration_date'] = self.compute_next_year_date(time.strftime('%Y-%m-%d'))
1087         #default['name'] = current_object.name
1088         default['ins_ref'] = ''
1089         default['state'] = 'open'
1090         default['notes'] = ''
1091         default['date'] = time.strftime('%Y-%m-%d')
1092
1093         #default['odometer'] = current_object.odometer
1094         #default['odometer_unit'] = current_object.odometer_unit
1095         return super(fleet_vehicle_log_contract, self).copy(cr, uid, id, default, context=context)
1096
1097     def contract_close(self, cr, uid, ids, *args):
1098         self.write(cr, uid, ids, {'state': 'closed'})
1099         return True
1100
1101     def contract_open(self, cr, uid, ids, *args):
1102         self.write(cr, uid, ids, {'state': 'open'})
1103         return True
1104     def create(self, cr, uid, data, context=None):
1105         data['cost_type'] = 'contract'
1106         cost_id = super(fleet_vehicle_log_contract, self).create(cr, uid, data, context=context)
1107         return cost_id
1108
1109
1110 ############################
1111 ############################
1112 #Vehicle.log.contract.state class
1113 ############################
1114 ############################
1115
1116 class fleet_contract_state(osv.Model):
1117     _name = 'fleet.contract.state'
1118     _columns = {
1119         'name':fields.char('Contract Status',size=32),
1120     }