from django.db import models from django.contrib.auth.models import User from django.contrib.auth import get_user_model User = get_user_model() class GameHistory(models.Model): player_1_id = models.ForeignKey(User, on_delete=models.CASCADE, related_name='player_1') player_2_id = models.ForeignKey(User, on_delete=models.CASCADE, related_name='player_2') player_1_score = models.IntegerField(default=0) player_2_score = models.IntegerField(default=0) winner_id = models.ForeignKey(User, on_delete=models.CASCADE, related_name='winner') date_played = models.DateTimeField(auto_now_add=True) duration = models.FloatField(default=0.0) class Meta: db_table = 'users_game_history' def __str__(self): return (f'{self.player_1_id} vs {self.player_2_id} ' f'on {self.date_played} ' f'winner is {self.winner_id} ' f'score is {self.player_1_score} - {self.player_2_score}') @property def player_1(self): return {'username': self.player_1_id.username, 'score': self.player_1_score} @property def player_2(self): return {'username': self.player_2_id.username, 'score': self.player_2_score} @property def winner(self): return {'username': self.winner_id.username} class Stats(models.Model): player_id = models.ForeignKey(User, on_delete=models.CASCADE, related_name='player') wins = models.IntegerField(default=0) losses = models.IntegerField(default=0) win_rate = models.FloatField(default=0.0) total_games_played = models.IntegerField(default=0) total_hours_played = models.FloatField(default=0.0) goal_scored = models.IntegerField(default=0) goal_conceded = models.IntegerField(default=0) class Meta: db_table = 'users_stats' def __str__(self): return (f'{self.player_id} wins: {self.wins}, losses: {self.losses}, win rate: {self.win_rate:.2f}, ' f'total games: {self.total_games_played}, total hours played: {self.total_hours_played:.2f}, ' f'goal scored: {self.goal_scored}, goal conceded: {self.goal_conceded}')