nautobot 2.0.0a3__py3-none-any.whl → 2.0.0b1__py3-none-any.whl

nautobot/extras/templates/extras/configcontext.html

@@ -67,7 +67,7 @@
                             {% if object.roles.all %}
                                 <ul>
                                     {% for role in object.roles.all %}
-                                        <li><a href="{% url 'dcim:device_list' %}?role={{ role.slug }}">{{ role }}</a></li>
+                                        <li><a href="{% url 'dcim:device_list' %}?role={{ role.name }}">{{ role }}</a></li>
                                     {% endfor %}
                                 </ul>
                             {% else %}

nautobot/extras/templates/extras/configcontextschema.html

@@ -1,9 +1,24 @@
 {% extends 'generic/object_detail.html' %}
+{% load buttons %}
+{% load plugins %}
 {% load helpers %}
 
+{% block buttons %}
+        {% plugin_buttons object %}
+        {% if perms.extras.add_configcontextschema %}
+            {% clone_button object %}
+        {% endif %}
+        {% if perms.extras.change_configcontextschema %}
+            {% edit_button object key="pk" %}
+        {% endif %}
+        {% if perms.extras.delete_configcontextschema %}
+            {% delete_button object key="pk" %}
+        {% endif %}
+{% endblock buttons %}
+
 {% block extra_nav_tabs %}
     <li role="presentation"{% if active_tab == 'validation' %} class="active"{% endif %}>
-        <a href="{% url 'extras:configcontextschema_object_validation' slug=object.slug %}">Validation</a>
+        <a href="{% url 'extras:configcontextschema_object_validation' pk=object.pk %}">Validation</a>
     </li>
 {% endblock extra_nav_tabs %}
 

nautobot/extras/templates/extras/customfield.html

@@ -2,19 +2,6 @@
 {% load buttons %}
 {% load helpers %}
 
-
-{% block buttons %}
-        {% if object.clone_fields and perms.extras.add_customfield %}
-            {% clone_button object %}
-        {% endif %}
-        {% if perms.extras.change_customfield %}
-            {% edit_button object key="pk" %}
-        {% endif %}
-        {% if perms.extras.delete_customfield %}
-            {% delete_button object key="pk" %}
-        {% endif %}
-{% endblock buttons %}
-
 {% block content_left_page %}
             <div class="panel panel-default">
                 <div class="panel-heading">

nautobot/extras/templates/extras/gitrepository.html

@@ -4,14 +4,14 @@
 {% block extra_buttons %}
         {% if perms.extras.change_gitrepository %}
             <form class="form-inline" style="display: inline-block"
-                  method="post" action="{% url 'extras:gitrepository_dryrun' slug=object.slug %}">
+                  method="post" action="{% url 'extras:gitrepository_dryrun' pk=object.pk %}">
                 {% csrf_token %}
                 <button type="submit" class="btn btn-info">
                     <i class="mdi mdi-book-refresh" aria-hidden="true"></i> Dry-Run
                 </button>
             </form>
             <form class="form-inline" style="display: inline-block"
-                  method="post" action="{% url 'extras:gitrepository_sync' slug=object.slug %}">
+                  method="post" action="{% url 'extras:gitrepository_sync' pk=object.pk %}">
                 {% csrf_token %}
                 <button type="submit" class="btn btn-primary">
                     <i class="mdi mdi-source-branch-sync" aria-hidden="true"></i> Sync
@@ -22,7 +22,7 @@
 
 {% block extra_nav_tabs %}
         <li role="presentation"{% if active_tab == 'result' %} class="active"{% endif %}>
-             <a href="{% url 'extras:gitrepository_result' slug=object.slug %}">Synchronization Status</a>
+             <a href="{% url 'extras:gitrepository_result' pk=object.pk %}">Synchronization Status</a>
         </li>
 {% endblock extra_nav_tabs %}
 

nautobot/extras/templates/extras/inc/jobresult.html

@@ -40,6 +40,16 @@
             {% endif %}
             </td>
         </tr>
+        <tr>
+            <td>Return Value</td>
+            <td id="pending-result-loader">
+            {% if result.date_done %}
+                {{ result.result | placeholder }}
+            {% else %}
+                <img src="{% static 'img/ajax-loader.gif' %}">
+            {% endif %}
+            </td>
+        </tr>
     </table>
 </div>
 

nautobot/extras/templates/extras/inc/panel_jobhistory.html

