X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/34511b8161f4fa25634a9659137a1143a1511c21..873c22275a94e60243469f1b7f0eedb7b2e76f39:/catalogue/models.py

diff --git a/catalogue/models.py b/catalogue/models.py
index 4af4810..55ee821 100644
--- a/catalogue/models.py
+++ b/catalogue/models.py
@@ -168,10 +168,14 @@ class Lesson(models.Model):
     def populate_description(self, wldoc=None, infile=None):
         if wldoc is None:
             wldoc = self.wldocument(infile)
-        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:
                 from lxml import etree
-                self.description = etree.tostring(nagl.getnext(),
+                self.description = etree.tostring(header.getnext(),
                         method='text', encoding='unicode').strip()
                 self.save()
                 return