Deep audit identified 106 findings; 102 fixed, 4 deferred. Covers 8 areas: - Settings & deploy: env-gated DEBUG/SECRET_KEY, HTTPS headers, gunicorn, celery worker - Auth (registered_user): password write_only, request.data fixes, transaction safety, proper HTTP status codes - Workout app: IDOR protection, get_object_or_404, prefetch_related N+1 fixes, transaction.atomic - Video/scripts: path traversal sanitization, HLS trigger guard, auth on cache wipe - Models (exercise/equipment/muscle/superset): null-safe __str__, stable IDs, prefetch support - Generator views: helper for registered_user lookup, logger.exception, bulk_update, transaction wrapping - Generator core (rules/selector/generator): push-pull ratio, type affinity normalization, modality checks, side-pair exact match, word-boundary regex, equipment cache clearing - Generator services (plan_builder/analyzer/normalizer): transaction.atomic, muscle cache, bulk_update, glutes classification fix Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
95 lines
3.3 KiB
Python
95 lines
3.3 KiB
Python
from django.db import models
|
|
from django.conf import settings
|
|
from exercise.models import *
|
|
from registered_user.models import RegisteredUser
|
|
|
|
WORKOUT_LEVEL = (
|
|
(0, "na"),
|
|
(1, "easy"),
|
|
(2, "moderate"),
|
|
(3, "average"),
|
|
(4, "hard"),
|
|
(5, "death"),
|
|
)
|
|
|
|
# Create your models here.
|
|
class Workout(models.Model):
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
name = models.CharField(max_length=255)
|
|
description = models.CharField(null=True, blank=True, max_length=255)
|
|
registered_user = models.ForeignKey(
|
|
RegisteredUser,
|
|
on_delete=models.CASCADE
|
|
)
|
|
estimated_time = models.FloatField(blank=True, null=True)
|
|
|
|
def __str__(self):
|
|
return str(self.id) + ": " + self.name + " - " + (self.description or "") + " - by: " + str(self.registered_user.first_name) + " - on: " + str(self.created_at)
|
|
|
|
class WorkoutExercise(models.Model):
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
workout = models.ForeignKey(
|
|
Workout,
|
|
on_delete=models.CASCADE,
|
|
related_name='workout_exercise_workout'
|
|
)
|
|
exercise = models.ForeignKey(
|
|
Exercise,
|
|
on_delete=models.CASCADE,
|
|
related_name='workout_exercise_exercise'
|
|
)
|
|
weight = models.IntegerField(null=True, blank=True)
|
|
reps = models.IntegerField(null=True, blank=True)
|
|
duration = models.IntegerField(null=True, blank=True)
|
|
|
|
def __str__(self):
|
|
return self.workout.name + " : " + self.exercise.name
|
|
|
|
def duration_audio(self):
|
|
if self.duration is not None:
|
|
return str(settings.MEDIA_URL) + "quantities_audio/" + "for_"+str(self.duration)+"_seconds.m4a"
|
|
|
|
def weight_audio(self):
|
|
if self.weight is not None:
|
|
return str(settings.MEDIA_URL) + "quantities_audio/" + "for_"+str(self.weight)+"_pounds.m4a"
|
|
|
|
class CompletedWorkout(models.Model):
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
registered_user = models.ForeignKey(
|
|
RegisteredUser,
|
|
on_delete=models.CASCADE,
|
|
related_name='completed_workout_user'
|
|
)
|
|
workout = models.ForeignKey(
|
|
Workout,
|
|
on_delete=models.CASCADE,
|
|
related_name='completed_workout_workout'
|
|
)
|
|
difficulty = models.PositiveSmallIntegerField(
|
|
choices=WORKOUT_LEVEL
|
|
)
|
|
total_time = models.IntegerField(null=True, blank=True)
|
|
workout_start_time = models.DateTimeField(null=False, blank=False)
|
|
notes = models.TextField(null=True, blank=True)
|
|
health_kit_workout_uuid = models.CharField(max_length=1000, blank=True, null=True)
|
|
|
|
def __str__(self):
|
|
return self.registered_user.first_name + " : " + self.registered_user.last_name + " : " + self.workout.name + " : " + str(self.difficulty)
|
|
|
|
class PlannedWorkout(models.Model):
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
on_date = models.DateField(auto_now=False,null=False, blank=False)
|
|
workout = models.ForeignKey(
|
|
Workout,
|
|
on_delete=models.CASCADE,
|
|
related_name='planned_workout_workout'
|
|
)
|
|
registered_user = models.ForeignKey(
|
|
RegisteredUser,
|
|
on_delete=models.CASCADE,
|
|
related_name='planned_workout_registered_user'
|
|
) |