Merge pull request 'DB操作で利用するworkout_repository.pyを実装' (#2) from develop into main
Reviewed-on: #2
This commit is contained in:
commit
1561aff5d9
@ -0,0 +1,31 @@
|
||||
from app.models.workout import Workout
|
||||
|
||||
class WorkoutRepository:
|
||||
def __init__(self):
|
||||
self.workouts = []
|
||||
|
||||
def add_workout(self, workout: Workout):
|
||||
self.workouts.append(workout)
|
||||
|
||||
def get_all_workouts(self):
|
||||
return self.workouts
|
||||
|
||||
def get_workout_by_id(self, workout_id: int):
|
||||
for workout in self.workouts:
|
||||
if workout.id == workout_id:
|
||||
return workout
|
||||
return None
|
||||
|
||||
def update_workout(self, workout_id: int, updated_workout: Workout):
|
||||
for i, workout in enumerate(self.workouts):
|
||||
if workout.id == workout_id:
|
||||
self.workouts[i] = updated_workout
|
||||
return True
|
||||
return False
|
||||
|
||||
def delete_workout(self, workout_id: int):
|
||||
for i, workout in enumerate(self.workouts):
|
||||
if workout.id == workout_id:
|
||||
del self.workouts[i]
|
||||
return True
|
||||
return False
|
Loading…
x
Reference in New Issue
Block a user