295 lines
16 KiB
Python
295 lines
16 KiB
Python
# -*- coding: utf-8 -*-
|
||
import logging
|
||
from odoo import api, fields, models, tools, exceptions
|
||
import datetime
|
||
from dateutil.relativedelta import relativedelta
|
||
from werkzeug import urls
|
||
import functools
|
||
import base64
|
||
|
||
_logger = logging.getLogger(__name__)
|
||
|
||
try:
|
||
from jinja2.sandbox import SandboxedEnvironment
|
||
|
||
mako_template_env = SandboxedEnvironment(
|
||
block_start_string="<%",
|
||
block_end_string="%>",
|
||
variable_start_string="${",
|
||
variable_end_string="}",
|
||
comment_start_string="<%doc>",
|
||
comment_end_string="</%doc>",
|
||
line_statement_prefix="%",
|
||
line_comment_prefix="##",
|
||
trim_blocks=True, # do not output newline after blocks
|
||
autoescape=True, # XML/HTML automatic escaping
|
||
)
|
||
mako_template_env.globals.update({
|
||
'str': str,
|
||
'quote': urls.url_quote,
|
||
'urlencode': urls.url_encode,
|
||
'datetime': datetime,
|
||
'len': len,
|
||
'abs': abs,
|
||
'min': min,
|
||
'max': max,
|
||
'sum': sum,
|
||
'filter': filter,
|
||
'reduce': functools.reduce,
|
||
'map': map,
|
||
'round': round,
|
||
'relativedelta': lambda *a, **kw: relativedelta(*a, **kw),
|
||
})
|
||
except ImportError:
|
||
_logger.warning("jinja2 not available, templating features will not work!")
|
||
|
||
class DmsDocumentChoiseTemplate(models.TransientModel):
|
||
_name = 'dms.choise_template'
|
||
_description = 'mklab_dms_document_choise_template'
|
||
|
||
temp_id = fields.Many2one(comodel_name='dms.template', string='Шаблон', domain="['|', ('company_id', '=', False), ('company_id', '=', company_id)]")
|
||
doc_id = fields.Many2one(comodel_name='dms.document', string='Документ')
|
||
company_id = fields.Many2one(comodel_name='res.company', string='Компания', required=True, index=True, default=lambda self: self.env.company)
|
||
|
||
def get_choise(self):
|
||
for s in self:
|
||
if s.temp_id:
|
||
s.doc_id.text = s.temp_id.text
|
||
|
||
class DmsDocumentTemplate(models.Model):
|
||
_name = 'dms.template'
|
||
_description = 'mklab_dms_document_template'
|
||
|
||
name = fields.Char(string='Номер', required='1', tracking=True) # Номер — для исходящих с использованием нумератора, для входящих вносится вручную
|
||
text = fields.Html(string='Текст', required='1', tracking=True) # Текст — hmtl
|
||
company_id = fields.Many2one(comodel_name='res.company', string='Компания', required=True, index=True, default=lambda self: self.env.company, tracking=True)
|
||
|
||
class DmsDocument(models.Model):
|
||
_name = 'dms.document'
|
||
_description = 'Документы'
|
||
_inherit = ['portal.mixin', 'mail.thread', 'mail.activity.mixin', 'utm.mixin']
|
||
_description = 'mklab_dms_document'
|
||
|
||
def create_pdf(self):
|
||
report_action = self.env.ref('mklab_dms_document.action_report_dms_document').sudo()
|
||
for s in self:
|
||
if s.type_document=='outgoing':
|
||
pdf, _ = report_action._render_qweb_pdf(report_action.report_name, [s.id])
|
||
s.incoming_file = base64.b64encode(pdf)
|
||
s.incoming_file_type = 'pdf'
|
||
if not s.incoming_file and s.incoming_file_type == 'pdf':
|
||
raise exceptions.UserError('Не указан файл для сохранения')
|
||
if not s.incoming_file_other and s.incoming_file_type != 'pdf':
|
||
raise exceptions.UserError('Не указан файл для сохранения')
|
||
if not s.directory_id:
|
||
raise exceptions.UserError('Не указана директория для сохранения файла')
|
||
content = s.incoming_file if s.incoming_file_type == 'pdf' else s.incoming_file_other
|
||
save_file = self.env['dms.file'].sudo().create({
|
||
'name': s.name,
|
||
'directory_id': s.directory_id.id,
|
||
'content': content,
|
||
})
|
||
s.file = save_file
|
||
s.state = 'done'
|
||
|
||
def open_choise_template(self):
|
||
company_id = False
|
||
if self.company_id:
|
||
company_id = self.company_id.id
|
||
return {
|
||
'type': 'ir.actions.act_window',
|
||
'name': "Выберите шаблон",
|
||
'res_model': 'dms.choise_template',
|
||
'target': 'new',
|
||
'view_mode': 'form',
|
||
'context': {'default_doc_id': self.id, 'company_id': company_id},
|
||
}
|
||
|
||
def unlink(self):
|
||
for s in self:
|
||
if s.state=='done':
|
||
raise exceptions.ValidationError('Нельзя удалить документ, записанный в хранилище')
|
||
res = super().unlink()
|
||
return res
|
||
|
||
def get_number(self):
|
||
type_document = str(self.env.context.get('default_type_document'))
|
||
if self.type_document:
|
||
type_document = self.type_document
|
||
if type_document == 'incoming':
|
||
self.name = self.env['ir.sequence'].next_by_code('dms.document_number')
|
||
return self.env['ir.sequence'].next_by_code('dms.document_number')
|
||
elif type_document == 'outgoing':
|
||
self.name = self.env['ir.sequence'].next_by_code('dms.document_number_out')
|
||
return self.env['ir.sequence'].next_by_code('dms.document_number_out')
|
||
elif type_document == 'internal':
|
||
self.name = self.env['ir.sequence'].next_by_code('dms.document_number_internal')
|
||
return self.env['ir.sequence'].next_by_code('dms.document_number_internal')
|
||
|
||
type_document = fields.Selection([('incoming','Входящий документ'),('outgoing','Исходящий документ'),('internal','Внутренний документ')], string='Тип', required="1", tracking=True) # Тип — список выбора (входящее, исходящее, внутреннее)
|
||
name = fields.Char(string='Номер', default=lambda r: r.get_number(), tracking=True) # Номер — для исходящих с использованием нумератора, для входящих вносится вручную
|
||
partner_id = fields.Many2one(comodel_name='res.partner', string='Контакт', required="1", domain="['|', ('company_id', '=', False), ('company_id', '=', company_id)]", tracking=True) # Контакт — от или для кого
|
||
date = fields.Date(string='Дата документа', default=lambda r: datetime.datetime.today(), required="1", tracking=True) # Дата документа
|
||
text = fields.Html(string='Текст') # Текст — hmtl
|
||
incoming_file = fields.Binary(string='Входящий файл (PDF)')
|
||
incoming_file_other = fields.Binary(string='Входящий файл')
|
||
incoming_file_type = fields.Selection([('pdf','PDF'),('other','Другой')], string='Тип входящего файла', default='pdf',tracking=True)
|
||
text_str = fields.Text(string='Текст') # Текст — hmtl
|
||
text_render = fields.Html(string='Текст render') # Текст — hmtl
|
||
file = fields.Many2one(comodel_name='dms.file', string='Связанный файл', domain="['|', ('company_id', '=', False), ('company_id', '=', company_id)]", tracking=True) # Связанный dms.file
|
||
link_model = fields.Many2one(comodel_name='ir.model', string='Связанная модель', tracking=True) # Связанная модель (не обязательно)
|
||
res_id = fields.Integer(string='ID документа в связанной модели', tracking=True) # ID документа в связанной модели (res_id, не обязательно)
|
||
# Эти поля потребуются только для заполнения переменных внутри текста
|
||
print_head = fields.Boolean(string='Печатать шапку', default=False, tracking=True) # Признак печати шапки - булево
|
||
state = fields.Selection([('draft','Черновик'),('done','Записано в хранилище')], string='Статус', default='draft', tracking=True) # Статус — черновик, записано в хранилище
|
||
parent_id = fields.Many2one(comodel_name='dms.document', string='Ссылка на родительский документ', domain="[('id', '!=', id),'|',('partner_id', '=', partner_id),('partner_id.parent_id', '=', partner_id),'|', ('company_id', '=', False), ('company_id', '=', company_id)]", tracking=True) #Ссылка на другой (родительский) документ
|
||
child_ids = fields.One2many(comodel_name='dms.document', inverse_name='parent_id', string='Direct subordinates')
|
||
subordinate_ids = fields.One2many(comodel_name='dms.document', string='Subordinates', compute='_compute_subordinates', compute_sudo=True)
|
||
company_id = fields.Many2one(comodel_name='res.company', string='Компания', required=True, index=True, default=lambda self: self.env.company, tracking=True)
|
||
directory_id = fields.Many2one(
|
||
comodel_name="dms.directory",
|
||
string="Директория хранения",
|
||
domain="[('permission_create', '=', True), '|', ('company_id', '=', False), ('company_id', '=', company_id)]",
|
||
context={'dms_directory_show_path': True},
|
||
ondelete="restrict",
|
||
auto_join=True,
|
||
index=True,
|
||
tracking=True
|
||
#domain="['|', ('company_id', '=', False), ('company_id', '=', company_id)]"
|
||
)
|
||
|
||
child_all_count = fields.Integer(string='Indirect Surbordinates Count', compute='_compute_subordinates', store=False, compute_sudo=True)
|
||
|
||
def _get_subordinates(self, parents=None):
|
||
if not parents:
|
||
parents = self.env[self._name]
|
||
|
||
indirect_subordinates = self.env[self._name]
|
||
parents |= self
|
||
direct_subordinates = self.child_ids - parents
|
||
for child in direct_subordinates:
|
||
child_subordinate = child._get_subordinates(parents=parents)
|
||
indirect_subordinates |= child_subordinate
|
||
return indirect_subordinates | direct_subordinates
|
||
|
||
|
||
@api.depends('child_ids', 'child_ids.child_all_count')
|
||
def _compute_subordinates(self):
|
||
for doc in self:
|
||
doc.subordinate_ids = doc._get_subordinates()
|
||
doc.child_all_count = len(doc.subordinate_ids)
|
||
|
||
|
||
def render_template(self):
|
||
for s in self:
|
||
template = s.text
|
||
template_new = s.text
|
||
model = s.link_model.model
|
||
res_id = s.res_id
|
||
render_result = 'Не удалось обработать текст'
|
||
substrstart = "${"
|
||
substrend = "}"
|
||
substr_new_start = "<"
|
||
substr_new_end = ">"
|
||
flag = True
|
||
start = 0
|
||
s.text_str = ''
|
||
end = len(template)-1
|
||
while flag:
|
||
try: #search for the item
|
||
index = template.index(substrstart,start,end)
|
||
start = index+len(substrstart)
|
||
indexend = template.index(substrend,start,end)
|
||
substr = template[index:indexend+1]
|
||
substr_new_str = substr
|
||
#s.text_str += str(substr)+ '; '
|
||
if substr.find(substr_new_start)>0 and substr.find(substr_new_end)>0:
|
||
new_flag = True
|
||
start_new = 0
|
||
end_new = len(substr)-1
|
||
while new_flag:
|
||
try: #search for the item
|
||
index_new = substr.index(substr_new_start,start_new,end_new)
|
||
start_new = index_new+len(substr_new_start)
|
||
indexend_new = substr.index(substr_new_end,start_new,end_new)
|
||
substr_new = substr[index_new:indexend_new+1]
|
||
#s.text_str += str(index_new) + '-' + str(indexend_new) +':'+str(substr_new)+ '; '
|
||
substr_new_str = substr_new_str.replace(substr_new,'')
|
||
except ValueError:
|
||
new_flag=False
|
||
template_new = template_new.replace(substr,substr_new_str)
|
||
except ValueError:
|
||
flag = False
|
||
parent = ''
|
||
if s.parent_id:
|
||
type_doc_parent = dict(self._fields['type_document'].selection).get(s.parent_id.type_document)
|
||
parent = ' В ответ на {} {} от {}.'.format(type_doc_parent, s.parent_id.name,s.parent_id.date.strftime('%d.%m.%Y'))
|
||
type_doc = dict(self._fields['type_document'].selection).get(s.type_document)
|
||
s.text_str = '{} {} от {}.{}'.format(type_doc, s.name,s.date.strftime('%d.%m.%Y'), parent)
|
||
template = template_new
|
||
#res = re.findall(r"(${})", template)
|
||
try:
|
||
template = mako_template_env.from_string(tools.ustr(template))
|
||
except:
|
||
template = mako_template_env.from_string(tools.ustr(''))
|
||
user = self.env.user
|
||
variables = {
|
||
'user': user
|
||
}
|
||
if model and res_id:
|
||
record = self.env[model].browse(res_id)
|
||
variables['object'] = record
|
||
|
||
try:
|
||
render_result = template.render(variables)
|
||
except Exception:
|
||
_logger.error("Failed to render template %r using values %r" % (template, variables))
|
||
render_result = u""
|
||
if render_result == u"False":
|
||
render_result = u""
|
||
s.text_render = render_result
|
||
return render_result
|
||
|
||
# @api.model
|
||
# def create(self, vals):
|
||
# if 'name' in vals and 'type_document' in vals:
|
||
# if vals.get('name', ('New')) == '/' or vals['name'] == '' or vals['name'] == False:
|
||
# if vals['type_document'] == 'incoming':
|
||
# vals['name'] = self.env['ir.sequence'].next_by_code('dms.document_number')
|
||
# if vals['type_document'] == 'outgoing':
|
||
# vals['name'] = self.env['ir.sequence'].next_by_code('dms.document_number_out')
|
||
# if vals['type_document'] == 'internal':
|
||
# vals['name'] = self.env['ir.sequence'].next_by_code('dms.document_number_internal')
|
||
# for s in self:
|
||
# if not s.name:
|
||
# if s.type_document == 'incoming':
|
||
# raise exceptions.ValidationError(self.env['ir.sequence'].next_by_code('dms.document_number'))
|
||
# s.name = self.env['ir.sequence'].next_by_code('dms.document_number')
|
||
# if s.type_document == 'outgoing':
|
||
# s.name = self.env['ir.sequence'].next_by_code('dms.document_number_out')
|
||
# if s.type_document == 'internal':
|
||
# s.name = self.env['ir.sequence'].next_by_code('dms.document_number_internal')
|
||
# res = super(mklab_dms_document, self).create(vals)
|
||
# return res
|
||
|
||
# def write(self, vals):
|
||
# if 'name' in vals and 'type_document' in vals:
|
||
# if vals.get('name', ('New')) == '/' or vals['name'] == '' or vals['name'] == False:
|
||
# if vals['type_document'] == 'incoming':
|
||
# vals['name'] = self.env['ir.sequence'].next_by_code('dms.document_number')
|
||
# if vals['type_document'] == 'outgoing':
|
||
# vals['name'] = self.env['ir.sequence'].next_by_code('dms.document_number_out')
|
||
# if vals['type_document'] == 'internal':
|
||
# vals['name'] = self.env['ir.sequence'].next_by_code('dms.document_number_internal')
|
||
# if not self.name:
|
||
# if self.type_document == 'incoming':
|
||
# raise exceptions.ValidationError(self.env['ir.sequence'].next_by_code('dms.document_number'))
|
||
# self.name = self.env['ir.sequence'].next_by_code('dms.document_number')
|
||
# if self.type_document == 'outgoing':
|
||
# self.name = self.env['ir.sequence'].next_by_code('dms.document_number_out')
|
||
# if self.type_document == 'internal':
|
||
# self.name = self.env['ir.sequence'].next_by_code('dms.document_number_internal')
|
||
# res = super(mklab_dms_document, self).write(vals)
|
||
# return res
|
||
|