Merge pull request #168 from ractf/hotfix/ctftime-unannounced-api-update

Add incrementing position field for CTFtime view
ractf · Aug 9, 2021 · 86da7d9 · 86da7d9
2 parents 423b6e6 + f1c1adb
commit 86da7d9
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/leaderboard/serializers.py b/src/leaderboard/serializers.py
@@ -6,8 +6,16 @@
 
 
 class CTFTimeSerializer(serializers.BaseSerializer):
+    position: int = 0
+
     def to_representation(self, instance):
-        return {"team": instance.name, "score": instance.leaderboard_points}
+        # TODO: Use SerializerFields for team, score and position.
+        return {"team": instance.name, "score": instance.leaderboard_points, "pos": self.get_position(instance)}
+
+    def get_position(self, _) -> int:
+        """Return an incrementing field representing leaderboard positions."""
+        self.position += 1
+        return self.position
 
 
 class LeaderboardTeamScoreSerializer(serializers.ModelSerializer):