Work - request relation, m2m implementation

work.py

 #This file is part of Tryton.  The COPYRIGHT file at the top level of
 #this repository contains the full copyright notices and license terms.
 
 from trytond.model import ModelView, ModelSQL, fields
 from trytond.wizard import Wizard
 from datetime import datetime, timedelta
 from collections import deque, defaultdict
 from heapq import heappop, heappush
-import itertools

[ced, 2009/07/03 10:44:44]

Not used

[bch, 2009/07/03 14:51:47]

On 2009/07/03 10:44:44, ced wrote:
> Not used

Done.

 
 def intfloor(x):
     return int(round(x, 4))
 
 
 class Work(ModelSQL, ModelView):
     _name = 'project.work'
 
     predecessors = fields.Many2Many('project.predecessor_successor',
             'successor', 'predecessor', 'Predecessors',
@@ skipping 72 matching lines @@
         :param context: the context
         :return: a dictionary with all field names as key and
             a dictionary as value with id as key
         '''
         req_ref_obj = self.pool.get('res.request.reference')
 
         res = {}
 
         if 'requests' in names:
             requests = dict((i, []) for i in ids)
-            req_ref_ids = req_ref_obj.search(cursor, user, [
-                    ('reference', 'in', ['project.work,%s' % i for i in ids]),
-                    ], context=context)
-            req_refs = req_ref_obj.browse(cursor, user, req_ref_ids,
-                    context=context)
-            for req_ref in req_refs:
-                _, work_id = req_ref.reference.split(',')

[ced, 2009/07/03 10:44:44]

work_id could be empty

[bch, 2009/07/03 14:51:47]

On 2009/07/03 10:44:44, ced wrote:
> work_id could be empty

Why, isn't a reference always "model,id" ?

[ced, 2009/07/03 15:03:36]

On 2009/07/03 14:51:47, Bertrand Chenal wrote:
> On 2009/07/03 10:44:44, ced wrote:
> > work_id could be empty
> 
> Why, isn't a reference always "model,id" ?

Tryton allow to have empty id.

[ced, 2009/07/03 15:31:20]

But the search clause will ensure that there is an id.
So it is ok

-                requests[int(work_id)].append(req_ref.request.id)
+
+            for i in range(0, len(ids), cursor.IN_MAX):
+                sub_ids = ids[i:i + cursor.IN_MAX]
+
+                req_ref_ids = req_ref_obj.search(cursor, user, [
+                        ('reference', 'in', [
+                                'project.work,%s' % i for i in sub_ids
+                                ]
+                         ),
+                        ], context=context)
+                req_refs = req_ref_obj.browse(cursor, user, req_ref_ids,
+                        context=context)
+                for req_ref in req_refs:
+                    _, work_id = req_ref.reference.split(',')
+                    requests[int(work_id)].append(req_ref.request.id)
 
             res['requests'] = requests
 
         if 'duration' in names:
             all_ids = self.search(cursor, user, [
                     ('parent', 'child_of', ids),
                     ('active', '=', True)], context=context) + ids
             all_ids = list(set(all_ids))
 
             works = self.browse(cursor, user, all_ids, context=context)
@@ skipping 49 matching lines @@
 
         return res
 
     def set_function_fields(self, cursor, user, id, name, value, arg,
             context=None):
         request_obj = self.pool.get('res.request')
         req_ref_obj = self.pool.get('res.request.reference')
 
         if name == 'requests':
             work = self.browse(cursor, user, id, context=context)
-            current = dict((req.id, req) for req in work.requests)

[ced, 2009/07/03 10:44:44]

currents

[bch, 2009/07/03 14:51:47]

On 2009/07/03 10:44:44, ced wrote:
> currents

Done.

-            to_unlink = []
-            to_link = []
-            to_delete = []
+            currents = dict((req.id, req) for req in work.requests)
             for v in value:
+                to_unlink = []
+                to_link = []
                 operator = v[0]
+
                 ids = len(v) > 1 and v[1] or []
                 if operator == 'set':

[ced, 2009/07/03 10:44:44]

I think it is better to make the operation directly like in
model/fields/many2many.py

[bch, 2009/07/03 14:51:47]

On 2009/07/03 10:44:44, ced wrote:
> I think it is better to make the operation directly like in
> model/fields/many2many.py

Done.

-                    to_link.extend((i for i in ids if i not in current))
-                    to_unlink.extend((i for i in current if i not in ids))
+                    to_link.extend((i for i in ids if i not in currents))
+                    to_unlink.extend((i for i in currents if i not in ids))
                 elif operator == 'add':
-                    to_link.extend((i for i in ids if i not in current))
+                    to_link.extend((i for i in ids if i not in currents))
                 elif operator == 'unlink':
-                    to_unlink.extend((i for i in ids if i in current))
+                    to_unlink.extend((i for i in ids if i in currents))
                 elif operator == 'unlink_all':
-                    to_unlink.extend(current)
+                    to_unlink.extend(currents)
                 elif operator == 'delete':
-                    to_delete.extend((i for i in ids if i in current))
+                    request_obj.delete(cursor, user, to_delete, context=context)
                 else:
                     raise Exception('Operation not supported')
 
-            if to_delete:
-                request_obj.delete(cursor, user, to_delete, context=context)
-            print to_unlink

[ced, 2009/07/03 10:44:44]

print statement

[bch, 2009/07/03 14:51:47]

On 2009/07/03 10:44:44, ced wrote:
> print statement

Done.

-            if to_unlink:
                 req_ref_ids = []
                 for i in to_unlink:
-                    request = current[i]
+                    request = currents[i]
                     for ref in request.references:
                         if int(ref.reference.split(',')[1]) == id:
                             req_ref_ids.append(ref.id)
-                print req_ref_ids

[ced, 2009/07/03 10:44:44]

print statement

[bch, 2009/07/03 14:51:47]

On 2009/07/03 10:44:44, ced wrote:
> print statement

Done.

-                req_ref_obj.delete(cursor, user, req_ref_ids, context=context)
-            for i in to_link:
-                req_ref_obj.create(cursor, user, {
-                        'request': i,
-                        'reference': 'project.work,%s' % id,
-                        }, context=context)
+                req_ref_obj.delete(cursor, user, req_ref_ids,
+                        context=context)
+
+                for i in to_link:
+                    req_ref_obj.create(cursor, user, {
+                            'request': i,
+                            'reference': 'project.work,%s' % id,
+                            }, context=context)
             return
-
 
         fun_fields = ('actual_start_date', 'actual_finish_date',
                       'constraint_start_date', 'constraint_finish_date')
         db_fields = ('actual_start_time', 'actual_finish_time',
                      'constraint_start_time', 'constraint_finish_time')
         for fun_field, db_field in zip(fun_fields, db_fields):
             if fun_field == name:
                 self.write(cursor, user, id, {db_field: value}, context=context)
                 break
 
@@ skipping 415 matching lines @@
         res = super(Work, self).delete(cursor, user, ids, context=context)
 
         for work_id in to_update:
             self.reset_leveling(cursor, user, work_id, context=context)
             self.compute_dates(cursor, user, work_id, context=context)
 
         return res
 Work()
 
 
+
 class PredecessorSuccessor(ModelSQL):
     'Predecessor - Successor'
     _name = 'project.predecessor_successor'
     _description = __doc__
 
     predecessor = fields.Many2One('project.work', 'Predecessor',
             ondelete='CASCADE', required=True, select=1)
     successor = fields.Many2One('project.work', 'Successor',
             ondelete='CASCADE', required=True, select=1)
 
@@ skipping 68 matching lines @@
                 },
             },
         }
 
     def _leveling(self, cursor, user, data, context=None):
         work_obj = self.pool.get('project.work')
         work_obj.create_leveling(cursor, user, data['id'], context=context)
         return {}
 
 Leveling()