writes csv export logic adapted from StorageLocation

Author: john-huang-121

app/controllers/audits_controller.rb

@@ -7,6 +7,14 @@ def index
     @selected_location = filter_params[:at_location]
     @audits = current_organization.audits.includes(:line_items, :storage_location).class_filter(filter_params)
     @storage_locations = StorageLocation.with_audits_for(current_organization).select(:id, :name)
+
+    respond_to do |format|
+      format.html
+      format.csv do
+        inventory = View::Inventory.new(current_organization.id)
+        send_data Audit.generate_csv_from_inventory(@audits, inventory), filename: "Audits-#{Time.zone.today}.csv"
+      end
+    end
   end
 
   def show

app/models/audit.rb

@@ -19,6 +19,7 @@ class Audit < ApplicationRecord
   belongs_to :storage_location
   belongs_to :adjustment, optional: true
 
+  include Exportable
   include Itemizable
   include Filterable
   scope :at_location, ->(location_id) { where(storage_location_id: location_id) }
@@ -49,6 +50,28 @@ def user_is_organization_admin_of_the_organization
     end
   end
 
+  def self.generate_csv_from_inventory(audits, inventory)
+    all_items = inventory.all_items.uniq(&:item_id).sort_by(&:name)
+    additional_headers = all_items.map(&:name).uniq
+    headers = csv_export_headers + additional_headers
+
+    CSV.generate(write_headers: true, headers: headers) do |csv|
+      audits.map do |audit|
+        sl = audit.storage_location # preloaded earlier with includes to avoid n+1
+        total_quantity = inventory.quantity_for(storage_location: sl.id)
+
+        row = [audit.updated_at.strftime("%B %d %Y"), audit.status, sl.name, sl.address, sl.square_footage, sl.warehouse_type, total_quantity] +
+          all_items.map { |i| inventory.quantity_for(storage_location: sl.id, item_id: i.item_id) }
+
+        csv << row.map { |attr| normalize_csv_attribute(attr) }
+      end
+    end
+  end
+
+  def self.csv_export_headers
+    ["Audit Date", "Audit Status", "Name", "Address", "Square Footage", "Warehouse Type", "Total Inventory"]
+  end
+
   private
 
   def line_items_unique_by_item_id

app/views/audits/index.html.erb

@@ -46,12 +46,13 @@
                 <%= filter_button %>
                 <%= clear_filter_button %>
 
-                <div class="btn-group pull-right">
+                <span class="float-right">
+                  <%= download_button_to(audits_path(format: :csv), {text: "Export Audit", size: "md"}) %>
                   <%= new_button_to new_audit_path, {text: "New Audit"} %>
-                </div>
+                </span>
             <% end # form %>
             </div>
-            </div>
+          </div>
         <!-- /.card -->
       </div>
       <!--/.col (left) -->