htsworkflow/frontend/samples/admin.py

   1 from django.contrib import admin
   2 from django.contrib.admin import widgets
   3 from django.db import models
   4 from django.utils.translation import ugettext_lazy as _
   5
   6 from htsworkflow.frontend.samples.models import Antibody, Cellline, Condition, ExperimentType, Species, Affiliation, Library, Tag
   7
   8 class Library_Inline(admin.TabularInline):
   9   model = Library
  10
  11 class CelllineOptions(admin.ModelAdmin):
  12     list_display = ('cellline_name', 'nickname', 'notes')
  13     search_fields = ('cellline_name', 'nickname', 'notes')
  14     fieldsets = (
  15       (None, {
  16           'fields': (('cellline_name'),('notes'),)
  17       }),
  18      )
  19
  20 class ExperimentTypeOptions(admin.ModelAdmin):
  21   model = ExperimentType
  22   #list_display = ('name',)
  23   #fieldsets = ( (None, { 'fields': ('name',) }), )
  24
  25 class LibraryOptions(admin.ModelAdmin):
  26     date_hierarchy = "creation_date"
  27     save_as = True
  28     save_on_top = True
  29     search_fields = (
  30         'library_id',
  31         'library_name',
  32         'cell_line__cellline_name',
  33         'library_species__scientific_name',
  34         'library_species__common_name',
  35     )
  36     list_display = (
  37         'library_id',
  38         #'aligned_reads',
  39         #'DataRun',
  40         'library_name',
  41         'public',
  42         #'experiment_type',
  43         #'organism',
  44         #'antibody_name',
  45         #'cell_line',
  46         #'libtags',
  47         #'made_for',
  48         'affiliation',
  49         #'made_by',
  50         'undiluted_concentration',
  51         'creation_date',
  52         'stopping_point',
  53         #'condition',
  54
  55     )
  56     list_filter = (
  57         'experiment_type',
  58         'library_species',
  59         'tags',
  60         #'made_for',
  61         'affiliations',
  62         'made_by',
  63         'antibody',
  64         'cell_line',
  65         'condition',
  66         'stopping_point',
  67         'hidden')
  68     list_display_links = ('library_id', 'library_name',)
  69     fieldsets = (
  70       (None, {
  71         'fields': (
  72           ('library_id','library_name','hidden'),
  73           ('library_species'),
  74           ('experiment_type', 'replicate'),
  75           ('cell_line','condition','antibody'),)
  76          }),
  77          ('Creation Information:', {
  78              'fields' : (('made_for', 'made_by', 'creation_date'), ('stopping_point', 'amplified_from_sample'), ('avg_lib_size','undiluted_concentration', 'ten_nM_dilution', 'successful_pM'), 'notes',)
  79          }),
  80          ('Library/Project Affiliation:', {
  81              'fields' : (('affiliations'), ('tags'),)
  82          }),
  83          )
  84
  85     # some post 1.0.2 version of django has formfield_overrides
  86     # which would replace this code with:
  87     # formfield_overrids = {
  88     #   models.ManyToMany: { 'widget': widgets.FilteredSelectMultiple }
  89     # }
  90     def formfield_for_dbfield(self, db_field, **kwargs):
  91       if db_field.name == 'affiliations':
  92         kwargs['widget'] = widgets.FilteredSelectMultiple(db_field.verbose_name, (db_field.name in self.filter_vertical))
  93       rv = super(LibraryOptions, self).formfield_for_dbfield(db_field, **kwargs)
  94       print db_field.name, kwargs
  95       return rv
  96
  97 class AffiliationOptions(admin.ModelAdmin):
  98     list_display = ('name','contact','email')
  99     fieldsets = (
 100       (None, {
 101           'fields': (('name','contact','email'))
 102       }),
 103     )
 104
 105 # class UserOptions(admin.ModelAdmin):
 106 #   inlines = [Library_Inline]
 107
 108 class AntibodyOptions(admin.ModelAdmin):
 109     search_fields = ('antigene','nickname','catalog','antibodies','source','biology','notes')
 110     list_display = ('antigene','nickname','antibodies','catalog','source','biology','notes')
 111     list_filter = ('antibodies','source')
 112     fieldsets = (
 113       (None, {
 114           'fields': (('antigene','nickname','antibodies'),('catalog','source'),('biology'),('notes'))
 115       }),
 116      )
 117
 118 class SpeciesOptions(admin.ModelAdmin):
 119     fieldsets = (
 120       (None, {
 121           'fields': (('scientific_name', 'common_name'),)
 122       }),
 123     )
 124
 125 class ConditionOptions(admin.ModelAdmin):
 126     list_display = (('condition_name'), ('notes'),)
 127     fieldsets = (
 128       (None, {
 129           'fields': (('condition_name'),('nickname'),('notes'),)
 130       }),
 131      )
 132
 133 class TagOptions(admin.ModelAdmin):
 134     list_display = ('tag_name', 'context')
 135     fieldsets = (
 136         (None, {
 137           'fields': ('tag_name', 'context')
 138           }),
 139         )
 140
 141 admin.site.register(Affiliation, AffiliationOptions)
 142 admin.site.register(Antibody, AntibodyOptions)
 143 admin.site.register(Cellline, CelllineOptions)
 144 admin.site.register(Condition, ConditionOptions)
 145 admin.site.register(ExperimentType, ExperimentTypeOptions)
 146 admin.site.register(Library, LibraryOptions)
 147 admin.site.register(Species, SpeciesOptions)
 148 admin.site.register(Tag, TagOptions)