Django Administraatori võimsuse vallandamine: Kohandatud administreerimistegevused selgitatud

            # myapp/models.py
from django.db import models

class Product(models.Model):
    name = models.CharField(max_length=200)
    price = models.DecimalField(max_digits=10, decimal_places=2)
    is_discounted = models.BooleanField(default=False)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.name

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest
from .models import Product

@admin.register(Product)
class ProductAdmin(admin.ModelAdmin):
    list_display = ('name', 'price', 'is_discounted', 'created_at')
    list_filter = ('is_discounted', 'created_at')
    search_fields = ('name',)

    # Define the custom admin action function
    def make_discounted(self, request: HttpRequest, queryset: QuerySet):
        updated_count = queryset.update(is_discounted=True)
        self.message_user(
            request,
            f\"{updated_count} toode(t) märgiti edukalt allahinnatuks.\",
            messages.SUCCESS
        )
    make_discounted.short_description = \"Märgi valitud tooted allahinnatuks\"

    # Register the action with the ModelAdmin
    actions = [make_discounted]

            

              
                Copy
              
            
          

            # myapp/models.py (assuming a Post model)
from django.db import models

class Post(models.Model):
    title = models.CharField(max_length=255)
    content = models.TextField()
    status = models.CharField(max_length=20, default='draft', choices=[
        ('draft', 'Draft'),
        ('pending', 'Pending Review'),
        ('approved', 'Approved'),
        ('rejected', 'Rejected'),
    ])
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.title

            

              
                Copy
              
            
          

            # myapp/forms.py
from django import forms

class ConfirmationForm(forms.Form):
    confirm = forms.BooleanField(
        label="Kas olete kindel, et soovite seda tegevust sooritada?",
        required=True,
        widget=forms.HiddenInput # Kuva käsitleme mallis
    )
    _selected_action = forms.CharField(widget=forms.HiddenInput)
    action = forms.CharField(widget=forms.HiddenInput)

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Post
from .forms import ConfirmationForm

