From b57f68a1b00194881f8578fd7900e58db7ff30b2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Agathe=20Moll=C3=A9?= <agathe.molle@savoirfairelinux.com>
Date: Wed, 10 Jun 2015 15:48:06 -0400
Subject: [PATCH] [FIX] Factorize and clean

---
 super_calendar/models/super_calendar.py      | 144 ++++++++++---------
 super_calendar/views/super_calendar_view.xml |  18 ++-
 2 files changed, 82 insertions(+), 80 deletions(-)

diff --git a/super_calendar/models/super_calendar.py b/super_calendar/models/super_calendar.py
index 536fb30d9..149c3434b 100644
--- a/super_calendar/models/super_calendar.py
+++ b/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(
diff --git a/super_calendar/views/super_calendar_view.xml b/super_calendar/views/super_calendar_view.xml
index 4a482736e..1c1623a70 100644
--- a/super_calendar/views/super_calendar_view.xml
+++ b/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"/>