addons/fleet/fleet.py

   1 from itertools import chain
   2 from osv import osv, fields
   3 import time
   4
   5 class fleet_vehicle_model_type(osv.Model):
   6     _name = 'fleet.vehicle.type'
   7     _description = 'Type of the vehicle'
   8     _columns = {
   9         'name' : fields.char('Name', size=32, required=True),
  10     }
  11
  12 class fleet_vehicle_tag(osv.Model):
  13     _name = 'fleet.vehicle.tag'
  14     _columns = {
  15         'name': fields.char('Name', required=True, translate=True),
  16     }
  17
  18 class fleet_vehicle_state(osv.Model):
  19     _name = 'fleet.vehicle.state'
  20     _columns = {
  21         'name': fields.char('Name', required=True),
  22         'sequence': fields.integer('Order',help="Used to order the note stages")
  23     }
  24     _order = 'sequence asc'
  25
  26 class fleet_vehicle_model(osv.Model):
  27
  28     def name_get(self, cr, uid, ids, context=None):
  29         if context is None:
  30             context = {}
  31         if not ids:
  32             return []
  33         reads = self.browse(cr, uid, ids, context=context)
  34         res = []
  35         for record in reads:
  36             name = record.modelname
  37             if record.brand.name:
  38                 name = record.brand.name+' / '+name
  39             res.append((record.id, name))
  40         return res
  41
  42     def _model_name_get_fnc(self, cr, uid, ids, prop, unknow_none, context=None):
  43         res = self.name_get(cr, uid, ids, context=context)
  44         return dict(res)
  45
  46     _name = 'fleet.vehicle.model'
  47     _description = 'Model of a vehicle'
  48
  49     _columns = {
  50         'name' : fields.function(_model_name_get_fnc, type="char", string='Name', store=True),
  51         'modelname' : fields.char('Model name', size=32, required=True),
  52         'brand' : fields.many2one('fleet.vehicle.model.brand', 'Model brand', required=True, help='Brand of the vehicle'),
  53         'vendors': fields.many2many('res.partner','fleet_vehicle_model_vendors','model_id', 'partner_id',string='Vendors',required=False),
  54         'image': fields.related('brand','image',type="binary",string="Logo",store=False)
  55     }
  56
  57 class fleet_vehicle_model_brand(osv.Model):
  58     _name = 'fleet.vehicle.model.brand'
  59     _description = 'Brand model of the vehicle'
  60     _columns = {
  61         'name' : fields.char('Brand Name',size=32, required=True),
  62         'image': fields.binary("Logo",help="This field holds the image used as logo for the brand, limited to 128x128px."),
  63     }
  64
  65 class fleet_vehicle(osv.Model):
  66
  67     _inherit = 'mail.thread'
  68
  69     def name_get(self, cr, uid, ids, context=None):
  70         if context is None:
  71             context = {}
  72         if not ids:
  73             return []
  74         reads = self.browse(cr, uid, ids, context=context)
  75         res = []
  76         for record in reads:
  77             if record.license_plate:
  78                 name = record.license_plate
  79             if record.model_id.modelname:
  80                 name = record.model_id.modelname + ' / ' + name
  81             if record.model_id.brand.name:
  82                 name = record.model_id.brand.name+' / '+ name
  83             res.append((record.id, name))
  84         return res
  85
  86     def _vehicle_name_get_fnc(self, cr, uid, ids, prop, unknow_none, context=None):
  87         res = self.name_get(cr, uid, ids, context=context)
  88         return dict(res)
  89
  90     def act_show_log_services(self, cr, uid, ids, context=None):
  91         """ This opens log view to view and add new log for this vehicle
  92             @return: the service log view
  93         """
  94         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid ,'fleet','act_show_log_services', context)
  95         res['context'] = {
  96             'default_vehicle_id': ids[0]
  97         }
  98         res['domain']=[('vehicle_id','=', ids[0])]
  99         return res
 100
 101     def act_show_log_fuel(self, cr, uid, ids, context=None):
 102         """ This opens log view to view and add new log for this vehicle
 103             @return: the service log view
 104         """
 105         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid ,'fleet','act_show_log_fuel', context)
 106         res['context'] = {
 107             'default_vehicle_id': ids[0]
 108         }
 109         res['domain']=[('vehicle_id','=', ids[0])]
 110         return res
 111
 112     _name = 'fleet.vehicle'
 113     _description = 'Fleet Vehicle'
 114
 115     _columns = {
 116         'name' : fields.function(_vehicle_name_get_fnc, type="char", string='Name', store=True),
 117         'company_id': fields.many2one('res.company', 'Company'),
 118         'license_plate' : fields.char('License plate', size=32, required=True, help='License plate number of the vehicle (ie: plate number for a car)'),
 119         'vin_sn' : fields.char('Chassis Number', size=32, required=False, help='Unique number written on the vehicle motor (VIN/SN number)'),
 120         'driver' : fields.many2one('res.partner', 'Driver',required=False, help='Driver of the vehicle'),
 121         'model_id' : fields.many2one('fleet.vehicle.model', 'Model', required=True, help='Model of the vehicle'),
 122         'log_ids' : fields.one2many('fleet.vehicle.log', 'vehicle_id', 'Other Logs'),
 123         'log_fuel' : fields.one2many('fleet.vehicle.log.fuel','vehicle_id', 'Fuel Logs'),
 124         'log_services' : fields.one2many('fleet.vehicle.log.services','vehicle_id', 'Services Logs'),
 125         'log_insurances' : fields.one2many('fleet.vehicle.log.insurance','vehicle_id', 'Insurances'),
 126         'acquisition_date' : fields.date('Acquisition date', required=False, help='Date when the vehicle has been bought'),
 127         'acquisition_price' : fields.integer('Price', help='Price of the bought vehicle'),
 128         'color' : fields.char('Color',size=32, help='Color of the vehicle'),
 129         'state': fields.many2one('fleet.vehicle.state', 'State', help='Current state of the vehicle', domain='[]'),
 130         'location' : fields.char('Location',size=32, help='Location of the vehicle (garage, ...)'),
 131         'doors' : fields.integer('Number of doors', help='Number of doors of the vehicle'),
 132         'tag_ids' :fields.many2many('fleet.vehicle.tag','vehicle_vehicle_tag_rel','vehicle_tag_id','tag_id','Tags'),
 133
 134         'transmission' : fields.selection([('manual', 'Manual'),('automatic','Automatic')], 'Transmission', help='Transmission Used by the vehicle',required=False),
 135         'fuel_type' : fields.selection([('gasoline', 'Gasoline'),('diesel','Diesel'),('electric','Electric'),('hybrid','Hybrid')], 'Fuel Type', help='Fuel Used by the vehicle',required=False),
 136         'horsepower' : fields.integer('Horsepower',required=False),
 137         'horsepower_tax': fields.float('Horsepower Taxation'),
 138         'power' : fields.integer('Power (kW)',required=False,help='Power in kW of the vehicle'),
 139         'co2' : fields.float('CO2 Emissions',required=False,help='CO2 emissions of the vehicle'),
 140
 141         'image': fields.related('model_id','image',type="binary",string="Logo",store=False)
 142     }
 143     _defaults = {
 144         'doors' : 5,
 145     }
 146
 147     def on_change_model(self, cr, uid, ids, model_id, context=None):
 148         # print "ids: %r" % (ids,)
 149         # print "model_id: %r" % (model_id,)
 150         # print "context: %r" % (context,)
 151         # import logging
 152         # logger = logging.getLogger('fleet.vehicle')
 153         # logger.info('Hello')
 154
 155         # import ipdb
 156         # ipdb.set_trace()
 157
 158         if not model_id:
 159             return {}
 160
 161         model = self.pool.get('fleet.vehicle.model').browse(cr, uid, model_id, context=context)
 162
 163         print "model: %r" % (model.name,)
 164
 165         return {
 166             'value' : {
 167                 'message' : "You have selected this %s model" % (model.name,),
 168             }
 169         }
 170     def create(self, cr, uid, data, context=None):
 171         vehicle_id = super(fleet_vehicle, self).create(cr, uid, data, context=context)
 172         try:
 173             vehicle = self.browse(cr, uid, vehicle_id, context=context)
 174             self.message_post(cr, uid, [vehicle_id], body='Vehicle %s has been added to the fleet!' % (vehicle.license_plate), context=context)
 175         except:
 176             pass # group deleted: do not push a message
 177         return vehicle_id
 178
 179     def write(self, cr, uid, ids, vals, context=None):
 180         vehicle_id = super(fleet_vehicle,self).write(cr, uid, ids, vals, context)
 181         try:
 182             changes = {}
 183             for key,value in vals.items():
 184                 if key == 'license_plate' or key == 'driver':
 185                     changes[key] = value
 186             if len(changes) > 0:
 187                 self.message_post(cr, uid, [vehicle_id], body='Vehicle edited. Changes : '+ str(changes), context=context)
 188                 #self.message_post(cr, uid, [vehicle_id], body='Vehicle edited. Changes : '+ ','.join(chain(*str(changes.items()))), context=context)
 189         except Exception as e:
 190             print e
 191             pass
 192         return vehicle_id
 193
 194 class fleet_vehicle_odometer(osv.Model):
 195     _name='fleet.vehicle.odometer'
 196     _description='Odometer log for a vehicle'
 197
 198     _columns = {
 199         'name' : fields.char('Name',size=64),
 200
 201         'date' : fields.date('Date'),
 202         'value' : fields.float('Odometer Value'),
 203         'unit' : fields.selection([('kilometers', 'Kilometers'),('miles','Miles')], 'Odometer Unit', help='Unit of the measurement',required=False),
 204         'vehicle_id' : fields.many2one('fleet.vehicle', 'Vehicle', required=True),
 205     }
 206     _defaults = {
 207         'date' : time.strftime('%Y-%m-%d')
 208     }
 209
 210 class fleet_vehicle_log_fuel(osv.Model):
 211
 212     _inherits = {'fleet.vehicle.odometer': 'odometer_id'}
 213
 214     def on_change_liter(self, cr, uid, ids, liter, price_per_liter, amount, context=None):
 215
 216         if liter > 0 and price_per_liter > 0:
 217             return {'value' : {'amount' : float(liter) * float(price_per_liter),}}
 218         elif liter > 0 and amount > 0:
 219             return {'value' : {'price_per_liter' : float(amount) / float(liter),}}
 220         elif price_per_liter > 0 and amount > 0:
 221             return {'value' : {'liter' : float(amount) / float(price_per_liter),}}
 222         else :
 223             return {}
 224
 225     def on_change_price_per_liter(self, cr, uid, ids, liter, price_per_liter, amount, context=None):
 226
 227         liter = float(liter);
 228         price_per_liter = float(price_per_liter);
 229         if price_per_liter > 0 and liter > 0:
 230             return {'value' : {'amount' : float(liter) * float(price_per_liter),}}
 231         elif price_per_liter > 0 and amount > 0:
 232             return {'value' : {'liter' : float(amount) / float(price_per_liter),}}
 233         elif liter > 0 and amount > 0:
 234             return {'value' : {'price_per_liter' : float(amount) / float(liter),}}
 235         else :
 236             return {}
 237
 238     def on_change_amount(self, cr, uid, ids, liter, price_per_liter, amount, context=None):
 239
 240         if amount > 0 and liter > 0:
 241             return {'value' : {'price_per_liter' : float(amount) / float(liter),}}
 242         elif amount > 0 and price_per_liter > 0:
 243             return {'value' : {'liter' : float(amount) / float(price_per_liter),}}
 244         elif liter > 0 and price_per_liter > 0:
 245             return {'value' : {'amount' : float(liter) * float(price_per_liter),}}
 246         else :
 247             return {}
 248
 249
 250     _name = 'fleet.vehicle.log.fuel'
 251
 252     _columns = {
 253         'name' : fields.char('Name',size=64),
 254
 255         'liter' : fields.float('Liter'),
 256         'price_per_liter' : fields.float('Price per liter'),
 257         'amount': fields.float('Total price'),
 258         'purchaser_id' : fields.many2one('res.partner', 'Purchaser'),
 259         'inv_ref' : fields.char('Invoice Reference', size=64),
 260         'vendor_id' : fields.many2one('res.partner', 'Vendor', domain="[('supplier','=',True)]"),
 261         'notes' : fields.text('Notes'),
 262     }
 263     _defaults = {
 264         'purchaser_id': lambda self, cr, uid, ctx: uid,
 265     }
 266
 267 class fleet_vehicle_log_services(osv.Model):
 268
 269     _inherits = {'fleet.vehicle.odometer': 'odometer_id'}
 270
 271     _name = 'fleet.vehicle.log.services'
 272     _columns = {
 273
 274         'name' : fields.char('Name',size=64),
 275
 276         'amount' :fields.float('Cost', help="Total cost of the service"),
 277         'service_ids' :fields.many2many('fleet.service.type','vehicle_service_type_rel','vehicle_service_type_id','service_id','Services completed'),
 278         'purchaser_id' : fields.many2one('res.partner', 'Purchaser'),
 279         'inv_ref' : fields.char('Invoice Reference', size=64),
 280         'vendor_id' :fields.many2one('res.partner', 'Vendor', domain="[('supplier','=',True)]"),
 281         'notes' : fields.text('Notes'),
 282     }
 283     _defaults = {
 284         'purchaser_id': lambda self, cr, uid, ctx: uid,
 285     }
 286
 287 class fleet_insurance_type(osv.Model):
 288     _name = 'fleet.insurance.type'
 289     _columns = {
 290         'name': fields.char('Name', required=True, translate=True),
 291     }
 292
 293 class fleet_vehicle_log_insurance(osv.Model):
 294     _inherits = {'fleet.vehicle.odometer': 'odometer_id'}
 295
 296     _name = 'fleet.vehicle.log.insurance'
 297     _columns = {
 298
 299         'name' : fields.char('Name',size=64),
 300
 301         'insurance_type' : fields.many2one('fleet.insurance.type', 'Type', required=False, help='Type of the insurance'),
 302         'start_date' : fields.date('Start date', required=False, help='Date when the coverage of the insurance begins'),
 303         'expiration_date' : fields.date('Expiration date', required=False, help='Date when the coverage of the insurance expirates'),
 304         'price' : fields.float('Price', help="Cost of the insurance for the specified period"),
 305         'insurer_id' :fields.many2one('res.partner', 'Insurer', domain="[('supplier','=',True)]"),
 306         'purchaser_id' : fields.many2one('res.partner', 'Purchaser'),
 307         'ins_ref' : fields.char('Insurance Reference', size=64),
 308         'notes' : fields.text('Notes'),
 309     }
 310     _defaults = {
 311         'purchaser_id': lambda self, cr, uid, ctx: uid,
 312     }
 313
 314 class fleet_service_type(osv.Model):
 315     _name = 'fleet.service.type'
 316     _columns = {
 317         'name': fields.char('Name', required=True, translate=True),
 318     }
 319
 320 #class hr_employee(osv.Model):
 321 #    _inherit = 'hr.employee'
 322
 323 #    _columns = {
 324 #        'vehicle_id' : fields.one2many('fleet.vehicle','driver', 'Vehicle',type="char"),
 325 #        'log_ids' : fields.one2many('fleet.vehicle.log', 'employee_id', 'Logs'),
 326 #    }
 327