Apparently I should've rendered the emails in plain text.

[htsworkflow.git] / htsworkflow / frontend / experiments / views.py

diff --git a/htsworkflow/frontend/experiments/views.py b/htsworkflow/frontend/experiments/views.py
index 38880166cd00fe644624c4d382608f288ec74a9d..3505a3bf594e7cfbd0b50afce09f6374e504280d 100755 (executable)
--- a/htsworkflow/frontend/experiments/views.py
+++ b/htsworkflow/frontend/experiments/views.py
@@ -1,27 +1,35 @@
 # Create your views here.
 #from django.template import Context, loader
 #shortcut to the above modules
-from django.shortcuts import render_to_response, get_object_or_404
-#from htswfrontend.fctracker.models import *
-from htswfrontend.exp_track.models import *
-from django.http import HttpResponse
+from django.contrib.auth.decorators import user_passes_test
 from django.core.exceptions import ObjectDoesNotExist
+from django.core.mail import EmailMessage, mail_managers
+from django.http import HttpResponse
+from django.shortcuts import render_to_response, get_object_or_404
+from django.template import Context
+from django.template.loader import get_template
+
+from htsworkflow.frontend.experiments.models import *
+from htsworkflow.frontend.experiments.experiments import \
+     estimateFlowcellDuration, \
+     getUsersForFlowcell, \
+     makeEmailLaneMap
 
 def index(request):
-    all_runs = DataRun.objects.all().order_by('-run_start_time')
-    #t = loader.get_template('exptrack/index.html')
+    all_runs = DataRun.objects.order_by('-run_start_time')
+    #t = loader.get_template('experiments/index.html')
     #c = Context({
     #    'data_run_list': all_runs,
     #})
     #return HttpResponse(t.render(c)) 
     # shortcut to the above module usage
-    return render_to_response('exptrack/index.html',{'data_run_list': all_runs}) 
+    return render_to_response('experiments/index.html',{'data_run_list': all_runs}) 
     
 def detail(request, run_folder):
     html_str = '<h2>Exp Track Details Page</h2>'
     html_str += 'Run Folder: '+run_folder
     r = get_object_or_404(DataRun,run_folder=run_folder)
-    return render_to_response('exptrack/detail.html',{'run_f': r})
+    return render_to_response('experiments/detail.html',{'run_f': r})
 
 def makeFCSheet(request,fcid):
   # get Flowcell by input fcid
@@ -32,29 +40,77 @@ def makeFCSheet(request,fcid):
   except ObjectDoesNotExist:
     pass
   lanes = ['1','2','3','4','5','6','7','8']
-  return render_to_response('exptrack/flowcellSheet.html',{'fc': rec})
-
-def test_Libs(request):
-  str = ''
-  str += '<table border=1><tr><th>Lib ID</th><th>Current Libaray Name (Free Text)</th><th>Auto-composed Libaray Name (antibody + celline + libid + species + [replicate])</th></tr>'
-  allLibs = Library.objects.all()
-  #allLibs = Library.objects.filter(antibody__isnull=False)
-  for L in allLibs:
-    str += '<tr>'
-    str += '<td>'+L.library_id+'</td><td>'+L.library_name+'</td>'   
-    str += '<td>'
-    str += L.experiment_type+'_'
-    if L.cell_line.cellline_name != 'Unknown':
-      str += L.cell_line.cellline_name+'_'
-
-    try:
-      if L.antibody is not None:
-        str += L.antibody.nickname + '_'
-    except Antibody.DoesNotExist:
-      pass
-  
-    str += 'Rep'+L.replicate.__str__()
-    str += '</td></tr>' 
-
-  str += '</table>'
-  return HttpResponse(str)  
+  return render_to_response('experiments/flowcellSheet.html',{'fc': rec})
+
+
+@user_passes_test(lambda u: u.is_staff)
+def startedEmail(request, pk):
+    """
+    Create the we have started processing your samples email
+    """
+    fc = get_object_or_404(FlowCell, id=pk)
+
+    send = request.REQUEST.get('send',False)
+    if send in ('1', 'on', 'True', 'true', True):
+        send = True
+    else:
+        send = False
+
+    bcc_managers = request.REQUEST.get('bcc', False)
+    if bcc_managers in ('on', '1', 'True', 'true'):
+        bcc_managers = True
+    else:
+        bcc_managers = False
+
+    email_lane = makeEmailLaneMap(fc)
+    flowcell_users = getUsersForFlowcell(fc)
+    estimate_low, estimate_high = estimateFlowcellDuration(fc)
+    email_verify = get_template('experiments/email_preview.html')
+    email_template = get_template('experiments/started_email.txt')
+    sender = settings.NOTIFICATION_SENDER
+
+    warnings = []
+    emails = []
+
+    emailless_users = []
+    for user in flowcell_users:
+        # provide warning
+        if user.email is None or len(user.email) == 0:
+            warnings.append((user.admin_url(), user.username))
+    user=None
+    
+    for user_email in email_lane.keys():
+        sending = ""
+        # build body
+        context = Context({u'flowcell': fc,
+                   u'lanes': email_lane[user_email],
+                   u'runfolder': 'blank',
+                   u'finish_low': estimate_low,
+                   u'finish_high': estimate_high,
+                  })
+
+        # build view
+        subject = "Flowcell  %s" % ( fc.flowcell_id )
+        body = email_template.render(context)
+
+        if send:
+            email = EmailMessage(subject, body, sender, to=[user_email])
+            if bcc_managers:
+                email.bcc = settings.MANAGERS
+            email.send()
+
+        emails.append((user_email, subject, body, sending))
+
+    verify_context = Context({
+        'send': send,
+        'warnings': warnings,
+        'emails': emails,
+        'from': sender,
+        'site_managers': settings.MANAGERS,
+        })
+    return HttpResponse(email_verify.render(verify_context))
+    
+def finishedEmail(request, pk):
+    """
+    """
+    return HttpResponse("I've got nothing.")