addons/pad/ir_attachment.py

   1 # -*- coding: utf-8 -*-
   2 from osv import fields, osv
   3 import random
   4 import string
   5
   6 class ir_attachment(osv.osv):
   7     _inherit = 'ir.attachment'
   8
   9     def pad_generate_url(self, cr, uid, model, id):
  10         pad_url_template = self.pool.get('res.users').browse(cr,uid,[uid])[0].company_id.pad_url_template
  11         s = string.ascii_uppercase + string.digits
  12         salt = ''.join([s[random.randint(0, len(s) - 1)] for i in range(8)])
  13         template_vars = {
  14             'db' : cr.dbname,
  15             'model' : model,
  16             'id' : id,
  17             'salt' : salt,
  18             'name' : '',
  19         }
  20         return pad_url_template % template_vars
  21
  22     def pad_get(self, cr, uid, model, id):
  23         attachment = self.search(cr, uid, [('res_model', '=', model), ('res_id', '=', id), ('type', '=', 'url'), ('name', '=', 'Pad')])
  24         if attachment:
  25             return self.read(cr, uid, attachment)[0]['url']
  26         else:
  27             url = self.pad_generate_url(cr, uid, model, id)
  28             self.create(cr, uid, {
  29                 'res_model' : model,
  30                 'res_id' : id,
  31                 'type' : 'url',
  32                 'name' : 'Pad',
  33                 'url' : url,
  34             })
  35             return url
  36
  37 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: