1414from django .contrib .contenttypes .models import ContentType
1515from django .utils import timezone
1616from django .db .models import Sum , Count , Q
17+ from django .contrib .admin .utils import NestedObjects
18+ from django .db import DEFAULT_DB_ALIAS
1719from dojo .filters import ProductFilter , ProductFindingFilter , EngagementFilter
1820from dojo .forms import ProductForm , EngForm , DeleteProductForm , DojoMetaDataForm , JIRAPKeyForm , JIRAFindingForm , AdHocFindingForm , \
1921 EngagementPresetsForm , DeleteEngagementPresetsForm
@@ -533,13 +535,6 @@ def delete_product(request, pid):
533535 product = get_object_or_404 (Product , pk = pid )
534536 form = DeleteProductForm (instance = product )
535537
536- from django .contrib .admin .utils import NestedObjects
537- from django .db import DEFAULT_DB_ALIAS
538-
539- collector = NestedObjects (using = DEFAULT_DB_ALIAS )
540- collector .collect ([product ])
541- rels = collector .nested ()
542-
543538 if request .method == 'POST' :
544539 if 'id' in request .POST and str (product .id ) == request .POST ['id' ]:
545540 form = DeleteProductForm (request .POST , instance = product )
@@ -553,6 +548,10 @@ def delete_product(request, pid):
553548 extra_tags = 'alert-success' )
554549 return HttpResponseRedirect (reverse ('product' ))
555550
551+ collector = NestedObjects (using = DEFAULT_DB_ALIAS )
552+ collector .collect ([product ])
553+ rels = collector .nested ()
554+
556555 product_tab = Product_Tab (pid , title = "Product" , tab = "settings" )
557556 return render (request , 'dojo/delete_product.html' ,
558557 {'product' : product ,
@@ -883,13 +882,6 @@ def delete_engagement_presets(request, pid, eid):
883882 preset = get_object_or_404 (Engagement_Presets , id = eid )
884883 form = DeleteEngagementPresetsForm (instance = preset )
885884
886- from django .contrib .admin .utils import NestedObjects
887- from django .db import DEFAULT_DB_ALIAS
888-
889- collector = NestedObjects (using = DEFAULT_DB_ALIAS )
890- collector .collect ([preset ])
891- rels = collector .nested ()
892-
893885 if request .method == 'POST' :
894886 if 'id' in request .POST :
895887 form = DeleteEngagementPresetsForm (request .POST , instance = preset )
@@ -901,6 +893,10 @@ def delete_engagement_presets(request, pid, eid):
901893 extra_tags = 'alert-success' )
902894 return HttpResponseRedirect (reverse ('engagement_presets' , args = (pid ,)))
903895
896+ collector = NestedObjects (using = DEFAULT_DB_ALIAS )
897+ collector .collect ([preset ])
898+ rels = collector .nested ()
899+
904900 product_tab = Product_Tab (pid , title = "Delete Engagement Preset" , tab = "settings" )
905901 return render (request , 'dojo/delete_presets.html' ,
906902 {'product' : product ,
0 commit comments