python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Lesson.java

package org.jetbrains.plugins.coursecreator.format;

import com.google.gson.annotations.Expose;
import com.intellij.psi.PsiDirectory;
import org.jetbrains.annotations.NotNull;

import java.util.*;

public class Lesson implements Comparable{
  @Expose public String name;
  @Expose public List<Task> task_list = new ArrayList<Task>();

  public int myIndex;
  public Map<String, Task> myTasksMap = new HashMap<String, Task>();

  public Lesson() {}

  public Lesson(@NotNull final String name) {
    this.name = name;
  }

  public void addTask(@NotNull final Task task, PsiDirectory taskDirectory) {
    myTasksMap.put(taskDirectory.getName(), task);
    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() {
    return task_list;
  }

  public void setIndex(int index) {
    myIndex = index;
  }

  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();
  }
}