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