@admin.register(Post)
class PostAdmin(admin.ModelAdmin):
    list_display = ('title', 'status', 'created_at')
    list_filter = ('status',)
    search_fields = ('title',)

    def mark_posts_approved(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        # Check if the user confirmed the action
        if 'apply' in request.POST:
            form = ConfirmationForm(request.POST)
            if form.is_valid():
                updated_count = queryset.update(status='approved')
                self.message_user(
                    request,
                    f\"{updated_count} postitus(t) märgiti edukalt kinnitatuks.\",
                    messages.SUCCESS
                )
                return HttpResponseRedirect(request.get_full_path())
        
        # If not confirmed, or GET request, show confirmation page
        else:
            # Store the selected objects' primary keys in a hidden field
            # This is crucial for passing the selection across the confirmation page
            context = self.admin_site.each_context(request)
            context['queryset'] = queryset
            context['form'] = ConfirmationForm(initial={
                '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
                'action': 'mark_posts_approved',
            })
            context['action_name'] = self.mark_posts_approved.short_description
            context['title'] = _("Kinnita tegevus")

            # Render a custom confirmation template
            return render(request, 'admin/confirmation_action.html', context)

    mark_posts_approved.short_description = _("Märgi valitud postitused kinnitatuks")

    actions = [mark_posts_approved]

            

              
                Copy
              
            
          

            {# templates/admin/confirmation_action.html #}
{% extends "admin/base_site.html" %}
{% load i18n admin_urls static admin_modify %}

{% block extrastyle %}{{ block.super }}
    
{% endblock %}

{% block content %}
    

        

            {% csrf_token %}
            
{{ action_name }}
            
Olete teostamas tegevust "{{ action_name }}" järgmiste {{ queryset.count }} objektiga:
            

                {% for obj in queryset %}
                    
{{ obj }}
                {% endfor %}
            
            
Olete täiesti kindel, et soovite jätkata?

            {{ form.as_p }}

            
            {% for item in form._selected_action.value %}
                
            {% endfor %}

            

                
                
            
        
    
{% endblock %}

            

              
                Copy
              
            
          

            # myapp/forms.py
from django import forms
from .models import Post

class ChangePostStatusForm(forms.Form):
    _selected_action = forms.CharField(widget=forms.HiddenInput)
    action = forms.CharField(widget=forms.HiddenInput)

    new_status = forms.ChoiceField(
        label="Uus olek",
        choices=Post.STATUS_CHOICES, # Eeldades, et STATUS_CHOICES on defineeritud Post mudelis
        required=True
    )
    comment = forms.CharField(
        label="Põhjus/Kommentaar (valikuline)",
        required=False,
        widget=forms.Textarea(attrs={'rows': 3})
    )

# Add STATUS_CHOICES to Post model
# myapp/models.py
from django.db import models

class Post(models.Model):
    STATUS_CHOICES = [
        ('draft', 'Mustand'),
        ('pending', 'Ootel ülevaatamist'),
        ('approved', 'Kinnitatud'),
        ('rejected', 'Tagasi lükatud'),
    ]
    title = models.CharField(max_length=255)
    content = models.TextField()
    status = models.CharField(max_length=20, default='draft', choices=STATUS_CHOICES)
    comment_history = models.TextField(blank=True, null=True)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.title

            

              
                Copy
              
            
          

            # myapp/admin.py (continued)
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Post
from .forms import ChangePostStatusForm # Import the new form

# ... (ProductAdmin and PostAdmin definitions, other imports)

@admin.register(Post)
class PostAdmin(admin.ModelAdmin):
    list_display = ('title', 'status', 'created_at')
    list_filter = ('status',)
    search_fields = ('title',)

    # Existing mark_posts_approved action...

    def change_post_status_with_comment(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        form = None
        if 'apply' in request.POST:
            form = ChangePostStatusForm(request.POST)
            if form.is_valid():
                new_status = form.cleaned_data['new_status']
                comment = form.cleaned_data['comment']

                updated_count = 0
                for post in queryset:
                    post.status = new_status
                    if comment:
                        post.comment_history = (post.comment_history or '') + f\"\n[{request.user.username}] muutis olekuks {new_status} kommentaariga: {comment}\"
                    post.save()
                    updated_count += 1

                self.message_user(
                    request,
                    f\"{updated_count} postitus(t) muutis(id) olekut '{new_status}' ja kommentaar lisati.\",
                    messages.SUCCESS
                )
                return HttpResponseRedirect(request.get_full_path())
        
        # If not confirmed, or GET request, show the input form
        if not form:
            form = ChangePostStatusForm(initial={
                '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
                'action': 'change_post_status_with_comment',
            })

        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = form
        context['action_name'] = self.change_post_status_with_comment.short_description
        context['title'] = _("Muuda postituse olekut ja lisa kommentaar")

        return render(request, 'admin/change_status_action.html', context)

    change_post_status_with_comment.short_description = _("Muuda valitud postituste olekut (kommentaariga)")

    actions = [
        mark_posts_approved,
        change_post_status_with_comment
    ]

            

              
                Copy
              
            
          

            {# templates/admin/change_status_action.html #}
{% extends "admin/base_site.html" %}
{% load i18n admin_urls static admin_modify %}

{% block extrastyle %}{{ block.super }}
    
{% endblock %}

{% block content %}
    

        

            {% csrf_token %}
            
{{ action_name }}
            
Olete muutmas olekut järgmistel {{ queryset.count }} objektiga:
            

                {% for obj in queryset %}
                    
{{ obj }}
                {% endfor %}
            
            
Palun valige uus olek ja soovi korral lisage kommentaar:

            {% for field in form %}
                {% if field.is_hidden %}
                    {{ field }}
                {% else %}
                    

                        {{ field.errors }}
                        {{ field.label_tag }}
                        {{ field }}
                        {% if field.help_text %}
{{ field.help_text }}
{% endif %}
                    
                {% endif %}
            {% endfor %}

            

                
                
            
        
    
{% endblock %}

            

              
                Copy
              
            
          

            # myapp/admin.py (continued)
import requests # You'll need to install requests: pip install requests
# ... other imports ...

# Assuming ProductAdmin from earlier

class ProductAdmin(admin.ModelAdmin):
    # ... existing list_display, list_filter, search_fields ...

    def sync_products_to_external_ims(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        # Check for confirmation (similar to previous examples if needed)
        if 'apply' in request.POST:
            # Simulate an external API endpoint
            EXTERNAL_IMS_API_URL = \"https://api.example.com/v1/products/sync/\"
            API_KEY = \"your_secret_api_key\" # In a real app, use settings.py or environment variables

            successful_syncs = 0
            failed_syncs = []

            for product in queryset:
                data = {
                    \"product_id\": product.id,
                    \"name\": product.name,
                    \"price\": str(product.price), # Convert Decimal to string for JSON
                    \"is_discounted\": product.is_discounted,
                    # Add other relevant product data
                }
                headers = {
                    \"Authorization\": f\"Bearer {API_KEY}\",
                    \"Content-Type\": \"application/json\"
                }
                
                try:
                    response = requests.post(EXTERNAL_IMS_API_URL, json=data, headers=headers, timeout=5) # 5-second timeout
                    response.raise_for_status() # Raise an HTTPError for bad responses (4xx or 5xx)
                    successful_syncs += 1
                except requests.exceptions.RequestException as e:
                    failed_syncs.append(f\"Toode {product.name} (ID: {product.id}): {e}\")
                except Exception as e:
                    failed_syncs.append(f\"Toode {product.name} (ID: {product.id}): Ootamatu viga: {e}\")

            if successful_syncs > 0:
                self.message_user(
                    request,
                    f\"{successful_syncs} toode(t) sünkroniseeriti edukalt välise IMS-iga.\",
                    messages.SUCCESS
                )
            if failed_syncs:
                error_message = f\"Ebaõnnestus sünkroniseerida {len(failed_syncs)} toode(t):\n\" + \"\n\".join(failed_syncs)
                self.message_user(request, error_message, messages.ERROR)
            
            return HttpResponseRedirect(request.get_full_path())

        # Initial GET request or non-apply POST request: show confirmation (if desired)
        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = ConfirmationForm(initial={
            '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
            'action': 'sync_products_to_external_ims',
        })
        context['action_name'] = self.sync_products_to_external_ims.short_description
        context['title'] = _("Kinnita andmete sünkroniseerimine")

        return render(request, 'admin/confirmation_action.html', context) # Re-use confirmation template

    sync_products_to_external_ims.short_description = _("Sünkroniseeri valitud tooted välise IMS-iga")

    actions = [
        # ... other actions ...
        sync_products_to_external_ims,
    ]

            

              
                Copy
              
            
          

            # myapp/admin.py (continued)
import csv
from django.http import HttpResponse
# ... other imports ...

class PostAdmin(admin.ModelAdmin):
    # ... existing list_display, list_filter, search_fields, actions ...

    def export_posts_as_csv(self, request: HttpRequest, queryset: QuerySet) -> HttpResponse:
        response = HttpResponse(content_type='text/csv')
        response['Content-Disposition'] = 'attachment; filename="posts_export.csv"'

        writer = csv.writer(response)
        # Write header row
        writer.writerow(['Pealkiri', 'Olek', 'Loodud kell', 'Sisu eelvaade'])

        for post in queryset:
            writer.writerow([
                post.title,
                post.get_status_display(), # Use get_FOO_display() for choice fields
                post.created_at.strftime(\"%Y-%m-%d %H:%M:%S\"),
                post.content[:100] + '...' if len(post.content) > 100 else post.content # Truncate long content
            ])
        
        self.message_user(
            request,
            f\"{queryset.count()} postitus(t) eksporditi edukalt CSV-faili.\",
            messages.SUCCESS
        )
        return response

    export_posts_as_csv.short_description = _("Ekspordi valitud postitused CSV-na")

    actions = [
        # ... other actions ...
        export_posts_as_csv,
    ]

            

              
                Copy
              
            
          

            # myproject/celery.py (standard Celery setup)
import os
from celery import Celery

# Set the default Django settings module for the 'celery' program.
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'myproject.settings')

app = Celery('myproject')

# Using a string here means the worker will not have to
# pickle the object when using Windows.
app.config_from_object('django.conf:settings', namespace='CELERY')

# Load task modules from all registered Django app configs.
app.autodiscover_tasks()

@app.task(bind=True)
def debug_task(self):
    print(f'Request: {self.request!r}')

            

              
                Copy
              
            
          

            # myapp/tasks.py
import requests
from celery import shared_task
from django.contrib.auth import get_user_model
from django.apps import apps

@shared_task
def sync_product_to_external_ims_task(product_id, admin_user_id):
    Product = apps.get_model('myapp', 'Product')
    User = get_user_model()

    try:
        product = Product.objects.get(pk=product_id)
        admin_user = User.objects.get(pk=admin_user_id)

        EXTERNAL_IMS_API_URL = \"https://api.example.com/v1/products/sync/\"
        API_KEY = \"your_secret_api_key\" # Use environment variables or Django settings

        data = {
            \"product_id\": product.id,
            \"name\": product.name,
            \"price\": str(product.price),
            \"is_discounted\": product.is_discounted,
        }
        headers = {
            \"Authorization\": f\"Bearer {API_KEY}\",
            \"Content-Type\": \"application/json\"
        }

        response = requests.post(EXTERNAL_IMS_API_URL, json=data, headers=headers, timeout=10)
        response.raise_for_status()
        
        # Log success (e.g., to Django logs or a specific model for tracking)
        print(f\"Toode {product.name} (ID: {product.id}) sünkroniseeritud {admin_user.username} poolt edukalt.\")

    except Product.DoesNotExist:
        print(f\"Toodet ID-ga {product_id} ei leitud.\")
    except User.DoesNotExist:
        print(f\"Administraatorit ID-ga {admin_user_id} ei leitud.\")
    except requests.exceptions.RequestException as e:
        print(f\"API sünkroniseerimine ebaõnnestus toote {product_id} jaoks: {e}\")
    except Exception as e:
        print(f\"Ootamatu viga sünkroniseerimisel toote {product_id} jaoks: {e}\")


            

              
                Copy
              
            
          

            # myapp/admin.py (continued)
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Product # Assuming Product model from earlier
from .tasks import sync_product_to_external_ims_task # Import your Celery task

class ProductAdmin(admin.ModelAdmin):
    # ... existing list_display, list_filter, search_fields ...

    def async_sync_products_to_external_ims(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        if 'apply' in request.POST:
            admin_user_id = request.user.id
            for product in queryset:
                # Enqueue the task for each selected product
                sync_product_to_external_ims_task.delay(product.id, admin_user_id)
            
            self.message_user(
                request,
                f\"{queryset.count()} toote sünkroniseerimise ülesanne(d) on järjekorda pandud.\",
                messages.SUCCESS
            )
            return HttpResponseRedirect(request.get_full_path())

        # Initial GET request or non-apply POST request: show confirmation
        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = ConfirmationForm(initial={
            '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
            'action': 'async_sync_products_to_external_ims',
        })
        context['action_name'] = self.async_sync_products_to_external_ims.short_description
        context['title'] = _("Kinnita asünkroonne andmete sünkroniseerimine")

        return render(request, 'admin/confirmation_action.html', context) # Re-use confirmation template

    async_sync_products_to_external_ims.short_description = _("Lisa asünkroonne sünkroniseerimine valitud toodetele IMS-iga järjekorda")

    actions = [
        # ... other actions ...
        async_sync_products_to_external_ims,
    ]

            

              
                Copy
              
            
          

            def sensitive_action(self, request, queryset):
    if not request.user.has_perm('myapp.can_perform_sensitive_action'):
        self.message_user(request, _("Teil puuduvad õigused selle tegevuse sooritamiseks."), messages.ERROR)
        return HttpResponseRedirect(request.get_full_path())
    # ... sensitive action logic ...

            

              
                Copy
              
            
          

            # myapp/models.py
class Product(models.Model):
    # ... fields ...
    class Meta:
        permissions = [
            ("can_perform_sensitive_action", "Saab sooritada tundlikku tootetegevust"),
        ]

            

              
                Copy
              
            
          

            # myapp/admin.py
class ProductAdmin(admin.ModelAdmin):
    # ... actions definition ...

    def get_actions(self, request: HttpRequest):
        actions = super().get_actions(request)
        # Eemaldage tegevus 'make_discounted', kui kasutajal puudub spetsiifiline õigus
        if not request.user.has_perm('myapp.change_product'): # Või kohandatud õigus nagu 'can_discount_product'
            if 'make_discounted' in actions:
                del actions['make_discounted']
        return actions

            

              
                Copy
              
            
          

            # myapp/admin_actions.py
from django.contrib import messages
from django.http import HttpRequest, HttpResponseRedirect
from django.db.models import QuerySet

def mark_as_active(modeladmin, request: HttpRequest, queryset: QuerySet):
    updated = queryset.update(is_active=True)
    modeladmin.message_user(request, f\"{updated} üksus(t) märgiti aktiivseks.\", messages.SUCCESS)
mark_as_active.short_description = \"Märgi valitud aktiivseks\"

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin
from .models import MyModel
from .admin_actions import mark_as_active

@admin.register(MyModel)
class MyModelAdmin(admin.ModelAdmin):
    list_display = ('name', 'is_active')
    actions = [mark_as_active]

            

              
                Copy
              
            
          

            # myapp/tests.py
from django.test import TestCase, Client
from django.contrib.auth import get_user_model
from django.urls import reverse
from django.contrib import admin
from .models import Product
from .admin import ProductAdmin # Import your ModelAdmin

User = get_user_model()

class ProductAdminActionTests(TestCase):
    def setUp(self):
        self.admin_user = User.objects.create_superuser('admin', 'admin@example.com', 'password')
        self.client = Client()
        self.client.login(username='admin', password='password')

        self.p1 = Product.objects.create(name="Product A", price=10.00, is_discounted=False)
        self.p2 = Product.objects.create(name="Product B", price=20.00, is_discounted=False)
        self.p3 = Product.objects.create(name="Product C", price=30.00, is_discounted=True)

        self.admin_site = admin.AdminSite()
        self.model_admin = ProductAdmin(Product, self.admin_site)

    def test_make_discounted_action(self):
        # Simulate selecting products and performing the action
        change_list_url = reverse('admin:myapp_product_changelist')
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [self.p1.pk, self.p2.pk],
            'action': 'make_discounted',
            'index': 0, # Required for some Django admin internal logic
        }, follow=True)

        self.assertEqual(response.status_code, 200)
        self.assertContains(response, '2 toode(t) märgiti edukalt allahinnatuks.')
        
        self.p1.refresh_from_db()
        self.p2.refresh_from_db()
        self.p3.refresh_from_db()

        self.assertTrue(self.p1.is_discounted)
        self.assertTrue(self.p2.is_discounted)
        self.assertTrue(self.p3.is_discounted) # This one was already discounted

    def test_make_discounted_action_confirmation(self):
        # For actions with confirmation, you'd test the two-step process
        change_list_url = reverse('admin:myapp_post_changelist') # Assuming Post model for confirmation example
        post1 = Post.objects.create(title='Test Post 1', content='...', status='draft')
        post2 = Post.objects.create(title='Test Post 2', content='...', status='draft')

        # Step 1: Request confirmation page
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [post1.pk, post2.pk],
            'action': 'mark_posts_approved',
            'index': 0,
        })
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"Kinnita tegevus", response.content) # Check if confirmation page is rendered

        # Step 2: Submit confirmation form
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [post1.pk, post2.pk],
            'action': 'mark_posts_approved',
            'apply': 'Jah, olen kindel',
            'confirm': 'on', # Value for a checkbox if rendered as checkbox
            '_selected_action': [str(post1.pk), str(post2.pk)], # Must be passed back from form
            'index': 0,
        }, follow=True)

        self.assertEqual(response.status_code, 200)
        self.assertContains(response, '2 postitus(t) märgiti edukalt kinnitatuks.')

        post1.refresh_from_db()
        post2.refresh_from_db()
        self.assertEqual(post1.status, 'approved')
        self.assertEqual(post2.status, 'approved')

            

              
                Copy