keannu125/StudE-Backend
1
0
Fork 0
mirror of https://github.com/lemeow125/StudE-Backend.git synced 2024-11-17 06:19:24 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Auto add courses post-migration based on csv records that we have

Browse source
This commit is contained in:
Keannu Bernasol 2023-07-23 23:15:52 +08:00
parent f550134998
commit 80f8aa5d66
1 changed files with 40 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
stude/courses/models.py
View file

@ -1,3 +1,5 @@
import os
from django.conf import settings
from django.db.models.signals import post_migrate
from django.dispatch import receiver
from django.db import models
@ -14,6 +16,8 @@ class Course(models.Model):
return self.name
'''
# Old Subject migration code
@receiver(post_migrate)
def populate_courses(sender, **kwargs):
if sender.name == 'courses':
@ -26,3 +30,39 @@ def populate_courses(sender, **kwargs):
Course.objects.get_or_create(
name='Bachelor of Science in Data Science', shortname='BSDS')
# Add more predefined records as needed
'''
# Create subjects based on records that we have
@receiver(post_migrate)
def populate_subjects(sender, **kwargs):
if sender.name == 'courses':
root_path = os.path.join(settings.MEDIA_ROOT, 'records')
csv_files = [f for f in os.listdir(root_path) if f.endswith('.csv')]
added_courses = 0
existing_courses = 0
print('Adding courses\n')
for csv_file in csv_files:
# The filename contains coursename
filename = os.path.splitext(csv_file)[0]
# Splitting the filename and constructing the shortname
shortname = ''
for word in filename.split(' '):
if word[0].isupper():
shortname += word[0]
if shortname != None:
# If course already exists with relevant info, skip over it
if (Course.objects.filter(name=filename, shortname=shortname).exists()):
existing_courses += 1
continue
# Else add the course
else:
Course.objects.get_or_create(
name=filename, shortname=shortname)
added_courses += 1
print('Added', added_courses, 'courses')
print(existing_courses, 'existing courses skipped\n')