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