Skip to content

Commit b9d40bc

Browse files
committed
[Data]: Django data generation script
1 parent a8e94fa commit b9d40bc

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

app/data/generate_dump_django.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,12 +111,15 @@ def writeDjangoTeach():
111111
"recovery_question": teacher["recovery_question"],
112112
"recovery_answer": teacher["recovery_answer"],
113113
}
114-
user = User.objects.create(**user_data)
114+
user = User.objects.get_or_create(**user_data)[0]
115115
user.save()
116+
dept = Department.objects.get(id=teacher["department_id"])
117+
116118
teacher_data = {
117119
"id": teacher["id"],
118120
"created": teacher["created"],
119121
"user": user,
122+
"department": dept,
120123
}
121124
teacher = Teacher.objects.create(**teacher_data)
122125
teacher.save()

0 commit comments

Comments
 (0)