addons/pad/pad.py

   1 # -*- coding: utf-8 -*-
   2 from osv import fields, osv
   3 import random
   4 import re
   5 import string
   6 import urllib2
   7 import logging
   8 from tools.translate import _
   9 from openerp.tools import html2plaintext
  10 from py_etherpad import EtherpadLiteClient
  11
  12 _logger = logging.getLogger(__name__)
  13
  14 class pad_common(osv.osv_memory):
  15     _name = 'pad.common'
  16
  17     def pad_generate_url(self, cr, uid, context=None):
  18         company = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id;
  19
  20         pad = {
  21             "server" : company.pad_server,
  22             "key" : company.pad_key,
  23         }
  24
  25         # make sure pad server in the form of http://hostname
  26         if not pad["server"]:
  27             return ''
  28         if not pad["server"].startswith('http'):
  29             pad["server"] = 'http://' + pad["server"]
  30         pad["server"] = pad["server"].rstrip('/')
  31         # generate a salt
  32         s = string.ascii_uppercase + string.digits
  33         salt = ''.join([s[random.randint(0, len(s) - 1)] for i in range(10)])
  34         #path
  35         path = '%s-%s-%s' % (cr.dbname.replace('_','-'), self._name, salt)
  36         # contruct the url
  37         url = '%s/p/%s' % (pad["server"], path)
  38
  39         #if create with content
  40         if "field_name" in context and "model" in context and "object_id" in context:
  41             myPad = EtherpadLiteClient( pad["key"], pad["server"]+'/api')
  42             myPad.createPad(path)
  43
  44             #get attr on the field model
  45             model = self.pool.get(context["model"])
  46             field = model._all_columns[context['field_name']]
  47             real_field = field.column.pad_content_field
  48
  49             #get content of the real field
  50             for record in model.browse(cr, uid, [context["object_id"]]):
  51                 if record[real_field]:
  52                     myPad.setText(path, html2plaintext(record[real_field]))
  53                     #Etherpad for html not functional
  54                     #myPad.setHTML(path, record[real_field])
  55
  56         return {
  57             "server": pad["server"],
  58             "path": path,
  59             "url": url,
  60         }
  61
  62     def pad_get_content(self, cr, uid, url, context=None):
  63         content = ''
  64         if url:
  65             try:
  66                 page = urllib2.urlopen('%s/export/html'%url).read()
  67                 mo = re.search('<body>(.*)</body>',page)
  68                 if mo:
  69                     content = mo.group(1)
  70             except:
  71                 _logger.warning("No url found '%s'.", url)
  72         return content
  73
  74     # TODO
  75     # reverse engineer protocol to be setHtml without using the api key
  76
  77     def write(self, cr, uid, ids, vals, context=None):
  78         self._set_pad_value(cr, uid, vals, context)
  79         return super(pad_common, self).write(cr, uid, ids, vals, context=context)
  80
  81     def create(self, cr, uid, vals, context=None):
  82         self._set_pad_value(cr, uid, vals, context)
  83         return super(pad_common, self).create(cr, uid, vals, context=context)
  84
  85     # Set the pad content in vals
  86     def _set_pad_value(self, cr, uid, vals, context=None):
  87         for k,v in vals.items():
  88             field = self._all_columns[k].column
  89             if hasattr(field,'pad_content_field'):
  90                 vals[field.pad_content_field] = self.pad_get_content(cr, uid, v, context=context)
  91
  92     def copy(self, cr, uid, id, default=None, context=None):
  93         if not default:
  94             default = {}
  95         for k, v in self._all_columns.iteritems():
  96             field = v.column
  97             if hasattr(field,'pad_content_field'):
  98                 pad = self.pad_generate_url(cr, uid, context)
  99                 default[k] = pad['url']
 100         return super(pad_common, self).copy(cr, uid, id, default, context)
 101
 102 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: