From 4eee1c124ee0c42b926e9eb4dfc62de67c33db10 Mon Sep 17 00:00:00 2001
From: Nathan Shafer <nate@torzo.com>
Date: Tue, 14 Oct 2014 15:50:14 -0700
Subject: [PATCH] Added support for aggregated fields in CSV output.

---
 djqscsv/djqscsv.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/djqscsv/djqscsv.py b/djqscsv/djqscsv.py
index 42c52fb..bedc703 100644
--- a/djqscsv/djqscsv.py
+++ b/djqscsv/djqscsv.py
@@ -90,6 +90,10 @@ def write_csv(queryset, file_obj, **kwargs):
     if extra_columns:
         field_names += extra_columns
 
+    aggregate_columns = list(values_qs.query.aggregate_select)
+    if aggregate_columns:
+        field_names += aggregate_columns
+
     if field_order:
         # go through the field_names and put the ones
         # that appear in the ordering list first