129 lines
4.3 KiB
Python
129 lines
4.3 KiB
Python
from django.contrib import admin
|
|
from django.urls import reverse
|
|
from django.utils.html import format_html
|
|
from yeast.models import Yeast, Strain, Manufacturer, Storage, Batch
|
|
from yeast.forms import YeastModelForm
|
|
|
|
import beer
|
|
|
|
from config.extras import BREWFATHER_APP_ROOT
|
|
|
|
import logging
|
|
logger = logging.getLogger('django')
|
|
|
|
class BatchInline(admin.TabularInline):
|
|
model = Batch
|
|
extra = 0
|
|
|
|
class SampleInline(admin.TabularInline):
|
|
model = Yeast
|
|
extra = 0
|
|
|
|
class StrainInline(admin.TabularInline):
|
|
model = Strain
|
|
extra = 5
|
|
|
|
class ParentInline(admin.TabularInline):
|
|
verbose_name = 'Parent Samples'
|
|
model = Batch.parent.through
|
|
|
|
class YeastAdmin(admin.ModelAdmin):
|
|
list_display = [ 'batch', 'url', 'lot_number', 'age', 'storage', 'viability', 'generation_num', 'cellcount', 'pitched', 'date_pitched', 'pitched_batch']
|
|
list_editable = ['pitched', 'date_pitched', 'pitched_batch', 'lot_number']
|
|
|
|
def batch_url(self, obj):
|
|
if obj.pitched_batch:
|
|
bf_id = obj.pitched_batch.brewfather_id
|
|
return format_html("<a href='https://web.brewfather.app/tabs/batches/batch/{batch_id}'>{batch_id}</a>", batch_id=bf_id)
|
|
|
|
def url(self, obj):
|
|
if obj.data_web:
|
|
return format_html("<a href='{url}'>{url}</a>", url=obj.data_web)
|
|
|
|
class StrainAdmin(admin.ModelAdmin):
|
|
list_display = ['name', 'long_name', 'manufacturer', 'avilable_batches']
|
|
inlines = [
|
|
BatchInline,
|
|
]
|
|
list_editable = ['long_name', 'manufacturer']
|
|
|
|
def avilable_batches(self, obj):
|
|
related_objs = [x for x in obj.batch_set.all() if not x.consumed]
|
|
|
|
urls = []
|
|
|
|
for related_obj in related_objs:
|
|
url_text = '{}: {}'.format(related_obj.get_source_display(), related_obj.production_date.strftime("%Y-%m-%d"))
|
|
url = reverse('admin:yeast_batch_change', args=[related_obj.id]) # replace 'myapp' with your app name
|
|
urls.append('<a href="{}">{}</a>'.format(url, url_text))
|
|
return format_html(', '.join(urls))
|
|
|
|
avilable_batches.short_description = 'Available Batches'
|
|
|
|
class StorageAdmin(admin.ModelAdmin):
|
|
list_display = ['name', 'viability_loss', 'viability_interval']
|
|
inlines = [
|
|
SampleInline,
|
|
]
|
|
|
|
class ManufacturerAdmin(admin.ModelAdmin):
|
|
list_display = ['name', 'url']
|
|
inlines = [
|
|
StrainInline,
|
|
]
|
|
|
|
def url(self, obj):
|
|
if obj.website:
|
|
return format_html("<a href='{url}'>{url}</a>", url=obj.website)
|
|
|
|
|
|
class BatchAdmin(admin.ModelAdmin):
|
|
list_display = ['strain', 'consumed', 'source', 'parent_samples', 'production_date', 'avilable_samples', 'used_samples']
|
|
form = YeastModelForm
|
|
filter_horizontal = ['parent']
|
|
inlines = [
|
|
#ParentInline,
|
|
SampleInline,
|
|
]
|
|
|
|
def save_related(self, request, form, formsets, change):
|
|
super(BatchAdmin, self).save_related(request, form, formsets, change)
|
|
if form.instance.source_batch:
|
|
relate_samples = [x for x in Yeast.objects.all() if x.pitched_batch==form.instance.source_batch]
|
|
for sample in relate_samples:
|
|
logger.critical(sample)
|
|
form.instance.parent.add(sample)
|
|
|
|
def parent_samples(self, obj):
|
|
obj.parent.all()
|
|
|
|
def used_samples(self, obj):
|
|
related_objs = list(set(obj.yeast_set.all()) - set(obj.remaining_samples))
|
|
|
|
urls = []
|
|
|
|
for related_obj in related_objs:
|
|
url = reverse('admin:yeast_yeast_change', args=[related_obj.id])
|
|
urls.append('<a href="{}">{}</a>'.format(url, related_obj))
|
|
return format_html(', '.join(urls))
|
|
|
|
used_samples.short_description = 'Used Samples'
|
|
|
|
def avilable_samples(self, obj):
|
|
related_objs = obj.remaining_samples
|
|
|
|
urls = []
|
|
|
|
for related_obj in related_objs:
|
|
url = reverse('admin:yeast_yeast_change', args=[related_obj.id]) # replace 'myapp' with your app name
|
|
urls.append('<a href="{}">{}</a>'.format(url, related_obj))
|
|
return format_html(', '.join(urls))
|
|
|
|
avilable_samples.short_description = 'Available Samples'
|
|
|
|
admin.site.register(Yeast, YeastAdmin)
|
|
admin.site.register(Strain, StrainAdmin)
|
|
admin.site.register(Manufacturer, ManufacturerAdmin)
|
|
admin.site.register(Storage, StorageAdmin)
|
|
admin.site.register(Batch, BatchAdmin)
|