🛍장고 백오피스 수정-3

Logentry

from django.contrib.admin.models import LogEntry, CHANGE   >>>> 2.
from django.contrib.contenttypes.models import ContentType >>>>> 2.

def refund(modeladmin, request, queryset):
    with transaction.atomic():
        qs = queryset.filter(~Q(status='환불'))
        ct = ContentType.objects.get_for_model(queryset.model)
        for obj in qs:
            obj.product.stock += obj.quantity
            obj.product.save()

            LogEntry.objects.log_action(   >>>>> 3.
                user_id = request.user.id,
                content_type_id=ct.pk,
                object_id=obj.pk,
                object_repr='주문 환불',
                action_flag=CHANGE,
                change_message='주문 환불'
            )
        qs.update(status='환불')

1. 장고 어드민 페이지에 모델을 변경하면 기록이 남는데 이 부분을 커스터마이징을 했다.

2. 실행하기 위해서 아래 것들을 임포트 했다.

   • from django.contrib.admin.models import LogEntry, CHANGE
   • from django.contrib.contenttypes.models import ContentType

3. 기존에 구현해 놓은 반복문에 Logentry객체에 접근해서 값들을 바꾸어준다.

Logentry 클래스

class LogEntry(models.Model):
    action_time = models.DateTimeField(
        _('action time'),
        default=timezone.now,
        editable=False,
    )
    user = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        models.CASCADE,
        verbose_name=_('user'),
    )
    content_type = models.ForeignKey(
        ContentType,
        models.SET_NULL,
        verbose_name=_('content type'),
        blank=True, null=True,
    )
    object_id = models.TextField(_('object id'), blank=True, null=True)
    # Translators: 'repr' means representation (https://docs.python.org/library/functions.html#repr)
    object_repr = models.CharField(_('object repr'), max_length=200)
    action_flag = models.PositiveSmallIntegerField(_('action flag'), choices=ACTION_FLAG_CHOICES)
    # change_message is either a string or a JSON structure
    change_message = models.TextField(_('change message'), blank=True)

    objects = LogEntryManager()  >>>>>>>>>>>>>>>>>>>>>>>>>

• Logentry클래스를 까보면 objects변수에 LogEntryManager() 클래스가 담겨있다.
• LogEntryManager()클래스를 살펴보자.

LogEntryManager클래스

class LogEntryManager(models.Manager):
    use_in_migrations = True

    def log_action(self, user_id, content_type_id, object_id, object_repr, action_flag, change_message=''):
        if isinstance(change_message, list):
            change_message = json.dumps(change_message)
        return self.model.objects.create(
            user_id=user_id,
            content_type_id=content_type_id,
            object_id=str(object_id),
            object_repr=object_repr[:200],
            action_flag=action_flag,
            change_message=change_message,
        )

• log_action 매소드가 있다. 이 부분을 바꿔주는 것이다.
• 2번째 인자인(self 제외) content_type_id 값에는 content_type_id가 필요한데....

from django.contrib.admin.models import LogEntry, CHANGE   >>>> 2.
from django.contrib.contenttypes.models import ContentType >>>>> 2.

def refund(modeladmin, request, queryset):
    with transaction.atomic():
        qs = queryset.filter(~Q(status='환불'))
        ct = ContentType.objects.get_for_model(queryset.model) >>>> 4. 
        for obj in qs:
            obj.product.stock += obj.quantity
            obj.product.save()

            LogEntry.objects.log_action(   >>>>> 3.
                user_id = request.user.id,
                content_type_id=ct.pk,
                object_id=obj.pk,
                object_repr='주문 환불',
                action_flag=CHANGE,
                change_message='주문 환불'
            )
        qs.update(status='환불')

4. ContentType 객체에 접근해서 모델을 뽑아온다.

• ContentType 를 까보면 objects 변수에 ContentTypeManager()클래스가 담겨있고 이걸 까보면

get_for_model 메소드

