pasteque/paste/models.py

from datetime import datetime, timedelta

import shortuuid
from django.core.validators import MaxLengthValidator
from django.db import models
from django.utils import timezone
from django.utils.translation import gettext_lazy as _

from webtools import settings


class Paste(models.Model):
    """Paste object."""

    filename = models.CharField(max_length=255, default="")
    slug = models.SlugField(unique=True, editable=False)
    content = models.TextField(
        validators=[MaxLengthValidator(settings.PASTE["max_characters"])]
    )
    size = models.IntegerField(default=0, editable=False)
    paste_time = models.DateTimeField(auto_now_add=True)
    viewcount = models.IntegerField(default=0, editable=False)

    def compute_size(self):
        """Computes size."""
        self.size = len(self.content)

    def incr_viewcount(self):
        """Increment view counter."""
        self.viewcount = self.viewcount + 1
        self.save()

    def __str__(self):
        excerpt = repr(self.content[:100]) + ("..." if len(self.content) > 100 else "")
        return f"{self.slug} - {excerpt}"

    def choose_slug(self):
        while True:
            uuid = shortuuid.uuid()
            for i in range(4, len(uuid)):
                potential_uuid = uuid[:i]
                if not type(self).objects.filter(slug=potential_uuid):
                    self.slug = potential_uuid
                    return