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 self.id def print(self): resp = requests.post( '{}/api/1/print/{}'.format(settings.LABEL_API, self.id)) resp.raise_for_status()