@@ -2,7 +2,7 @@
     {% for result in job_results %}
         <div class="list-group-item">
             <a href="{% url 'extras:jobresult' pk=result.pk %}">
-                {{ result.obj_type.name }} - {{ result.related_name }}
+                {{ result.job_model.name }}
             </a>
             <span class="pull-right" title="{{ result.date_created }}">{% include 'extras/inc/job_label.html' %}</span>
             <br>

nautobot/extras/templates/extras/job.html

@@ -62,30 +62,22 @@
                             </div>
                         {% endif %}
                         {% csrf_token %}
-                        {% if job_form.requires_input %}
-                            <div class="panel panel-default">
-                                <div class="panel-heading">
-                                    <strong>Job Data</strong>
-                                </div>
-                                <div class="panel-body">
-                                    {% render_form job_form %}
-                                </div>
-                                {% if job_model.has_sensitive_variables %}
-                                    <div class="panel-footer">
-                                        <span class="help-block">
-                                            <i class="mdi mdi-information-outline"></i>
-                                            This job may have sensitive variables and can only be executed immediately. This can be changed in the Job edit view to set `has_sensitive_variables` to `False` if appropriate.
-                                        </span>
-                                    </div>
-                                {% endif %}
+                        <div class="panel panel-default">
+                            <div class="panel-heading">
+                                <strong>Job Data</strong>
                             </div>
-                        {% else %}
-                             <div class="alert alert-info">
-                                <i class="mdi mdi-information"></i>
-                                This job does not require any input to run.
+                            <div class="panel-body">
+                                {% render_form job_form %}
                             </div>
-                            {% render_form job_form %}
-                        {% endif %}
+                            {% if job_model.has_sensitive_variables %}
+                                <div class="panel-footer">
+                                    <span class="help-block">
+                                        <i class="mdi mdi-information-outline"></i>
+                                        This job may have sensitive variables and can only be executed immediately. This can be changed in the Job edit view to set `has_sensitive_variables` to `False` if appropriate.
+                                    </span>
+                                </div>
+                            {% endif %}
+                        </div>
                         <div class="panel panel-default job_execution">
                             <div class="panel-heading">
                                 <strong>Job Execution</strong>
@@ -124,6 +116,8 @@
     var requires_approval = {{ job_model.approval_required|yesno:"true,false" }};
     var run_now_text;
     var schedule_text;
