web-app/add_exercises.py at main · SmaCoFit/web-app · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
from sqlmodel import Session, select
from database import engine
from models import Exercise, ExerciseCategory, ExerciseDifficulty
import uuid

def add_exercises():
    new_exercises = [
        {
            "name": "Push Up",
            "slug": "push-up",
            "description": "Latihan fundamental untuk membangun kekuatan tubuh bagian atas, khususnya otot dada, bahu, dan triceps.",
            "category": ExerciseCategory.strength,
            "muscleGroups": ["dada", "bahu", "triceps", "chest", "shoulders"],
            "difficulty": ExerciseDifficulty.beginner,
            "instructions": ["Posisi tengkurap dengan tangan selebar bahu", "Turunkan badan hingga dada hampir menyentuh lantai", "Dorong kembali ke posisi awal"],
            "isActive": True
        },
        {
            "name": "Sit Up",
            "slug": "sit-up",
            "description": "Latihan inti (core strength) klasik untuk melatih stabilitas abdomen serta kekuatan otot perut.",
            "category": ExerciseCategory.strength,
            "muscleGroups": ["perut", "inti", "abs", "core"],
            "difficulty": ExerciseDifficulty.beginner,
            "instructions": ["Berbaring telentang dengan lutut ditekuk", "Angkat tubuh bagian atas ke arah lutut", "Turunkan kembali perlahan"],
            "isActive": True
        },
        {
            "name": "Squat",
            "slug": "squat",
            "description": "Turun sampai paha sejajar lantai, lutut tidak melewati jari kaki.",
            "category": ExerciseCategory.strength,
            "muscleGroups": ["kaki", "bokong", "paha depan", "legs", "glutes"],
            "difficulty": ExerciseDifficulty.beginner,
            "instructions": ["Berdiri tegak", "Turunkan pinggul", "Kembali ke posisi awal"],
            "isActive": True
        },
        {
            "name": "Plank",
            "slug": "plank",
            "description": "Tahan posisi tubuh lurus selama waktu yang ditentukan.",
            "category": ExerciseCategory.strength,
            "muscleGroups": ["inti", "perut", "punggung bawah", "core", "abs"],
            "difficulty": ExerciseDifficulty.beginner,
            "instructions": ["Posisi tengkurap", "Angkat badan dengan siku", "Tahan posisi lurus"],
            "isActive": True
        },
        # Exercises below are NOT supported by pose detection — set isActive=False
        {
            "name": "Jumping Jack",
            "slug": "jumping-jack",
            "description": "Lompat sambil membuka kaki dan ayunkan tangan ke atas kepala.",
            "category": ExerciseCategory.cardio,
            "muscleGroups": ["seluruh tubuh", "kardio", "full body", "cardio"],
            "difficulty": ExerciseDifficulty.beginner,
            "instructions": ["Berdiri tegak", "Lompat dan buka kaki", "Kembali ke posisi awal"],
            "isActive": False
        },
        {
            "name": "High Knee",
            "slug": "high-knee",
            "description": "Lari di tempat dengan mengangkat lutut setinggi pinggang.",
            "category": ExerciseCategory.cardio,
            "muscleGroups": ["inti", "kardio", "core", "cardio"],
            "difficulty": ExerciseDifficulty.beginner,
            "instructions": ["Lari di tempat", "Angkat lutut tinggi", "Lakukan dengan cepat"],
            "isActive": False
        },
        {
            "name": "Shoulder Press",
            "slug": "shoulder-press",
            "description": "Angkat beban lurus ke atas kepala, lalu turunkan perlahan.",
            "category": ExerciseCategory.strength,
            "muscleGroups": ["bahu", "triceps", "shoulders"],
            "difficulty": ExerciseDifficulty.beginner,
            "instructions": ["Pegang beban", "Angkat lurus ke atas", "Turunkan perlahan"],
            "isActive": False
        }
    ]

    with Session(engine) as session:
        for ex_data in new_exercises:
            existing = session.exec(select(Exercise).where(Exercise.slug == ex_data["slug"])).first()
            if not existing:
                ex = Exercise(**ex_data)
                session.add(ex)
                print(f"Added {ex_data['name']}")
            else:
                # Update all important fields for existing exercises
                existing.isActive = ex_data["isActive"]
                existing.muscleGroups = ex_data["muscleGroups"]
                existing.difficulty = ex_data["difficulty"]
                existing.description = ex_data["description"]
                existing.instructions = ex_data["instructions"]
                existing.secondaryMuscles = ex_data.get("secondaryMuscles", [])
                session.add(existing)
                print(f"Updated {ex_data['name']} (isActive={ex_data['isActive']})")
        session.commit()
        print("Done!")

if __name__ == '__main__':
    add_exercises()