117 lines
4.7 KiB
Python
117 lines
4.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# OpenERP, Open Source Management Solution
|
|
# Copyright (C) 2010-today OpenERP SA (<http://www.openerp.com>)
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
#
|
|
##############################################################################
|
|
|
|
import tools
|
|
from osv import osv
|
|
from osv import fields
|
|
from tools.translate import _
|
|
|
|
import io, StringIO
|
|
from PIL import Image
|
|
|
|
class mail_group(osv.osv):
|
|
"""
|
|
A mail_group is a collection of users sharing messages in a discussion
|
|
group. Group users are users that follow the mail group, using the
|
|
subscription/follow mechanism of OpenSocial.
|
|
"""
|
|
|
|
_name = 'mail.group'
|
|
_inherit = ['mail.thread']
|
|
|
|
def action_group_join(self, cr, uid, ids, context={}):
|
|
return self.message_subscribe(cr, uid, ids, context=context);
|
|
|
|
def action_group_leave(self, cr, uid, ids, context={}):
|
|
return self.message_unsubscribe(cr, uid, ids, context=context);
|
|
|
|
def onchange_photo_mini(self, cr, uid, ids, value, context=None):
|
|
return {'value': {'photo': value, 'photo_mini': self._photo_resize(cr, uid, value) } }
|
|
|
|
def _set_photo_mini(self, cr, uid, id, name, value, args, context=None):
|
|
self.write(cr, uid, [id], {'photo': value}, context=context)
|
|
return True
|
|
|
|
def _photo_resize(self, cr, uid, photo, context=None):
|
|
image_stream = io.BytesIO(photo.decode('base64'))
|
|
img = Image.open(image_stream)
|
|
img.thumbnail((120, 100), Image.ANTIALIAS)
|
|
img_stream = StringIO.StringIO()
|
|
img.save(img_stream, "JPEG")
|
|
return img_stream.getvalue().encode('base64')
|
|
|
|
def _get_photo_mini(self, cr, uid, ids, name, args, context=None):
|
|
result = {}
|
|
for group in self.browse(cr, uid, ids, context=context):
|
|
if not group.photo:
|
|
result[group.id] = False
|
|
else:
|
|
result[group.id] = self._photo_resize(cr, uid, group.photo, context=context)
|
|
return result
|
|
|
|
def is_subscriber(self, cr, uid, ids, name, args, context=None):
|
|
result = {}
|
|
for id in ids:
|
|
result[id] = self.message_is_subscriber(cr, uid, [id], context=context)
|
|
return result
|
|
|
|
def get_messages_nbr(self, cr, uid, ids, name, args, context=None):
|
|
result = {}
|
|
for id in ids:
|
|
result[id] = self.message_get_messages_nbr(cr, uid, [id], context=context)
|
|
return result
|
|
|
|
def get_discussions_nbr(self, cr, uid, ids, name, args, context=None):
|
|
result = {}
|
|
for id in ids:
|
|
result[id] = self.message_get_discussions_nbr(cr, uid, [id], context=context)
|
|
return result
|
|
|
|
def get_members_nbr(self, cr, uid, ids, name, args, context=None):
|
|
result = {}
|
|
for id in ids:
|
|
result[id] = len(self.message_get_subscribers_ids(cr, uid, [id], context=context))
|
|
return result
|
|
|
|
_columns = {
|
|
'name': fields.char('Name', size=64, required=True),
|
|
'description': fields.text('Description'),
|
|
'responsible_id': fields.many2one('res.users', string='Responsible',
|
|
ondelete='set null', required=True, select=1),
|
|
'public': fields.boolean('Public', help='This group is visible by non members'),
|
|
'photo': fields.binary('Photo'),
|
|
'photo_mini': fields.function(_get_photo_mini, fnct_inv=_set_photo_mini, string='Photo Mini', type="binary",
|
|
store = {
|
|
'mail.group': (lambda self, cr, uid, ids, c={}: ids, ['photo'], 10),
|
|
}),
|
|
'joined': fields.function(is_subscriber, type='boolean', string='Joined'),
|
|
'messages_nbr': fields.function(get_messages_nbr, type='integer', string='Messages count'),
|
|
'discussions_nbr': fields.function(get_discussions_nbr, type='integer', string='Discussions count'),
|
|
'members_nbr': fields.function(get_members_nbr, type='integer', string='Members count'),
|
|
}
|
|
|
|
_defaults = {
|
|
'public': True,
|
|
'responsible_id': (lambda s, cr, uid, ctx: uid),
|
|
}
|
|
|
|
mail_group()
|