htsworkflow/frontend/samples/admin.py

   1 from django.contrib import admin
   2 from django.contrib.admin import widgets
   3 from django.contrib.admin.models import User
   4 from django.contrib.auth.admin import UserAdmin
   5 from django.contrib.auth.forms import UserCreationForm, UserChangeForm
   6 from django.template import Context, Template
   7 from django.db import models
   8 from django.utils.translation import ugettext_lazy as _
   9
  10 from htsworkflow.frontend.samples.models import Antibody, Cellline, Condition, ExperimentType, HTSUser, LibraryType, Species, Affiliation, Library, Tag
  11 from htsworkflow.frontend.experiments.models import Lane
  12 from htsworkflow.frontend.inventory.models import PrinterTemplate
  13 from htsworkflow.frontend.bcmagic.utils import print_zpl_socket
  14
  15 # Let's disable those pesky delete everything by accident features.
  16 admin.site.disable_action('delete_selected')
  17
  18 class AffiliationOptions(admin.ModelAdmin):
  19     list_display = ('name','contact','email')
  20     fieldsets = (
  21       (None, {
  22           'fields': (('name','contact','email','users'))
  23       }),
  24     )
  25
  26     # some post 1.0.2 version of django has formfield_overrides
  27     # which would replace this code with:
  28     # formfield_overrids = {
  29     #   models.ManyToMany: { 'widget': widgets.FilteredSelectMultiple }
  30     # }
  31     def formfield_for_dbfield(self, db_field, **kwargs):
  32       if db_field.name == 'users':
  33         kwargs['widget'] = widgets.FilteredSelectMultiple(db_field.verbose_name, (db_field.name in self.filter_vertical))
  34       rv = super(AffiliationOptions, self).formfield_for_dbfield(db_field, **kwargs)
  35     #  print db_field.name, kwargs
  36       return rv
  37
  38 class AntibodyOptions(admin.ModelAdmin):
  39     search_fields = ('antigene','nickname','catalog','antibodies','source','biology','notes')
  40     list_display = ('antigene','nickname','antibodies','catalog','source','biology','notes')
  41     list_filter = ('antibodies','source')
  42     fieldsets = (
  43       (None, {
  44           'fields': (('antigene','nickname','antibodies'),('catalog','source'),('biology'),('notes'))
  45       }),
  46      )
  47
  48 class CelllineOptions(admin.ModelAdmin):
  49     list_display = ('cellline_name', 'nickname', 'notes')
  50     search_fields = ('cellline_name', 'nickname', 'notes')
  51     fieldsets = (
  52       (None, {
  53           'fields': (('cellline_name'),('notes'),)
  54       }),
  55      )
  56
  57 class ConditionOptions(admin.ModelAdmin):
  58     list_display = (('condition_name'), ('notes'),)
  59     fieldsets = (
  60       (None, {
  61           'fields': (('condition_name'),('nickname'),('notes'),)
  62       }),
  63      )
  64
  65 class ExperimentTypeOptions(admin.ModelAdmin):
  66   model = ExperimentType
  67   #list_display = ('name',)
  68   #fieldsets = ( (None, { 'fields': ('name',) }), )
  69
  70 class HTSUserCreationForm(UserCreationForm):
  71     class Meta:
  72         model = HTSUser
  73         fields = ("username",'first_name','last_name')
  74
  75 class HTSUserChangeForm(UserChangeForm):
  76     class Meta:
  77         model = HTSUser
  78
  79 class HTSUserOptions(UserAdmin):
  80     form = HTSUserChangeForm
  81     add_form = HTSUserCreationForm
  82
  83 class LaneLibraryInline(admin.StackedInline):
  84   model = Lane
  85   extra = 0
  86
  87 class Library_Inline(admin.TabularInline):
  88   model = Library
  89
  90 class LibraryTypeOptions(admin.ModelAdmin):
  91     model = LibraryType
  92
  93 class LibraryOptions(admin.ModelAdmin):
  94     class Media:
  95         css = {
  96             "all": ("css/wide_account_number.css",)
  97             }
  98
  99     date_hierarchy = "creation_date"
 100     save_as = True
 101     save_on_top = True
 102     search_fields = (
 103         'id',
 104         'library_name',
 105         'cell_line__cellline_name',
 106         'library_species__scientific_name',
 107         'library_species__common_name',
 108     )
 109     list_display = (
 110         'id',
 111         #'aligned_reads',
 112         #'DataRun',
 113         'library_name',
 114         'public',
 115         #'experiment_type',
 116         #'organism',
 117         #'antibody_name',
 118         #'cell_line',
 119         #'libtags',
 120         #'made_for',
 121         'affiliation',
 122         #'made_by',
 123         'undiluted_concentration',
 124         'creation_date',
 125         'stopping_point',
 126         #'condition',
 127
 128     )
 129     list_filter = (
 130         'experiment_type',
 131         'library_species',
 132         'tags',
 133         #'made_for',
 134         'affiliations',
 135         'made_by',
 136         'antibody',
 137         'cell_line',
 138         'condition',
 139         'stopping_point',
 140         'hidden')
 141     list_display_links = ('id', 'library_name',)
 142     fieldsets = (
 143       (None, {
 144         'fields': (
 145           ('id','library_name','hidden'),
 146           ('library_species'),
 147           ('library_type', 'experiment_type', 'replicate'),
 148           ('cell_line','condition','antibody'),)
 149          }),
 150          ('Creation Information:', {
 151              'fields' : (('made_for', 'made_by', 'creation_date'), ('stopping_point', 'amplified_from_sample'), ('avg_lib_size','undiluted_concentration', 'ten_nM_dilution', 'successful_pM'), 'account_number', 'notes',)
 152          }),
 153          ('Library/Project Affiliation:', {
 154              'fields' : (('affiliations'), ('tags'),)
 155          }),
 156          )
 157     inlines = [
 158       LaneLibraryInline,
 159     ]
 160     actions = ['action_print_library_labels']
 161
 162
 163     def action_print_library_labels(self, request, queryset):
 164         """
 165         Django action which prints labels for the selected set of labels from the
 166         Django Admin interface.
 167         """
 168
 169         #Probably should ask if the user really meant to print all selected
 170         # libraries if the count is above X. X=10 maybe?
 171
 172         # Grab the library template
 173         #FIXME: Hardcoding library template name. Not a good idea... *sigh*.
 174         EVIL_HARDCODED_LIBRARY_TEMPLATE_NAME = "Library"
 175
 176         try:
 177             template = PrinterTemplate.objects.get(item_type__name=EVIL_HARDCODED_LIBRARY_TEMPLATE_NAME)
 178         except PrinterTemplate.DoesNotExist:
 179             self.message_user(request, "Could not find a library template with ItemType.name of '%s'" % \
 180                               (EVIL_HARDCODED_LIBRARY_TEMPLATE_NAME))
 181             return
 182
 183         # ZPL Template
 184         t = Template(template.template)
 185
 186         #Iterate over selected labels to print
 187         for library in queryset.all():
 188
 189             # Django Template Context
 190             c = Context({'library': library})
 191
 192             # Send rendered template to the printer that the template
 193             #  object has been attached to in the database.
 194             print_zpl_socket(t.render(c), host=template.printer.ip_address)
 195
 196         self.message_user(request, "%s labels printed." % (len(queryset)))
 197
 198     action_print_library_labels.short_description = "Print Labels"
 199
 200
 201
 202     # some post 1.0.2 version of django has formfield_overrides
 203     # which would replace this code with:
 204     # formfield_overrids = {
 205     #    models.ManyToMany: { 'widget': widgets.FilteredSelectMultiple }
 206     #}
 207     def formfield_for_dbfield(self, db_field, **kwargs):
 208       if db_field.name in ('affiliations', 'tags'):
 209         kwargs['widget'] = widgets.FilteredSelectMultiple(db_field.verbose_name,
 210                                                           (db_field.name in self.filter_vertical))
 211       rv = super(LibraryOptions, self).formfield_for_dbfield(db_field, **kwargs)
 212       return rv
 213
 214 class SpeciesOptions(admin.ModelAdmin):
 215     fieldsets = (
 216       (None, {
 217           'fields': (('scientific_name', 'common_name'),)
 218       }),
 219     )
 220
 221 class TagOptions(admin.ModelAdmin):
 222     list_display = ('tag_name', 'context')
 223     fieldsets = (
 224         (None, {
 225           'fields': ('tag_name', 'context')
 226           }),
 227         )
 228
 229 admin.site.register(Affiliation, AffiliationOptions)
 230 admin.site.register(Antibody, AntibodyOptions)
 231 admin.site.register(Cellline, CelllineOptions)
 232 admin.site.register(Condition, ConditionOptions)
 233 admin.site.register(ExperimentType, ExperimentTypeOptions)
 234 admin.site.register(HTSUser, HTSUserOptions)
 235 admin.site.register(LibraryType, LibraryTypeOptions)
 236 admin.site.register(Library, LibraryOptions)
 237 admin.site.register(Species, SpeciesOptions)
 238 admin.site.register(Tag, TagOptions)