- for nagl in wldoc.edoc.findall('.//naglowek_rozdzial'):
- if (nagl.text or '').strip() == u'Pomysł na lekcję':
+ if self.type == 'project':
+ lookup = u'Zadanie'
+ else:
+ lookup = u'Pomysł na lekcję'
+ for header in wldoc.edoc.findall('.//naglowek_rozdzial'):
+ if (header.text or '').strip() == lookup: