scope work unit export by client/project

ironmtntech · Jun 26, 2018 · cd78980 · cd78980
1 parent 4c814e9
commit cd78980
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/app/controllers/clients_controller.rb b/app/controllers/clients_controller.rb
@@ -86,8 +86,14 @@ def show
   end
 
   def export_work_units
-    @client = Client.find(params[:client_id])
-    @work_units = WorkUnit.for_client(@client)
+    require 'pry';binding.pry
+    if params[:scope] == 'projects'
+      @project = Project.find(params[:project_id])
+      @work_units = WorkUnit.for_project(@project)
+    elsif params[:scope] == 'clients'
+      @client = Client.find(params[:client_id])
+      @work_units = WorkUnit.for_client(@client)
+    end
     _params = params[:export_date]
     @start_date = Date.parse("#{_params['start_date(1i)']}/#{_params['start_date(2i)']}/#{_params['start_date(3i)']}")
     @end_date = Date.parse("#{_params['end_date(1i)']}/#{_params['end_date(2i)']}/#{_params['end_date(3i)']}") + 1.days

diff --git a/app/views/shared/_work_unit_index.html.haml b/app/views/shared/_work_unit_index.html.haml
@@ -4,7 +4,12 @@
   = date_select :export_date, :start_date
   = label_tag "End Date"
   = date_select :export_date, :end_date
-  = hidden_field_tag :client_id, params[:id]
+  - if params[:controller] == 'projects'
+    = hidden_field_tag :scope, 'projects'
+    = hidden_field_tag :project_id, params[:id]
+  - elsif params[:controller] == 'clients'
+    = hidden_field_tag :scope, 'clients'
+    = hidden_field_tag :client_id, params[:id]
   %br/
   %br/
   = submit_tag 'Export Selected Date Range', :class => "data_export_icon"