takahe/users/models/invite.py

import random

import urlman
from django.db import models


class Invite(models.Model):
    """
    An invite token, good for one signup.
    """

    # Should always be lowercase
    token = models.CharField(max_length=500, unique=True)

    # Is it limited to a specific email?
    email = models.EmailField(null=True, blank=True)

    # Admin note about this code
    note = models.TextField(null=True, blank=True)

    created = models.DateTimeField(auto_now_add=True)
    updated = models.DateTimeField(auto_now=True)

    class urls(urlman.Urls):
        admin = "/admin/invites/"
        admin_create = "{admin}create/"
        admin_view = "{admin}{self.pk}/"

    @classmethod
    def create_random(cls, email=None, note=None):
        return cls.objects.create(
            token="".join(
                random.choice("abcdefghkmnpqrstuvwxyz23456789") for i in range(20)
            ),
            email=email,
            note=note,
        )