| from peewee import * |
|
|
| from src.model.BaseModel import BaseModel |
| from src.model.Crew import Crew |
| from src.model.User import User |
| from src.model.enums.devil_fruit.DevilFruitAbilityType import DevilFruitAbilityType |
|
|
|
|
| class CrewAbility(BaseModel): |
| """ |
| CrewAbility class |
| """ |
|
|
| crew = ForeignKeyField(Crew, backref="crew_abilities") |
| ability_type = SmallIntegerField() |
| value = SmallIntegerField() |
| acquired_method = SmallIntegerField() |
| acquired_user = ForeignKeyField(User, backref="crew_abilities_acquired", null=True) |
| expiration_date = DateTimeField(null=True) |
| was_removed = BooleanField(default=False) |
| removed_user = ForeignKeyField(User, backref="crew_abilities_removed", null=True) |
|
|
| class Meta: |
| db_table = "crew_ability" |
|
|
| def get_ability_type(self): |
| """ |
| Returns the ability type |
| :return: The ability type |
| """ |
| return DevilFruitAbilityType(self.ability_type) |
|
|
| def get_description(self): |
| """ |
| Returns the description of the ability |
| :return: The description |
| """ |
| return self.get_ability_type().get_description() |
|
|
| def get_value_with_sign(self): |
| """ |
| Returns the value with sign |
| :return: The value with sign |
| """ |
| return self.get_ability_type().get_sign() + str(self.value) |
|
|
|
|
| CrewAbility.create_table() |
|
|
