Image

Complexity

Total Complexity

6

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Importance

Changes	3
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	delete()	0	3	1
A	__str__()	0	2	1
A	has_read_permission()	0	3	1
A	has_write_permission()	0	3	1
A	has_object_read_permission()	0	2	1
A	has_object_write_permission()	0	2	1

import os.path

from django.db import models

from dry_rest_permissions.generics import allow_staff_or_superuser

from sigma_core.models.user import User
from sigma_core.models.group import Group


def img_path(instance, filename):
    from django.utils.crypto import get_random_string
    extension = os.path.splitext(filename)[1]
    return "img/" + get_random_string(length=150, allowed_chars='abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_') + extension


class Image(models.Model):
    file = models.ImageField(max_length=255, upload_to=img_path)
    owner = models.ForeignKey(User)
    added = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.file.__str__()

    def delete(self, *args, **kwargs):
        self.file.delete(save=False)
        return super(Image, self).delete(*args, **kwargs)

    # Permissions
    @staticmethod
    def has_read_permission(request):
        return True

    def has_object_read_permission(self, request):
        return True

    @staticmethod
    def has_write_permission(request):
        return True

    def has_object_write_permission(self, request):
        return request.user == self.owner