70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
import random
|
|
from django.db.utils import IntegrityError
|
|
from APIs.models import Student
|
|
|
|
from APIs.constants import BRANCHES, BATCH_CHOICES, DEGREE_CHOICES
|
|
|
|
# To run this script run the following command:
|
|
# python manage.py runscript add_students --script-args <add_type> <num_of_stundents_to_add>
|
|
|
|
|
|
def run(*args):
|
|
|
|
# Throw error if invalid number of arguments passed
|
|
if not args:
|
|
raise ValueError("Invalid number of arguments passed")
|
|
|
|
if args[0] in ("manual", "man") :
|
|
if not args:
|
|
raise ValueError("Invalid number of arguments passed")
|
|
|
|
print("id\troll_no\tname\tbranch\tphone_number\tcpi\tdegree\tbatch")
|
|
for i in range(1, int(args[1])+1):
|
|
details = input()
|
|
details = details.split(",")
|
|
details = [i.strip() for i in details]
|
|
details = [int(details[0]), int(details[1]), details[2], details[3],
|
|
int(details[4]), float(details[5]), details[6], details[7]]
|
|
|
|
student = Student.objects.create(
|
|
id = details[0],
|
|
roll_no = details[1],
|
|
name = details[2],
|
|
branch = details[3],
|
|
phone_number = details[4],
|
|
cpi = details[5],
|
|
degree = details[6],
|
|
batch = details[7],
|
|
|
|
)
|
|
student.save()
|
|
|
|
elif args[0] in ("auto", "automatic"):
|
|
if not args:
|
|
raise ValueError("Invalid number of arguments passed")
|
|
|
|
for i in range(1, int(args[1])+1):
|
|
try:
|
|
student = Student.objects.create(
|
|
id = i-1,
|
|
roll_no = 220010000 + i,
|
|
name = "Student " + str(i),
|
|
branch = random.choice(BRANCHES),
|
|
phone_number = random.randint(1000000000, 9999999999),
|
|
cpi = random.random()*10,
|
|
degree = random.choice(DEGREE_CHOICES)[0],
|
|
batch = random.choice(BATCH_CHOICES)[0],
|
|
)
|
|
student.save()
|
|
except IntegrityError:
|
|
pass
|
|
|
|
elif args[0] in ("del", "delete"):
|
|
# delete students with name starting with Student
|
|
s = Student.objects.filter(name__startswith="Student")
|
|
s.delete()
|
|
|
|
else:
|
|
raise ValueError("Invalid argument passed")
|
|
|