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_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 pad
  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             try:
  43                 myPad.createPad(path)
  44             except urllib2.URLError:
  45                 raise osv.except_osv(_("Error"), _("Pad creation fail, \
  46                 either there is a problem with your pad server URL or with your connection."))
  47
  48             #get attr on the field model
  49             model = self.pool.get(context["model"])
  50             field = model._all_columns[context['field_name']]
  51             real_field = field.column.pad_content_field
  52
  53             #get content of the real field
  54             for record in model.browse(cr, uid, [context["object_id"]]):
  55                 if record[real_field]:
  56                     myPad.setText(path, html2plaintext(record[real_field]))
  57                     #Etherpad for html not functional
  58                     #myPad.setHTML(path, record[real_field])
  59
  60         return {
  61             "server": pad["server"],
  62             "path": path,
  63             "url": url,
  64         }
  65
  66     def pad_get_content(self, cr, uid, url, context=None):
  67         content = ''
  68         if url:
  69             try:
  70                 page = urllib2.urlopen('%s/export/html'%url).read()
  71                 mo = re.search('<body>(.*)</body>',page)
  72                 if mo:
  73                     content = mo.group(1)
  74             except:
  75                 _logger.warning("No url found '%s'.", url)
  76         return content
  77
  78     # TODO
  79     # reverse engineer protocol to be setHtml without using the api key
  80
  81     def write(self, cr, uid, ids, vals, context=None):
  82         self._set_pad_value(cr, uid, vals, context)
  83         return super(pad_common, self).write(cr, uid, ids, vals, context=context)
  84
  85     def create(self, cr, uid, vals, context=None):
  86         self._set_pad_value(cr, uid, vals, context)
  87         return super(pad_common, self).create(cr, uid, vals, context=context)
  88
  89     # Set the pad content in vals
  90     def _set_pad_value(self, cr, uid, vals, context=None):
  91         for k,v in vals.items():
  92             field = self._all_columns[k].column
  93             if hasattr(field,'pad_content_field'):
  94                 vals[field.pad_content_field] = self.pad_get_content(cr, uid, v, context=context)
  95
  96     def copy(self, cr, uid, id, default=None, context=None):
  97         if not default:
  98             default = {}
  99         for k, v in self._all_columns.iteritems():
 100             field = v.column
 101             if hasattr(field,'pad_content_field'):
 102                 pad = self.pad_generate_url(cr, uid, context)
 103                 default[k] = pad.get('url')
 104         return super(pad_common, self).copy(cr, uid, id, default, context)
 105
 106 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: