Update italy_yaml loader to support get_statement boolean

cms-dev · Aug 5, 2015 · 2e3c7f9 · 2e3c7f9
1 parent 518bcc4
commit 2e3c7f9
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 16 deletions.
diff --git a/cmscontrib/loaders/base_loader.py b/cmscontrib/loaders/base_loader.py
@@ -86,9 +86,11 @@ class TaskLoader(BaseLoader):
     def __init__(self, path, file_cacher):
         super(TaskLoader, self).__init__(path, file_cacher)
 
-    def get_task(self):
+    def get_task(self, get_statement):
         """Produce a Task object.
 
+        get_statement (boolean): whether the statement should be imported.
+
         return (Task): the Task object.
 
         """

diff --git a/cmscontrib/loaders/italy_yaml.py b/cmscontrib/loaders/italy_yaml.py
@@ -344,7 +344,7 @@ def get_user(self):
 
         return User(**args)
 
-    def get_task(self):
+    def get_task(self, get_statement):
         """See docstring in class Loader.
 
         """
@@ -387,21 +387,23 @@ def get_task(self):
         primary_language = load(conf, None, "primary_language")
         if primary_language is None:
             primary_language = 'it'
-        paths = [os.path.join(self.path, "statement", "statement.pdf"),
-                 os.path.join(self.path, "testo", "testo.pdf")]
-        for path in paths:
-            if os.path.exists(path):
-                digest = self.file_cacher.put_file_from_path(
-                    path,
-                    "Statement for task %s (lang: %s)" % (name,
-                                                          primary_language))
-                break
-        else:
-            logger.critical("Couldn't find any task statement, aborting...")
-            sys.exit(1)
-        args["statements"] = [Statement(primary_language, digest)]
 
-        args["primary_statements"] = '["%s"]' % (primary_language)
+        if get_statement:
+            paths = [os.path.join(self.path, "statement", "statement.pdf"),
+                     os.path.join(self.path, "testo", "testo.pdf")]
+            for path in paths:
+                if os.path.exists(path):
+                    digest = self.file_cacher.put_file_from_path(
+                        path,
+                        "Statement for task %s (lang: %s)" % (name,
+                                                              primary_language))
+                    break
+            else:
+                logger.critical("Couldn't find any task statement, aborting...")
+                sys.exit(1)
+            args["statements"] = [Statement(primary_language, digest)]
+
+            args["primary_statements"] = '["%s"]' % (primary_language)
 
         args["attachments"] = []  # FIXME Use auxiliary