def get_for_model(self, model, for_concrete_model=True):
        """
        Return the ContentType object for a given model, creating the
        ContentType if necessary. Lookups are cached so that subsequent lookups
        for the same model don't hit the database.
        """
        opts = self._get_opts(model, for_concrete_model)
        try:
            return self._get_from_cache(opts)
        except KeyError:
            pass

        # The ContentType entry was not found in the cache, therefore we
        # proceed to load or create it.
        try:
            # Start with get() and not get_or_create() in order to use
            # the db_for_read (see #20401).
            ct = self.get(app_label=opts.app_label, model=opts.model_name)
        except self.model.DoesNotExist:
            # Not found in the database; we proceed to create it. This time
            # use get_or_create to take care of any race conditions.
            ct, created = self.get_or_create(
                app_label=opts.app_label,
                model=opts.model_name,
            )
        self._add_to_cache(self.db, ct)
        return ct

• 수많은 메소드 중에 get_for_model 이란 메소드가 있다.

• 이 함수에 인자로 쿼리셋에서 반은 모델을 인자로 넘겨서 ct에 저장한다.

• Logentry설정을 마무리해준다.

changelist에 환불 버튼 추가

환불 버튼을 누르면 상태가 변하는 것을 만들어보자.

class OrderAdmin(admin.ModelAdmin):
    list_filter = ('status',)
    list_display = ('fcuser', 'product', 'styled_status', 'action')
    change_list_template = 'admin/order_change_list.html' >>>1
    actions = [
        refund
    ]
    def action(self, obj):  >>>> 2.
        if obj.status != '환불':
            return format_html(f'<input type="button" value="환불"token interpolation">{obj.id})" class="btn btn-primary btn-sm">')

• 저저번 포스트에서 했던 결제 완료 나 환불 글씨 색깔을 바꾸는 로직으로 버튼을 추가해주었다.

1. change_list_template 을 새로만든 html페이지로 연결해주었다.

2. action함수를 보면 obj를 인자로 받는데 print문으로 찍어보면 Order모델이 나온다.

   • 만약 상태가 환불이 아니면 환불버튼을 생성해준다. onclick옵션으로 함수를 만들고 인자로 obj의 id값을 넣어준다.

change_list_template

{% extends 'admin/change_list.html' %}
{% load static %}
{% block extrahead %}
{{ block.super }}
<script>
    function order_refund_submit(obj_id) {
        jQuery("#changelist-form").append('<input type="hiden" name="obj_id" value="' + obj_id + '">');
        jQuery("#changelist-form").submit();
    }
</script>
{% endblock %}  

• extends 'admin/ change_list.html' -> 장고안에 있는 어드민 페이지 리스트 html문서를 익스텐드했다.
• 블럭안에 제이쿼리 코드를 넣었다. 제이쿼리를 잘 몰라서 유추해보자면
  • 아까 action함수에서 onclick이벤트에 함수에 인자로 obj.id를 넘겨주었고
  • changelistform에 input타입이 hiden이고 value에 obj_id를 넣어주고 submit한다.

changelist_view

def changelist_view(self, request, extra_context=None):
        ## 원하는 동작
        extra_context = { 'title': '주문 목록' }
# 추가된 코드 
        if request.method == 'POST':
            obj_id = request.POST.get('obj_id')
            if obj_id:
                qs = Order.objects.filter(pk=obj_id)
                ct = ContentType.objects.get_for_model(qs.model)
                for obj in qs:
                    obj.product.stock += obj.quantity
                    obj.product.save()
                    LogEntry.objects.log_action(
                        user_id = request.user.id,
                        content_type_id=ct.pk,
                        object_id=obj.pk,
                        object_repr='주문 환불',
                        action_flag=CHANGE,
                        change_message='주문 환불'
                    )
                qs.update(status='환불')

• jQuery에서 넘겨준 값을 받는다.
• changelist_view도 장고에서 뷰단을 처리하는 로직과 비슷하다.
• POST로 받으면 코드를 실행하며
  • request.POST.get('obj_id')가 있으면 다음으로 넘어간다.
  • 나머지는 refund함수와 비슷하고 qs에 Order모델에 pk값이 obj_id(request.Post.get)인 값을 저장한다.