NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-18 05:28:16 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

14438 script model

Browse Source
This commit is contained in:
Arthur 2024-02-05 15:15:01 -08:00
parent 4cb3b63f79
commit fc890e36af
3 changed files with 25 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
netbox/extras/migrations/0107_alter_script_options.py
View File

@ -5,14 +5,15 @@ import django.db.models.deletion
def update_scripts(apps, schema_editor):
ScriptModule = apps.get_model('extras', 'ScriptModule')
from extras.models import ScriptModule
ScriptModuleNew = apps.get_model('extras', 'ScriptModule')
Script = apps.get_model('extras', 'Script')
for module in ScriptModule.objects.all():
for script, cls in module.get_module_scripts:
for script in module.get_module_scripts.keys():
Script.objects.create(
name=script,
script_module=module,
module=ScriptModuleNew.objects.get(file_root=module.file_root, file_path=module.file_path),
)
@ -23,26 +24,16 @@ class Migration(migrations.Migration):
]
operations = [
migrations.AddField(
model_name='script',
name='name',
field=models.CharField(default=None, max_length=79),
preserve_default=False,
),
migrations.AddField(
model_name='script',
name='module',
field=models.ForeignKey(default=None, on_delete=django.db.models.deletion.PROTECT, related_name='scripts', to='extras.scriptmodule'),
preserve_default=False,
),
migrations.AlterField(
model_name='script',
name='id',
field=models.BigAutoField(auto_created=True, primary_key=True, serialize=False),
),
migrations.AlterModelOptions(
name='script',
options={'ordering': ('name', 'pk')},
migrations.CreateModel(
name='Script',
fields=[
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False)),
('name', models.CharField(max_length=79)),
('module', models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, related_name='scripts', to='extras.scriptmodule')),
],
options={
'ordering': ('name', 'pk'),
},
),
migrations.RunPython(
code=update_scripts,

7
netbox/extras/models/scripts.py
View File

@ -32,9 +32,16 @@ class Script(EventRulesMixin, models.Model):
related_name='scripts'
)
def __str__(self):
return self.name
class Meta:
ordering = ('name', 'pk')
@cached_property
def python_class(self):
return self.module.get_module_scripts.get(self.name)
class ScriptModuleManager(models.Manager.from_queryset(RestrictedQuerySet)):

8
netbox/templates/extras/script_list.html
View File

@ -53,15 +53,15 @@
</thead>
<tbody>
{% with jobs=module.get_latest_jobs %}
{% for script_name, script_class in module.scripts.items %}
{% for script in module.scripts.all %}
<tr>
<td>
<a href="{% url 'extras:script' module=module.python_name name=script_name %}" name="script.{{ script_name }}">{{ script_class.name }}</a>
<a href="{% url 'extras:script' module=module.python_name name=script.name %}" name="script.{{ script.name }}">{{ script.python_class.name }}</a>
</td>
<td>
{{ script_class.Meta.description|markdown|placeholder }}
{{ script.python_class.Meta.description|markdown|placeholder }}
</td>
{% with last_result=jobs|get_key:script_class.class_name %}
{% with last_result=jobs|get_key:script.python_class.class_name %}
{% if last_result %}
<td>
<a href="{% url 'extras:script_result' job_pk=last_result.pk %}">{{ last_result.created|annotated_date }}</a>