addons/pad/pad.py

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