htsworkflow/frontend/samples/admin.py

   1 from django.contrib import admin
   2 from django.contrib.admin import widgets
   3 from django.contrib.auth.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         'experiment_type',
 129         'library_type',
 130         #'cell_line',
 131         'stopping_point',
 132         'made_by',
 133         'library_species',
 134         'affiliations',
 135         )
 136     list_display_links = ('id', 'library_name',)
 137     fieldsets = (
 138       (None, {
 139         'fields': (
 140           ('id','library_name','hidden'),
 141           ('library_species', 'experiment_type'),
 142           ('library_type', 'multiplex_id'),
 143           )
 144          }),
 145          ('Experiment Detail:', {
 146             'fields': (('cell_line', 'replicate',),
 147                        ('condition',),
 148                        ('antibody', ),
 149                        ),
 150             'classes': ('collapse',),
 151             }),
 152          ('Creation Information:', {
 153              'fields' : (('made_by', 'creation_date', 'stopping_point'),
 154                          ('amplified_from_sample'),
 155                          ('gel_cut_size', 'insert_size',
 156                           'undiluted_concentration'),
 157                          ('bioanalyzer_concentration','bioanalyzer_image_url'),
 158                          ('bioanalyzer_summary'),
 159                          ('notes'))
 160          }),
 161          ('Library/Project Affiliation:', {
 162              'fields' : (('account_number', 'affiliations'),)
 163          }),
 164          )
 165     inlines = [
 166       LaneLibraryInline,
 167     ]
 168     actions = ['action_print_library_labels']
 169
 170     def action_print_library_labels(self, request, queryset):
 171         """
 172         Django action which prints labels for the selected set of labels from the
 173         Django Admin interface.
 174         """
 175
 176         #Probably should ask if the user really meant to print all selected
 177         # libraries if the count is above X. X=10 maybe?
 178
 179         # Grab the library template
 180         #FIXME: Hardcoding library template name. Not a good idea... *sigh*.
 181         EVIL_HARDCODED_LIBRARY_TEMPLATE_NAME = "Library"
 182
 183         try:
 184             template = PrinterTemplate.objects.get(item_type__name=EVIL_HARDCODED_LIBRARY_TEMPLATE_NAME)
 185         except PrinterTemplate.DoesNotExist:
 186             self.message_user(request, "Could not find a library template with ItemType.name of '%s'" % \
 187                               (EVIL_HARDCODED_LIBRARY_TEMPLATE_NAME))
 188             return
 189
 190         # ZPL Template
 191         t = Template(template.template)
 192
 193         zpl_list = []
 194         #Iterate over selected labels to print
 195         for library in queryset.all():
 196
 197             # Django Template Context
 198             c = Context({'library': library})
 199
 200             # Send rendered template to the printer that the template
 201             #  object has been attached to in the database.
 202             zpl_list.append(t.render(c))
 203
 204         print_zpl_socket(zpl_list, host=template.printer.ip_address)
 205
 206         self.message_user(request, "%s labels printed." % (len(queryset)))
 207
 208     action_print_library_labels.short_description = "Print Labels"
 209
 210     def formfield_for_dbfield(self, db_field, **kwargs):
 211         # Override Field type
 212         if db_field.name in ('affiliations', 'tags'):
 213             kwargs['widget'] = widgets.FilteredSelectMultiple(
 214                 db_field.verbose_name,
 215                 (db_field.name in self.filter_vertical)
 216             )
 217         field = super(LibraryOptions, self).formfield_for_dbfield(db_field,
 218                                                                   **kwargs)
 219         # Override field attributes
 220         if db_field.name == "bioanalyzer_summary":
 221             field.widget.attrs["rows"] = "3"
 222         return field
 223
 224 class SpeciesOptions(admin.ModelAdmin):
 225     fieldsets = (
 226       (None, {
 227           'fields': (('scientific_name', 'common_name'),)
 228       }),
 229     )
 230
 231 class TagOptions(admin.ModelAdmin):
 232     list_display = ('tag_name', 'context')
 233     fieldsets = (
 234         (None, {
 235           'fields': ('tag_name', 'context')
 236           }),
 237         )
 238
 239 admin.site.register(Library, LibraryOptions)
 240 admin.site.register(Affiliation, AffiliationOptions)
 241 admin.site.register(Antibody, AntibodyOptions)
 242 admin.site.register(Cellline, CelllineOptions)
 243 admin.site.register(Condition, ConditionOptions)
 244 admin.site.register(ExperimentType, ExperimentTypeOptions)
 245 #admin.site.register(HTSUser, HTSUserOptions)
 246 admin.site.register(LibraryType, LibraryTypeOptions)
 247 admin.site.register(MultiplexIndex, MultiplexIndexOptions)
 248 admin.site.register(Species, SpeciesOptions)
 249 #admin.site.register(Tag, TagOptions)