+    var dry_run_text = "Run Job Now (DRYRUN)";
+    var schedule_dry_run_text = "Schedule Job (DRYRUN)";
 
     if (requires_approval) {
         run_now_text = "Request to Run Job Now";
@@ -138,10 +132,21 @@
          * Toggle form elements based on the job execution type.
          * If the type is `immediately` (the default), don't show
          * any of the scheduling related elements, else, show them.
+         * If dryrun is selected, the job may run without approval.
          **/
+        var dryrun = false;
+        if ($("#id_dryrun").prop("type") === "checkbox") {
+            dryrun = $("#id_dryrun").prop("checked")
+        } else if ($("#id_dryrun").prop("type") === "hidden") {
+            dryrun = ($("#id_dryrun").val() === "True")
+        }
         if ($("#id__schedule_type").val() == "immediately") {
             // Toggle run button
-            $("#id__run").html('<i class="mdi mdi-play"></i> ' + run_now_text);
+            if (dryrun){
+                $("#id__run").html('<i class="mdi mdi-play"></i> ' + dry_run_text);
+            } else {
+                $("#id__run").html('<i class="mdi mdi-play"></i> ' + run_now_text);
+            }
             $("#id__run").addClass("btn-primary");
             $("#id__run").removeClass("btn-warning");
 
@@ -151,7 +156,11 @@
             $("#id__schedule_start_time").parent().parent().hide();
         } else {
             // Toggle run button
-            $("#id__run").html('<i class="mdi mdi-clock"></i> ' + schedule_text);
+            if (dryrun){
+                $("#id__run").html('<i class="mdi mdi-clock"></i> ' + schedule_dry_run_text);
+            } else {
+                $("#id__run").html('<i class="mdi mdi-clock"></i> ' + schedule_text);
+            }
             $("#id__run").addClass("btn-warning");
             $("#id__run").removeClass("btn-primary");
 
@@ -171,11 +180,12 @@
     $(document).ready(function() {
         {% if not job_model.has_sensitive_variables %}
             $("#id__schedule_type").change(toggleExecutionType);
-            toggleExecutionType();
         {% else %}
             $("#id__schedule_type").val("immediately");
             $(".job_execution").hide();
         {% endif %}
+        $("#id_dryrun").change(toggleExecutionType);
+        toggleExecutionType();
     });
 </script>
 {% endblock %}

nautobot/extras/templates/extras/job_approval_request.html

@@ -32,9 +32,6 @@
         <li role="presentation" class="active">
             <a href="#run" role="tab" data-toggle="tab" class="active">Approval Request</a>
         </li>
-        <li role="presentation">
-            <a href="#source" role="tab" data-toggle="tab">Source</a>
-        </li>
     </ul>
     <div class="tab-content">
         <div role="tabpanel" class="tab-pane active" id="run">
@@ -66,22 +63,14 @@
                                     </div>
                                 </div>
                             {% endif %}
-                            {% if job_form.requires_input %}
-                                <div class="panel panel-default">
-                                    <div class="panel-heading">
-                                        <strong>Job Data</strong>
-                                    </div>
-                                    <div class="panel-body">
-                                        {% render_form job_form %}
-                                    </div>
+                            <div class="panel panel-default">
+                                <div class="panel-heading">
+                                    <strong>Job Data</strong>
                                 </div>
-                            {% else %}
-                                 <div class="alert alert-info">
-                                    <i class="mdi mdi-information"></i>
-                                    This job does not require any input to run.
+                                <div class="panel-body">
+                                    {% render_form job_form %}
                                 </div>
-                                {% render_form job_form %}
-                            {% endif %}
+                            </div>
                         {% else %}
                             <div class="panel panel-default">
                                 <div class="panel-heading"><strong>Job Data</strong></div>
@@ -110,13 +99,15 @@
                             </div>
                         </div>
                         <div class="pull-right">
-                            <button type="submit" name="_dry_run" id="id__dry_run" class="btn btn-primary"
-                                {% if not perms.extras.run_job or object.job_model is None or not object.job_model.runnable %}
-                                    disabled="disabled"
-                                {% endif %}
-                            >
-                                <i class="mdi mdi-play"></i> Dry Run
-                            </button>
+                            {% if "dryrun" in job_form.fields and object.job_model.supports_dryrun %}
+                                <button type="submit" name="_dry_run" id="id__dry_run" class="btn btn-primary"
+                                    {% if not perms.extras.run_job or object.job_model is None or not object.job_model.runnable %}
+                                        disabled="disabled"
+                                    {% endif %}
+                                >
+                                    <i class="mdi mdi-play"></i> Dry Run
+                                </button>
+                            {% endif %}
                             <button type="submit" name="_approve" id="id__approve" class="btn btn-success"
                                 {% if not perms.extras.approve_job or not perms.extras.change_scheduledjob or object.job_model is None or not object.job_model.runnable %}
                                     disabled="disabled"
@@ -139,11 +130,5 @@
                 </div>
             </div>
         </div>
-        <div role="tabpanel" class="tab-pane" id="source">
-            {% if object.job_model.installed and object.job_model.job_class is not None %}
-                <p><code>{{ object.job_model.job_class.filename }}</code></p>
-                <pre>{{ object.job_model.job_class.source }}</pre>
-            {% endif %}
-        </div>
     </div>
 {% endblock %}

nautobot/extras/templates/extras/job_detail.html

@@ -9,7 +9,7 @@
 
 {% block extra_buttons %}
     {% if perms.extras.run_job %}
-        <a href="{% url 'extras:job_run' slug=object.slug %}" class="btn btn-primary"
+        <a href="{% url 'extras:job_run' pk=object.pk %}" class="btn btn-primary"
            {% if not object.installed or not object.enabled %}disabled="disabled"{% endif %}>
             <i class="mdi mdi-play" aria-hidden="true"></i> Run/Schedule
         </a>
@@ -22,16 +22,6 @@
             <strong>Source Code</strong>
         </div>
         <table class="table table-hover panel-body attr-table">
-            <tr>
-                <td>Source</td>
-                <td>{{ object.source | bettertitle }}</td>
-            </tr>
-            {% if object.source == "git" %}
-                <tr>
-                    <td>Repository</td>
-                    <td>{{ object.git_repository|hyperlinked_object }}</td>
-                </tr>
-            {% endif %}
             <tr>
                 <td>Module</td>
                 <td><code>{{ object.module_name }}</code></td>
@@ -152,14 +142,18 @@
                 </td>
             </tr>
             <tr>
-                <td>Default to Committing Changes</td>
-                <td>{{ object.commit_default | render_boolean }}</td>
+                <td>Supports Dryrun</td>
+                <td colspan="2">{{ object.supports_dryrun | render_boolean }}</td>
+            </tr>
+            <tr>
+                <td>Default to Dryrun</td>
+                <td>{{ object.dryrun_default | render_boolean }}</td>
                 <td>
-                    {% if object.commit_default_override %}
+                    {% if object.dryrun_default_override %}
                         <span class="text-muted">
                             overridden; default is
                             {% if object.installed %}
-                                {{ object.job_class.commit_default | render_boolean }}
+                                {{ object.job_class.dryrun_default | render_boolean }}
                             {% else %}
                                 unknown (not currently installed)
                             {% endif %}
@@ -167,6 +161,10 @@
                     {% endif %}
                 </td>
             </tr>
+            <tr>
+                <td>Read-Only Execution</td>
+                <td colspan="2">{{ object.read_only | render_boolean }}</td>
+            </tr>
             <tr>
                 <td>Hide from Default UI</td>
                 <td>{{ object.hidden | render_boolean }}</td>
@@ -183,22 +181,6 @@
                     {% endif %}
                 </td>
             </tr>
-            <tr>
-                <td>Read-Only Execution</td>
-                <td>{{ object.read_only | render_boolean }}</td>
-                <td>
-                    {% if object.read_only_override %}
-                        <span class="text-muted">
-                            overridden; default is
-                            {% if object.installed %}
-                                {{ object.job_class.read_only | render_boolean }}
-                            {% else %}
-                                unknown (not currently installed)
-                            {% endif %}
-                        </span>
-                    {% endif %}
-                </td>
-            </tr>
             <tr>
                 <td>Has Sensitive Variables</td>
                 <td>{{ object.has_sensitive_variables | render_boolean }}</td>

nautobot/extras/templates/extras/job_edit.html

@@ -6,21 +6,6 @@
     <div class="panel panel-default">
         <div class="panel-heading"><strong>Job Source</strong></div>
         <div class="panel-body">
-            <div class="form-group">
-                <label class="col-md-3 control-label">Source</label>
-                <div class="col-md-9" style="padding-top: 7px">
-                    {{ obj.source | split:"." | first | bettertitle }}
-                    <span class="help-block">Source of the Python code for this job</span>
-                </div>
-            </div>
-            {% if obj.source == "git" %}
-                <div class="form-group">
-                    <label class="col-md-3 control-label">Repository</label>
-                    <div class="col-md-9" style="padding-top: 7px">
-                        <td>{{ obj.git_repository | placeholder }}</td>
-                    </div>
-                </div>
-            {% endif %}
             <div class="form-group">
                 <label class="col-md-3 control-label">Module</label>
                 <div class="col-md-9" style="padding-top: 7px">
@@ -48,6 +33,19 @@
                     </div>
                 </div>
             </div>
+            <div class="form-group">
+                <div class="col-md-9 col-md-offset-3">
+                    <div class="checkbox">
+                        <label>
+                            <input type="checkbox" disabled="true" {% if obj.supports_dryrun %}checked="checked"{% endif %}>
+                            Supports dryrun
+                        </label>
+                        <span class="help-block">
+                            Whether this job supports bypassing approval when running with dryrun argument set to true
+                        </span>
+                    </div>
+                </div>
+            </div>
             <div class="form-group">
                 <div class="col-md-9 col-md-offset-3">
                     <div class="checkbox">
@@ -81,6 +79,7 @@
         <div class="panel-body">
             {% render_field form.enabled %}
             {% include "extras/inc/overridable_field.html" with field=form.name override_field=form.name_override %}
+            {% render_field form.slug %}
             {% include "extras/inc/overridable_field.html" with field=form.grouping override_field=form.grouping_override %}
             {% include "extras/inc/overridable_field.html" with field=form.description override_field=form.description_override %}
         </div>
@@ -89,9 +88,8 @@
         <div class="panel-heading"><strong>Properties</strong></div>
         <div class="panel-body">
             {% include "extras/inc/overridable_field.html" with field=form.approval_required override_field=form.approval_required_override %}
-            {% include "extras/inc/overridable_field.html" with field=form.commit_default override_field=form.commit_default_override %}
+            {% include "extras/inc/overridable_field.html" with field=form.dryrun_default override_field=form.dryrun_default_override %}
             {% include "extras/inc/overridable_field.html" with field=form.hidden override_field=form.hidden_override %}
-            {% include "extras/inc/overridable_field.html" with field=form.read_only override_field=form.read_only_override %}
             {% include "extras/inc/overridable_field.html" with field=form.soft_time_limit override_field=form.soft_time_limit_override %}
             {% include "extras/inc/overridable_field.html" with field=form.time_limit override_field=form.time_limit_override %}
             {% include "extras/inc/overridable_field.html" with field=form.has_sensitive_variables override_field=form.has_sensitive_variables_override %}

nautobot/extras/templates/extras/jobresult.html

@@ -35,18 +35,20 @@
 {% block buttons %}
     {% if perms.extras.run_job %}
         {% if result.job_model and result.task_kwargs %}
-            <a href="{% url 'extras:job_run' slug=result.job_model.slug %}?kwargs_from_job_result={{ result.pk }}"
+            <a href="{% url 'extras:job_run' pk=result.job_model.pk %}?kwargs_from_job_result={{ result.pk }}"
                class="btn btn-success">
                 <span class="mdi mdi-repeat" aria-hidden="true"></span> Re-Run
             </a>
         {% elif result.job_model is not None %}
-            <a href="{% url 'extras:job_run' slug=result.job_model.slug %}"
+            <a href="{% url 'extras:job_run' pk=result.job_model.pk %}"
                class="btn btn-primary">
                 <span class="mdi mdi-play" aria-hidden="true"></span> Run
             </a>
         {% endif %}
     {% endif %}
-    {% export_button %}
+    <a href="{% url 'extras-api:joblogentry-list' %}?job_result={{ result.pk }}&format=csv" class="btn btn-success">
+        <span class="mdi mdi-database-export" aria-hidden="true"></span> Export
+    </a>
     {{ block.super }}
 {% endblock buttons %}
 
@@ -84,20 +86,36 @@
             <div class="col-md-6">
                 <div class="panel panel-default">
                     <div class="panel-heading">
-                        <strong>Job Inputs</strong>
+                        <strong>Job Positional Arguments</strong>
+                    </div>
+                    <div class="panel-body">
+                        {% include 'extras/inc/json_data.html' with data=result.task_args format="json" %}
+                    </div>
+                </div>
+                <div class="panel panel-default">
+                    <div class="panel-heading">
+                        <strong>Job Keyword Arguments</strong>
                     </div>
                     <div class="panel-body">
                         {% include 'extras/inc/json_data.html' with data=result.task_kwargs format="json" %}
                     </div>
                 </div>
+                <div class="panel panel-default">
+                    <div class="panel-heading">
+                        <strong>Job Celery Keyword Arguments</strong>
+                    </div>
+                    <div class="panel-body">
+                        {% include 'extras/inc/json_data.html' with data=result.celery_kwargs format="json" %}
+                    </div>
+                </div>
             </div>
             <div class="col-md-6">
                 <div class="panel panel-default">
                     <div class="panel-heading">
-                        <strong>Output Data</strong>
+                        <strong>Return Data</strong>
                     </div>
                     <div class="panel-body">
-                        {% include 'extras/inc/json_data.html' with data=result.data format="json" %}
+                        {% include 'extras/inc/json_data.html' with data=result.result format="json" %}
                     </div>
                 </div>
                 <div class="panel panel-default">

nautobot/extras/templates/extras/scheduledjob.html

@@ -8,7 +8,7 @@
 {% block buttons %}
     {% plugin_buttons object %}
     {% if user|can_delete:object %}
-        {% delete_button object key='pk' %}
+        {% delete_button object %}
     {% endif %}
 {% endblock buttons %}
 
@@ -119,7 +119,7 @@
               <strong>User Inputs</strong>
           </div>
           <table class="table table-hover panel-body attr-table">
-              {% for key, value in object.kwargs.data.items %}
+              {% for key, value in object.kwargs.items %}
               <tr>
                 <td>{{ labels|get_item:key }}</td>
                 <td>{% if value is None %}–{% else %}<code>{{ value }}</code>{% endif %}</td>

nautobot/extras/templates/extras/secret.html

@@ -1,6 +1,12 @@
 {% extends 'generic/object_detail.html' %}
 {% load helpers %}
 
+{% block extra_buttons %}
+    <a onClick="checkSecret()" class="btn btn-default">
+        Check Secret
+    </a>
+{% endblock extra_buttons %}
+
 {% block content_left_page %}
     <div class="panel panel-default">
         <div class="panel-heading">
@@ -35,3 +41,25 @@
 {% block content_right_page %}
     {% include 'utilities/obj_table.html' with table=groups_table table_template='panel_table.html' heading='Groups containing this secret' %}
 {% endblock content_right_page %}
+
+
+{% block javascript %}
+<script>
+    function checkSecret() {
+        $.ajax({
+            url: "{% url 'extras-api:secret-check' pk=object.pk %}",
+            dataType: 'json',
+            success: function(json) {
+                if(json.result) {
+                    alert("The secret was successfully retrieved.")
+                } else {
+                    alert("Error retrieving secret: \n\n" + json.message)
+                }
+            },
+            error: function(xhr) {
+                alert("Error checking secret: \n\n" + xhr.responseText);
+            }
+        });
+    }
+</script>
+{% endblock javascript %}

nautobot/extras/templatetags/job_buttons.py

@@ -144,6 +144,7 @@ def job_buttons(context, obj):
                 "object": obj,
                 "job": jb.job,
                 "hidden_inputs": hidden_inputs,
+                "disabled": "" if context["user"].has_perms(("extras.run_jobbutton", "extras.run_job")) else "disabled",
             }
             try:
                 text_rendered = render_jinja2(jb.text, button_context)

nautobot/extras/{tests/example_jobs → test_jobs}/api_test_job.py

@@ -1,9 +1,13 @@
 from django.conf import settings
 
-from nautobot.extras.jobs import Job, BooleanVar, IntegerVar, StringVar, ObjectVar
+from nautobot.core.celery import register_jobs
+from nautobot.extras.jobs import Job, BooleanVar, IntegerVar, StringVar, ObjectVar, get_task_logger
 from nautobot.extras.models import Role
 
 
+logger = get_task_logger(__name__)
+
+
 class APITestJob(Job):
     class Meta:
         name = "Job for API Tests"
@@ -15,10 +19,13 @@ class APITestJob(Job):
     var3 = BooleanVar()
     var4 = ObjectVar(model=Role)
 
-    def run(self, data, commit=True):
-        self.log_debug(message=data["var1"])
-        self.log_info(message=data["var2"])
-        self.log_success(message=data["var3"])
-        self.log_warning(message=data["var4"])
+    def run(self, var1, var2, var3, var4):
+        logger.debug(var1)
+        logger.info(var2)
+        logger.warning(var3)
+        logger.error(var4)
 
         return "Job complete"
+
+
+register_jobs(APITestJob)

nautobot/extras/test_jobs/atomic_transaction.py

@@ -0,0 +1,53 @@
+from django.db import transaction
+
+from nautobot.core.celery import register_jobs
+from nautobot.extras.jobs import Job, BooleanVar, get_task_logger
+from nautobot.extras.models import Status
+
+
+logger = get_task_logger(__name__)
+
+
+class SimulatedError(Exception):
+    """I'm just a dummy exception for the purpose of testing."""
+
+
+class TestAtomicDecorator(Job):
+    """
+    Job that uses @transaction.atomic decorator to roll back changes.
+    """
+
+    fail = BooleanVar()
+
+    @transaction.atomic
+    def run(self, fail=False):
+        try:
+            Status.objects.create(name="Test database atomic rollback 1")
+            if fail:
+                raise SimulatedError("simulated failure")
+        except Exception:
+            logger.error("Job failed, all database changes have been rolled back.")
+            raise
+        logger.info("Job succeeded.")
+
+
+class TestAtomicContextManager(Job):
+    """
+    Job that uses `with transaction.atomic()` context manager to roll back changes.
+    """
+
+    fail = BooleanVar()
+
+    def run(self, fail=False):
+        try:
+            with transaction.atomic():
+                Status.objects.create(name="Test database atomic rollback 2")
+                if fail:
+                    raise SimulatedError("simulated failure")
+        except Exception as err:
+            logger.error("Job failed, all database changes have been rolled back.")
+            raise err
+        logger.info("Job succeeded.")
+
+
+register_jobs(TestAtomicContextManager, TestAtomicDecorator)