Skip to content

Commit

Permalink
Remove order column from Generation model and use ID column instead
Browse files Browse the repository at this point in the history
  • Loading branch information
phrasmotica committed Jul 19, 2021
1 parent 98eef4d commit 206af8a
Show file tree
Hide file tree
Showing 6 changed files with 15 additions and 26 deletions.
4 changes: 1 addition & 3 deletions data/v2/build.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,9 +178,7 @@ def csv_record_to_objects(info):

def _build_generations():
def csv_record_to_objects(info):
yield Generation(
id=int(info[0]), region_id=int(info[1]), name=info[2], order=info[3]
)
yield Generation(id=int(info[0]), region_id=int(info[1]), name=info[2])

build_generic((Generation,), "generations.csv", csv_record_to_objects)

Expand Down
18 changes: 9 additions & 9 deletions data/v2/csv/generations.csv
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
id,main_region_id,identifier,order
1,1,generation-i,1
2,2,generation-ii,2
3,3,generation-iii,3
4,4,generation-iv,4
5,5,generation-v,5
6,6,generation-vi,6
7,7,generation-vii,7
8,8,generation-viii,8
id,main_region_id,identifier
1,1,generation-i
2,2,generation-ii
3,3,generation-iii
4,4,generation-iv
5,5,generation-v
6,6,generation-vi
7,7,generation-vii
8,8,generation-viii
6 changes: 0 additions & 6 deletions pokemon_v2/migrations/0011_typeefficacypast.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,10 +81,4 @@ class Migration(migrations.Migration):
to="pokemon_v2.Type",
),
),
migrations.AddField(
model_name="generation",
name="order",
field=models.IntegerField(default=1),
preserve_default=False,
),
]
2 changes: 0 additions & 2 deletions pokemon_v2/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -747,8 +747,6 @@ class Generation(HasName):
on_delete=models.CASCADE,
)

order = models.IntegerField()


class GenerationName(IsName, HasGeneration):
pass
Expand Down
3 changes: 1 addition & 2 deletions pokemon_v2/serializers.py
Original file line number Diff line number Diff line change
Expand Up @@ -678,7 +678,6 @@ class Meta:
"main_region",
"moves",
"names",
"order",
"pokemon_species",
"types",
"version_groups",
Expand Down Expand Up @@ -2031,7 +2030,7 @@ def remove_newer_types(self, relations, current_gen):
def type_is_present(self, type, current_gen):
type_obj = Type.objects.get(name=type["name"])
gen_introduced = Generation.objects.get(pk=type_obj.generation.id)
return gen_introduced.order <= current_gen.order
return gen_introduced.id <= current_gen.id

def get_type_pokemon(self, obj):

Expand Down
8 changes: 4 additions & 4 deletions pokemon_v2/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,11 +74,11 @@ def setup_region_name_data(cls, region, name="reg nm"):

# Generation Data
@classmethod
def setup_generation_data(cls, order=1, region=None, name="gen"):
def setup_generation_data(cls, region=None, name="gen"):

region = region or cls.setup_region_data(name="reg for " + name)

generation = Generation.objects.create(region=region, name=name, order=order)
generation = Generation.objects.create(region=region, name=name)
generation.save()

return generation
Expand Down Expand Up @@ -3420,7 +3420,7 @@ def test_type_api(self):
pokemon = self.setup_pokemon_data(name="pkmn for base tp")
pokemon_type = self.setup_pokemon_type_data(pokemon=pokemon, type=type)

generation = self.setup_generation_data(order=1, name="past gen")
generation = self.setup_generation_data(name="past gen")

no_damage_to = self.setup_type_data(
name="no damage to tp", generation=generation
Expand All @@ -3441,7 +3441,7 @@ def test_type_api(self):
name="double damage from tp", generation=generation
)

newer_generation = self.setup_generation_data(order=5, name="newer_generation")
newer_generation = self.setup_generation_data(name="newer_generation")

newer_type = self.setup_type_data(name="newer tp", generation=newer_generation)

Expand Down

0 comments on commit 206af8a

Please sign in to comment.