200010030
/
cdc-placement-website-backend
mirror of https://github.com/CDC-IITDH/cdc-placement-website-backend
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Using constants from APIs/constants.py now

This commit is contained in:
Challenger 2023-06-09 00:14:56 +05:30
parent 609eb1a159
commit 70711888ea
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
CDC_Backend/scripts/add_students.py
View File

@ -2,9 +2,12 @@ import random
from django.db.utils import IntegrityError from django.db.utils import IntegrityError
from APIs.models import Student from APIs.models import Student
from APIs.constants import BRANCHES, BATCH_CHOICES, DEGREE_CHOICES
# To run this script run the following command: # To run this script run the following command:
# python manage.py runscript add_students --script-args <add_type> <num_of_stundents_to_add> # python manage.py runscript add_students --script-args <add_type> <num_of_stundents_to_add>
def run(*args): def run(*args):
# Throw error if invalid number of arguments passed # Throw error if invalid number of arguments passed
@ -39,11 +42,11 @@ def run(*args):
id = i-1, id = i-1,
roll_no = 220010000 + i, roll_no = 220010000 + i,
name = "Student " + str(i), name = "Student " + str(i),
branch = random.choice(["CSE", "EE", "MECH", "CHEM", "CIVIL", "EP", "BSMS", "MNC"]), branch = random.choice(BRANCHES),
phone_number = random.randint(1000000000, 9999999999), phone_number = random.randint(1000000000, 9999999999),
cpi = random.random()*10, cpi = random.random()*10,
degree = "bTech", degree = random.choice(DEGREE_CHOICES)[0],
batch = random.randint(2018, 2023) batch = random.choice(BATCH_CHOICES)[0],
) )
student.save() student.save()
except IntegrityError: except IntegrityError: