# scripts/create_NEW_and_CANCELED_tasks.py

# Author: Justin Chen

# Django
from django.contrib.auth.models import User

# fetchcore
from fetchcore_server.api.models import Robot, RobotFootprint, Task
from django.conf import settings


def run():
    print('~Create NEW and CANCELED tasks started~')

    Robot.objects.all().delete()
    Task.objects.all().delete()
    print('All robots and tasks deleted')

    robot_name = 'freight0008'
    try:
        user = User.objects.get(username=robot_name)
    except User.DoesNotExist:
        user = User.objects.create(username=robot_name, password='robotics')
        print('User ' +  robot_name + 'created')

    robot_footprint = RobotFootprint.objects.get(name='FREIGHT100')
    robot = Robot.objects.create(
        name=user,
        footprint=robot_footprint,
    )
    print('Robot ' + robot_name + ' created')

    new_tasks = []
    for i in range(1, 5001):
        new_tasks.append(Task(
            type='RUNSCRIPT_TASK',
            name='NEW task ' + str(i) + ' pointing to ' + robot_name,
            robot=robot,
            status='NEW',
        ))
    Task.objects.bulk_create(new_tasks)
    print('5000 NEW tasks pointing to ' + robot_name + ' created')

    canceled_tasks = []
    for i in range(1, 5001):
        canceled_tasks.append(Task(
            type='RUNSCRIPT_TASK',
            name='CANCELED task ' + str(i) + ' pointing to ' + robot_name,
            robot=robot,
            status='CANCELED',
        ))
    Task.objects.bulk_create(canceled_tasks)
    print('5000 CANCELED tasks pointing to ' + robot_name + ' created')