spejstore-new/storage/models.py

from __future__ import unicode_literals

import uuid
import re

from django.db import models
from django.conf import settings
from django.contrib.auth.models import User
from django_hstore import hstore
from tree.fields import PathField
from tree.models import TreeModelMixin

import requests


STATES = (
    ('present', 'Present'),
    ('taken', 'Taken'),
    ('broken', 'Broken'),
    ('missing', 'Missing'),
)


class Category(models.Model):
    name = models.CharField(max_length=127)

    def __str__(self):
        return self.name


class Item(models.Model, TreeModelMixin):
    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

    parent = models.ForeignKey('self', null=True, blank=True)
    path = PathField()

    name = models.TextField()

    description = models.TextField(blank=True, null=True)
    state = models.CharField(max_length=31, choices=STATES, default=STATES[0][0])
    categories = models.ManyToManyField(Category, blank=True)
    owner = models.ForeignKey(User, null=True, blank=True, related_name='owned_items')

    taken_by = models.ForeignKey(User, null=True, blank=True, related_name='taken_items')
    taken_on = models.DateTimeField(blank=True, null=True)
    taken_until = models.DateTimeField(blank=True, null=True)

    props = hstore.DictionaryField(blank=True)

    objects = hstore.HStoreManager()

    def __str__(self):
        return '- ' * (self.get_level() or 0) + self.name

    def get_absolute_url(self):
        from django.urls import reverse
        return reverse('item-display', kwargs={'pk': str(self.pk)})

    def get_or_create_label(self, **kwargs):
        defaults = {
            'id': re.sub('[^A-Z0-9]', '', self.name.upper())[:16],
            }

        defaults.update(kwargs)

        obj, created = self.labels.get_or_create(**kwargs, defaults=defaults)

        return obj

    class Meta:
        ordering = ('path',)


class ItemImage(models.Model):
    item = models.ForeignKey(Item, related_name='images')
    image = models.ImageField()


class Label(models.Model):
    id = models.CharField(max_length=64, primary_key=True)
    item = models.ForeignKey(Item, related_name='labels')
    style = models.CharField(max_length=32, choices=(
        ('basic_99012_v1', 'Basic Dymo 89x36mm label'),
        ), default='basic_99012_v1')
    created = models.DateTimeField(auto_now_add=True, blank=True)

    def __str__(self):
        return '{}'.format(self.id)

    def print(self):
        resp = requests.post(
            '{}/api/1/print/{}'.format(settings.LABEL_API, self.id))
        resp.raise_for_status()