mirror of
https://github.com/taigaio/taiga-back
synced 2025-10-06 00:02:52 +02:00
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
#
|
|
# Copyright (c) 2021-present Kaleidos INC
|
|
|
|
from django_pglocks import advisory_lock
|
|
|
|
|
|
def detail_route(methods=['get'], **kwargs):
|
|
"""
|
|
Used to mark a method on a ViewSet that should be routed for detail requests.
|
|
"""
|
|
def decorator(func):
|
|
func.bind_to_methods = methods
|
|
func.detail = True
|
|
func.permission_classes = kwargs.get('permission_classes', [])
|
|
func.kwargs = kwargs
|
|
return func
|
|
return decorator
|
|
|
|
|
|
def list_route(methods=['get'], **kwargs):
|
|
"""
|
|
Used to mark a method on a ViewSet that should be routed for list requests.
|
|
"""
|
|
def decorator(func):
|
|
func.bind_to_methods = methods
|
|
func.detail = False
|
|
func.permission_classes = kwargs.get('permission_classes', [])
|
|
func.kwargs = kwargs
|
|
return func
|
|
return decorator
|
|
|
|
|
|
def model_pk_lock(func):
|
|
"""
|
|
This decorator is designed to be used in ModelViewsets methods to lock them based
|
|
on the model and the id of the selected object.
|
|
"""
|
|
def decorator(self, *args, **kwargs):
|
|
from taiga.base.utils.db import get_typename_for_model_class
|
|
pk = self.kwargs.get(self.pk_url_kwarg, None)
|
|
tn = get_typename_for_model_class(self.get_queryset().model)
|
|
key = "{0}:{1}".format(tn, pk)
|
|
|
|
with advisory_lock(key):
|
|
return func(self, *args, **kwargs)
|
|
|
|
return decorator
|