diff --git a/SAS/TMSS/src/tmss/tmssapp/populate.py b/SAS/TMSS/src/tmss/tmssapp/populate.py index 705997d3df7c71d180a5e1c39dfa7491faa42d61..2651767c7660b97652efa499a6746f20aa4728af 100644 --- a/SAS/TMSS/src/tmss/tmssapp/populate.py +++ b/SAS/TMSS/src/tmss/tmssapp/populate.py @@ -59,13 +59,47 @@ def _populate_task_draft_example(): :return: """ try: - from datetime import datetime + from datetime import datetime, timezone from lofar.sas.tmss.tmss.tmssapp import models from lofar.sas.tmss.test.tmss_test_data_django_models import TaskDraft_test_data if isTestEnvironment() or isDevelopmentEnvironment(): + for nr in range(0, 18): + models.Cycle.objects.create(name="Cycle %s" % nr, + description="Lofar Cycle %s" % nr, + start=datetime(2013+nr//2, 6 if nr%2==0 else 11, 1, 0, 0, 0, 0, tzinfo=timezone.utc), + stop=datetime(2013+(nr+1)//2, 6 if nr%2==1 else 11, 1, 0, 0, 0, 0, tzinfo=timezone.utc), + number=nr, + standard_hours=0, # TODO: fill in cycle hours + expert_hours=0, + filler_hours=0) + + tmss_project = models.Project.objects.create(cycle=models.Cycle.objects.get(number=14), + name="TMSS-Commissioning", + description="Project for all TMSS tests and commissioning", + priority=1, + can_trigger=False, + private_data=True, + expert=True, + filler=False) + + scheduling_set = models.SchedulingSet.objects.create(name="UC1 test set", + description="UC1 test set", + project=tmss_project) + + requirements_template = models.SchedulingUnitTemplate.objects.create(name="UC1 test scheduling unit template", + description="UC1 test scheduling unit template", + version="0.1", + schema={}) + + scheduling_unit_draft = models.SchedulingUnitDraft.objects.create(name="UC1 test scheduling unit", + description="UC1 test scheduling unit", + scheduling_set=scheduling_set, + requirements_template=requirements_template, + requirements_doc=get_default_json_object_for_schema(requirements_template.schema)) + obs_task_template = models.TaskTemplate.objects.get(name='correlator schema') - task_draft_data = TaskDraft_test_data(name="Test Target Observation", specifications_template=obs_task_template) + task_draft_data = TaskDraft_test_data(name="Test Target Observation", specifications_template=obs_task_template, scheduling_unit_draft=scheduling_unit_draft) obs_task_draft = models.TaskDraft.objects.create(**task_draft_data) pipeline_task_template = models.TaskTemplate.objects.get(name='preprocessing schema')