From a422904a4486f387944e60feb720561bf3cc6888 Mon Sep 17 00:00:00 2001
From: Hao Liu <44379968+TheRealHaoLiu@users.noreply.github.com>
Date: Wed, 23 Oct 2024 15:12:44 -0400
Subject: [PATCH] Add description and label to demo jt

---
 .../commands/create_preload_data.py           | 24 +++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/awx/main/management/commands/create_preload_data.py b/awx/main/management/commands/create_preload_data.py
index eb2ca87e80aa..8ffd995ea388 100644
--- a/awx/main/management/commands/create_preload_data.py
+++ b/awx/main/management/commands/create_preload_data.py
@@ -1,10 +1,11 @@
 # Copyright (c) 2015 Ansible, Inc.
 # All Rights Reserved
 
+import textwrap
 from django.core.management.base import BaseCommand
 from django.db import transaction
 from crum import impersonate
-from awx.main.models import User, Organization, Project, Inventory, CredentialType, Credential, Host, JobTemplate
+from awx.main.models import User, Organization, Project, Inventory, CredentialType, Credential, Host, JobTemplate, Label
 from awx.main.signals import disable_computed_fields
 
 
@@ -50,9 +51,11 @@ def _handle(self):
 
                     ssh_type = CredentialType.objects.filter(namespace='ssh').first()
                     c, _ = Credential.objects.get_or_create(
-                        credential_type=ssh_type, name='Demo Credential', inputs={'username': getattr(superuser, 'username', 'null')}, created_by=superuser
+                        credential_type=ssh_type,
+                        name='Demo Credential',
+                        inputs={'username': getattr(superuser, 'username', 'null')},
+                        created_by=superuser,
                     )
-
                     if superuser:
                         c.admin_role.members.add(superuser)
 
@@ -73,15 +76,28 @@ def _handle(self):
                         created_by=superuser,
                     )
 
+                    label, _ = Label.objects.get_or_create(organization=o, name='demo')
+
+                    jt_description = (
+                        "This template serves as a quick introduction to how Ansible Automation works. "
+                        "It runs a basic automation task, allowing you to see Ansible in action without "
+                        "needing to set up anything complex. You can use this template to understand "
+                        "how job templates operate and test how Ansible interacts with systems in a "
+                        "simple, safe environment."
+                    )
                     jt = JobTemplate.objects.filter(name='Demo Job Template').first()
                     if jt:
                         jt.project = p
+                        jt.description = jt_description
                         jt.inventory = i
                         jt.playbook = 'hello_world.yml'
                         jt.save()
                     else:
-                        jt, _ = JobTemplate.objects.get_or_create(name='Demo Job Template', playbook='hello_world.yml', project=p, inventory=i)
+                        jt, _ = JobTemplate.objects.get_or_create(
+                            name='Demo Job Template', description=jt_description, playbook='hello_world.yml', project=p, inventory=i
+                        )
                     jt.credentials.add(c)
+                    jt.labels.add(label)
 
                     print('Default organization added.')
                     print('Demo Credential, Inventory, and Job Template added.')