def handle(self, *args, **options):
if Lift.objects.exists() or Program.objects.exists():
print('Program data already loaded...exiting.')
print(ALREADY_LOADED_ERROR_MESSAGE)
return
print("Creating lifts.")
for lift_name in LIFT_NAMES:
lift = Lift(exercise=lift_name)
lift.save()
print("Loading program data.")
for row in DictReader(open('./program_data.csv')):
program = Program()
program.username = row['Username']
program.program = row['Program']
program.version = row['Version']
program.description = row['Description']
raw_submission_date = row['Date']
submission_date = UTC.localize(
datetime.strptime(raw_submission_date, DATETIME_FORMAT))
program.date = submission_date
program.save()
raw_lift_names = row['Lifts']
lift_names = [name for name in raw_lift_names.split(', ') if name]
for lift_name in lift_names:
lift = Lift.objects.get(exercise=lift_name)
program.lifts.add(lift)
program.save()
print("Done.")
def create(self, request):
try:
data = JSONParser().parse(request)
serializer = ProgramSerializer(data=data)
if serializer.is_valid():
pr = Program()
pr.name = serializer.validated_data.get('name')
pr.description = serializer.validated_data.get('description')
pr.level = serializer.validated_data.get('level')
pr.weeks = serializer.validated_data.get('weeks')
pr.save()
predef_routines = []
PF = serializer.validated_data.get('routines')
for c in PF:
predef_routines.append(c.id)
pr.predef_routines.set(predef_routines)
pr.save()
return Response(serializer.data,
status=status.HTTP_201_CREATED)
return Response(serializer.errors,
status=status.HTTP_400_BAD_REQUEST)
except Program.DoesNotExist:
return Response(status=status.HTTP_400_BAD_REQUEST)