class Work (models.Model):
    character = models.ForeignKey(Character)
    money = models.IntegerField()
    work_name = models.CharField(max_length=60)
    start = models.DateTimeField(auto_now_add=True)
    finish = models.DateTimeField(blank=True, null=True)
    duration = models.IntegerField()
    next = models.CharField(max_length=40)
    studying = models.CharField(max_length=40)
    collected = models.BooleanField(default=False)

    @classmethod
    def make_work(cls, number, character):
        if number > '1':
            return Work_for_money(number)
        else:
            return Work_to_learn(number)

class Work_for_money(Work):
    def __init__(self, number):
        if number == '0':
            self.name="Pomagasz Hagridowi"
            self.duration = 30
            self.money = 3
            self.energy = 5
        elif number == '1':
            self.name="Pomagasz Hagridowi"
            self.duration = 65
            self.money = 10
            self.energy = 15
        self.next = "/courtyard/hagrid/2"
        self.studying=""


class Work_to_learn(Work):
    def __init__(self, number, character):
        if number == '2':
            self.skill = "Transmutacja"
            self.name = "Studiujesz transmutacje"
        elif number == '3':
            self.skill = "Eliksiry"
            self.name = "Studiujesz Eliksiry"
        elif number == '4':
            self.skill = "Zaklecia"
            self.name = "Studiujesz Zaklecia"
        elif number == '5':
            self.skill = "Zielarstwo"
            self.name = "Studiujesz Zielarstwo"
        elif number == '6':
            self.skill = "Czarna magia"
            self.name = "Studiujesz Czarna magie"

        if character.skill_set.filter(skill_name=self.skill).exists():
            self.level = character.skill_set.get(skill_name=self.skill).level
        else:
            self.level = 0

        self.duration = 5
        self.next = "/stairs/library/2/"
        self.money = 0
        self.energy = 3
        self.studying = self.skill