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