Browse Source

[FIX] Factorize and clean

pull/139/head
Agathe Mollé 9 years ago
committed by Alejandro Santana
parent
commit
b57f68a1b0
  1. 144
      super_calendar/models/super_calendar.py
  2. 18
      super_calendar/views/super_calendar_view.xml

144
super_calendar/models/super_calendar.py

@ -39,7 +39,7 @@ def _models_get(self):
return [(model.model, model.name) for model in model_ids]
class super_calendar_configurator(models.Model):
class SuperCalendarConfigurator(models.Model):
_logger = logging.getLogger(__name__)
_name = 'super.calendar.configurator'
@ -66,77 +66,81 @@ class super_calendar_configurator(models.Model):
# Rebuild all calendar records
for configurator in configurator_ids:
for line in configurator.line_ids:
current_pool = self.env[line.name.model]
domain = line.domain and safe_eval(line.domain) or []
current_record_ids = current_pool.search(domain)
for cur_rec in current_record_ids:
f_user = line.user_field_id and line.user_field_id.name
f_descr = (line.description_field_id and
line.description_field_id.name)
f_date_start = (line.date_start_field_id and
line.date_start_field_id.name)
f_date_stop = (line.date_stop_field_id and
line.date_stop_field_id.name)
f_duration = (line.duration_field_id and
line.duration_field_id.name)
if (f_user and
cur_rec[f_user] and
cur_rec[f_user]._model._name != 'res.users'):
raise Exception(
_('Error'),
_("The 'User' field of record %s (%s) "
"does not refer to res.users")
% (cur_rec[f_descr], line.name.model))
if (((f_descr and cur_rec[f_descr]) or
line.description_code) and
cur_rec[f_date_start]):
duration = False
if (not line.duration_field_id and
line.date_stop_field_id and
cur_rec[f_date_start] and
cur_rec[f_date_stop]):
date_start = datetime.strptime(
cur_rec[f_date_start],
tools.DEFAULT_SERVER_DATETIME_FORMAT
)
date_stop = datetime.strptime(
cur_rec[f_date_stop],
tools.DEFAULT_SERVER_DATETIME_FORMAT
)
date_diff = (date_stop - date_start)
duration = date_diff.total_seconds() / 3600
elif line.duration_field_id:
duration = cur_rec[f_duration]
if line.description_type != 'code':
name = cur_rec[f_descr]
else:
parse_dict = {'o': cur_rec}
mytemplate = Template(line.description_code)
name = mytemplate.render(**parse_dict)
super_calendar_values = {
'name': name,
'model_description': line.description,
'date_start': cur_rec[f_date_start],
'duration': duration,
'user_id': (
f_user and
cur_rec[f_user] and
cur_rec[f_user].id or
False
),
'configurator_id': configurator.id,
'res_id': line.name.model+','+str(cur_rec['id']),
'model_id': line.name.id,
}
super_calendar_pool.create(super_calendar_values)
self._generate_record_from_line(configurator, line)
self._logger.info('Calendar generated')
return True
class super_calendar_configurator_line(models.Model):
@api.multi
def _generate_record_from_line(self, configurator, line):
super_calendar_pool = self.env['super.calendar']
current_pool = self.env[line.name.model]
domain = line.domain and safe_eval(line.domain) or []
current_record_ids = current_pool.search(domain)
for cur_rec in current_record_ids:
f_user = line.user_field_id and line.user_field_id.name
f_descr = (line.description_field_id and
line.description_field_id.name)
f_date_start = (line.date_start_field_id and
line.date_start_field_id.name)
f_date_stop = (line.date_stop_field_id and
line.date_stop_field_id.name)
f_duration = (line.duration_field_id and
line.duration_field_id.name)
if (f_user and
cur_rec[f_user] and
cur_rec[f_user]._model._name != 'res.users'):
raise Exception(
_('Error'),
_("The 'User' field of record %s (%s) "
"does not refer to res.users")
% (cur_rec[f_descr], line.name.model))
if (((f_descr and cur_rec[f_descr]) or
line.description_code) and
cur_rec[f_date_start]):
duration = False
if (not line.duration_field_id and
line.date_stop_field_id and
cur_rec[f_date_start] and
cur_rec[f_date_stop]):
date_start = datetime.strptime(
cur_rec[f_date_start],
tools.DEFAULT_SERVER_DATETIME_FORMAT
)
date_stop = datetime.strptime(
cur_rec[f_date_stop],
tools.DEFAULT_SERVER_DATETIME_FORMAT
)
date_diff = (date_stop - date_start)
duration = date_diff.total_seconds() / 3600
elif line.duration_field_id:
duration = cur_rec[f_duration]
if line.description_type != 'code':
name = cur_rec[f_descr]
else:
parse_dict = {'o': cur_rec}
mytemplate = Template(line.description_code)
name = mytemplate.render(**parse_dict)
super_calendar_values = {
'name': name,
'model_description': line.description,
'date_start': cur_rec[f_date_start],
'duration': duration,
'user_id': (
f_user and
cur_rec[f_user] and
cur_rec[f_user].id or
False
),
'configurator_id': configurator.id,
'res_id': line.name.model+','+str(cur_rec['id']),
'model_id': line.name.id,
}
super_calendar_pool.create(super_calendar_values)
class SuperCalendarConfiguratorLine(models.Model):
_name = 'super.calendar.configurator.line'
name = fields.Many2one(
@ -196,7 +200,7 @@ class super_calendar_configurator_line(models.Model):
)
class super_calendar(models.Model):
class SuperCalendar(models.Model):
_name = 'super.calendar'
name = fields.Char(

18
super_calendar/views/super_calendar_view.xml

@ -1,9 +1,9 @@
<?xml version="1.0"?>
<openerp>
<data>
<!-- Configurator -->
<record model="ir.ui.view" id="super_calendar_configurator_tree">
<field name="name">super_calendar_configurator_tree</field>
<field name="model">super.calendar.configurator</field>
@ -50,7 +50,7 @@
<field name="description_code" nolabel="1" attrs="{'required':[('description_type','==','code')]}"/>
</group>
</group>
</form>
</field>
<newline/>
@ -58,7 +58,7 @@
</form>
</field>
</record>
<record model="ir.actions.act_window" id="super_calendar_configurator">
<field name="name">Calendar Configurators</field>
<field name="res_model">super.calendar.configurator</field>
@ -66,9 +66,9 @@
<field name="view_mode">tree,form</field>
<field name="view_id" ref="super_calendar_configurator_tree"/>
</record>
<!-- Calendar -->
<record model="ir.ui.view" id="super_calendar_tree">
<field name="name">super_calendar_tree</field>
<field name="model">super.calendar</field>
@ -76,7 +76,6 @@
<tree string="Calendar">
<field name="name"/>
<field name="date_start"/>
<!--<field name="date_stop"/>-->
<field name="duration"/>
<field name="user_id"/>
<field name="configurator_id"/>
@ -95,7 +94,6 @@
<group>
<field name="name" readonly="1"/>
<field name="date_start" readonly="1"/>
<!--<field name="date_stop" readonly="1"/>-->
<field name="duration" readonly="1"/>
<field name="user_id" readonly="1"/>
<field name="configurator_id" readonly="1"/>
@ -139,7 +137,7 @@
</search>
</field>
</record>
<record model="ir.actions.act_window" id="super_calendar_action">
<field name="name">Super Calendar</field>
<field name="res_model">super.calendar</field>
@ -147,7 +145,7 @@
<field name="view_mode">calendar,tree,form</field>
<field name="view_id" ref="super_calendar"/>
</record>
<menuitem id="super_calendar_menu" name="Super Calendar" action="super_calendar_action"/>
<menuitem id="super_calendar_calendar" name="Calendar" parent="super_calendar_menu" />
<menuitem id="super_calendar_calendar_calendar" name="Calendar" parent="super_calendar_calendar" action="super_calendar_action"/>

Loading…
Cancel
Save