cubicweb: comparison mixins.py

equal deleted inserted replaced

-:3c281b6f16c6
+:8d246203730a
 if done is None:
 done = set()
 entity = view.cw_rset.get_entity(row, col)
 if entity.eid in done:
 msg = entity._cw._('loop in %(rel)s relation (%(eid)s)') % {
-'rel': entity.tree_attribute,
+'rel': entity.cw_adapt_to('ITree').tree_relation,
 'eid': entity.eid
 }
 return None, msg
 done.add(entity.eid)
 return done, entity
 def call(self, done=None, **kwargs):
 if done is None:
 done = set()
 super(TreeViewMixIn, self).call(done=done, **kwargs)
-def cell_call(self, row, col=0, vid=None, done=None, **kwargs):
+def cell_call(self, row, col=0, vid=None, done=None, maxlevel=None, **kwargs):
+assert maxlevel is None or maxlevel > 0
 done, entity = _done_init(done, self, row, col)
 if done is None:
 # entity is actually an error message
 self.w(u'<li class="badcontent">%s</li>' % entity)
 return
 self.open_item(entity)
 entity.view(vid or self.item_vid, w=self.w, **kwargs)
+if maxlevel is not None:
+maxlevel -= 1
+if maxlevel == 0:
+self.close_item(entity)
+return
 relatedrset = entity.children(entities=False)
-self.wview(self.__regid__, relatedrset, 'null', done=done, **kwargs)
+self.wview(self.__regid__, relatedrset, 'null', done=done,
+maxlevel=maxlevel, **kwargs)
 self.close_item(entity)
 def open_item(self, entity):
 self.w(u'<li class="%s">\n' % entity.__regid__.lower())
 def close_item(self, entity):