Being able to filter on paired end runs might be useful
[htsworkflow.git] / htsworkflow / frontend / experiments / admin.py
index 4181e5c91aef2aa0d2a7af9b2d0bddcc37ad65e6..6cf7aeb4c7a29a53fa4bfa4db0dc7ef4082e98c2 100644 (file)
@@ -1,10 +1,13 @@
+from itertools import chain
 from htsworkflow.frontend.experiments.models import \
      FlowCell, DataRun, DataFile, FileType, ClusterStation, Sequencer, Lane
 from django.contrib import admin
 from django.contrib.admin.widgets import FilteredSelectMultiple
 from django.forms import ModelForm
 from django.forms.fields import Field, CharField
-from django.forms.widgets import TextInput
+from django.forms.widgets import TextInput, Select
+from django.utils.encoding import force_unicode
+from django.utils.html import escape, conditional_escape
 from django.utils.translation import ugettext_lazy as _
 
 class DataFileForm(ModelForm):
@@ -34,6 +37,9 @@ class DataRunOptions(admin.ModelAdmin):
                    ('runfolder_name', 'cycle_start', 'cycle_stop'),
                    ('result_dir',),
                    ('last_update_time'),
+                   ('image_software', 'image_version'),
+                   ('basecall_software', 'basecall_version'),
+                   ('alignment_software', 'alignment_version'),
                    ('comment',))
       }),
     )
@@ -45,12 +51,12 @@ admin.site.register(DataRun, DataRunOptions)
 class FileTypeAdmin(admin.ModelAdmin):
     list_display = ('name', 'mimetype', 'regex')
 admin.site.register(FileType, FileTypeAdmin)
-  
+
 # lane form setup needs to come before Flowcell form config
 # as flowcell refers to the LaneInline class
 class LaneForm(ModelForm):
     comment = CharField(widget=TextInput(attrs={'size':'80'}), required=False)
-    
+
     class Meta:
         model = Lane
 
@@ -59,7 +65,6 @@ class LaneInline(admin.StackedInline):
     Controls display of Lanes on the Flowcell form.
     """
     model = Lane
-    max_num = 8
     extra = 8
     form = LaneForm
     raw_id_fields = ('library',)
@@ -90,8 +95,10 @@ class LaneOptions(admin.ModelAdmin):
       }),
     )
 admin.site.register(Lane, LaneOptions)
-    
+
 class FlowCellOptions(admin.ModelAdmin):
+    class Media:
+        css = { 'all': ('css/admin_flowcell.css',) }
     date_hierarchy = "run_date"
     save_on_top = True
     search_fields = ('flowcell_id',
@@ -100,7 +107,7 @@ class FlowCellOptions(admin.ModelAdmin):
         '=lane__library__id',
         'lane__library__library_name')
     list_display = ('flowcell_id','run_date','Lanes')
-    list_filter = ('sequencer','cluster_station')
+    list_filter = ('sequencer','cluster_station', 'paired_end')
     fieldsets = (
         (None, {
           'fields': ('run_date', ('flowcell_id','cluster_station','sequencer'),
@@ -115,18 +122,75 @@ class FlowCellOptions(admin.ModelAdmin):
     def formfield_for_dbfield(self, db_field, **kwargs):
         field = super(FlowCellOptions, self).formfield_for_dbfield(db_field,
                                                                    **kwargs)
+
         # Override field attributes
-        if db_field.name == "notes":
+        if db_field.name == 'sequencer':
+            # seems kind of clunky.
+            # the goal is to replace the default select/combo box with one
+            # that can strike out disabled options.
+            attrs = field.widget.widget.attrs
+            field.widget.widget = SequencerSelect(attrs=attrs, queryset=field.queryset)
+        elif db_field.name == "notes":
             field.widget.attrs["rows"] = "3"
         return field
 admin.site.register(FlowCell, FlowCellOptions)
 
 class ClusterStationOptions(admin.ModelAdmin):
-    list_display = ('name', )
-    fieldsets = ( ( None, { 'fields': ( 'name', ) } ), )
+    list_display = ('name', 'isdefault',)
+    fieldsets = ( ( None, { 'fields': ( 'name', 'isdefault') } ), )
 admin.site.register(ClusterStation, ClusterStationOptions)
 
+class SequencerSelect(Select):
+    def __init__(self, queryset=None, *args, **kwargs):
+        super(SequencerSelect, self).__init__(*args, **kwargs)
+        self.queryset = queryset
+
+    def render_options(self, choices, selected_choices):
+        # Normalize to strings.
+        selected_choices = set([force_unicode(v) for v in selected_choices])
+        output = []
+        for option_value, option_label in chain(self.choices, choices):
+            if isinstance(option_label, (list, tuple)):
+                output.append(u'<optgroup label="%s">' % escape(force_unicode(option_value)))
+                for option in option_label:
+                    output.append(self.render_option(selected_choices, *option))
+                output.append(u'</optgroup>')
+            else:
+                output.append(self.render_option(selected_choices, option_value, option_label))
+        return u'\n'.join(output)
+
+    # render_options blatently grabbed from 1.3.1 as the 1.2 version
+    # has render_option, which is what I needed to overload as a
+    # nested function in render_options
+    def render_options(self, choices, selected_choices):
+        # Normalize to strings.
+        selected_choices = set([force_unicode(v) for v in selected_choices])
+        output = []
+        for option_value, option_label in chain(self.choices, choices):
+            if isinstance(option_label, (list, tuple)):
+                output.append(u'<optgroup label="%s">' % escape(force_unicode(option_value)))
+                for option in option_label:
+                    output.append(self.render_option(selected_choices, *option))
+                output.append(u'</optgroup>')
+            else:
+                output.append(self.render_option(selected_choices, option_value, option_label))
+        return u'\n'.join(output)
+
+
+    def render_option(self, selected_choices, option_value, option_label):
+        disabled_sequencers = [ unicode(s.id) for s in self.queryset.filter(active=False) ]
+        option_value = unicode(option_value)
+        selected_html = (option_value in selected_choices) and u' selected="selected"' or ''
+        cssclass = "strikeout" if option_value in disabled_sequencers else ''
+        return u'<option class="%s" value="%s"%s>%s</option>' % (
+            cssclass, escape(option_value), selected_html,
+            conditional_escape(force_unicode(option_label)))
+
 class SequencerOptions(admin.ModelAdmin):
-    list_display = ('name', )
-    fieldsets = ( ( None, { 'fields': ( 'name', ) } ), )
+    list_display = ('name', 'active', 'isdefault', 'instrument_name', 'model')
+    fieldsets = ( ( None,
+                    { 'fields': (
+                        'name', ('active', 'isdefault'), 'instrument_name', 'serial_number',
+                        'model', 'comment') } ), )
+
 admin.site.register(Sequencer, SequencerOptions)