1 files changed, 25 insertions, 6 deletions
diff --git a/python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Lesson.java b/python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Lesson.java
index 38720140caf1..bd91e8ec30af 100644
--- a/python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Lesson.java
+++ b/python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Lesson.java
@@ -4,12 +4,9 @@ import com.google.gson.annotations.Expose;
 import com.intellij.psi.PsiDirectory;
 import org.jetbrains.annotations.NotNull;
 
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
-public class Lesson {
+public class Lesson implements Comparable{
   @Expose public String name;
   @Expose public List<Task> task_list = new ArrayList<Task>();
 
@@ -27,11 +24,15 @@ public class Lesson {
     task_list.add(task);
   }
 
+  public void setName(String name) {
+    this.name = name;
+  }
+
   public Task getTask(@NotNull final String name) {
     return myTasksMap.get(name);
   }
 
-  public List<Task> getTasklist() {
+  public List<Task> getTaskList() {
     return task_list;
   }
 
@@ -42,4 +43,22 @@ public class Lesson {
   public int getIndex() {
     return myIndex;
   }
+
+  public Map<String, Task> getTasksMap() {
+    return myTasksMap;
+  }
+
+  public void init() {
+    task_list.clear();
+    for (Task task : myTasksMap.values()) {
+      task_list.add(task);
+    }
+    Collections.sort(task_list);
+  }
+
+  @Override
+  public int compareTo(@NotNull Object o) {
+    Lesson lesson = (Lesson) o;
+    return myIndex - lesson.getIndex();
+  }
 }