Cohorts: Group by quarter

README.md

@@ -226,7 +226,7 @@ You can also use an array or hash for static data and enums.
 
 ```yml
 smart_variables:
-  period: ["day", "week", "month"]
+  period: ["day", "week", "month", "quarter"]
   status: {0: "Active", 1: "Archived"}
 ```
 

app/controllers/blazer/base_controller.rb

@@ -59,7 +59,7 @@ def refresh_query(query)
 
     def add_cohort_analysis_vars
       @bind_vars << "cohort_period" unless @bind_vars.include?("cohort_period")
-      @smart_vars["cohort_period"] = ["day", "week", "month"] if @smart_vars
+      @smart_vars["cohort_period"] = ["day", "week", "month", "quarter"] if @smart_vars
       # TODO create var_params method
       request.query_parameters["cohort_period"] ||= "week"
     end
@@ -87,7 +87,7 @@ def parse_smart_variables(var, data_source)
 
     def cohort_analysis_statement(statement)
       @cohort_period = params["cohort_period"] || "week"
-      @cohort_period = "week" unless ["day", "week", "month"].include?(@cohort_period)
+      @cohort_period = "week" unless ["day", "week", "month", "quarter"].include?(@cohort_period)
 
       # for now
       @conversion_period = @cohort_period
@@ -99,6 +99,8 @@ def cohort_analysis_statement(statement)
           7
         when "month"
           30
+        when "quarter"
+          90
         end
 
       statement.apply_cohort_analysis(period: @cohort_period, days: @cohort_days)

app/controllers/blazer/queries_controller.rb

@@ -440,6 +440,8 @@ def render_cohort_analysis
               current_date - 1
             when "week"
               current_date - 7
+            when "quarter"
+              current_date.prev_month(3)
             else
               current_date.prev_month
             end

lib/blazer/adapters/sql_adapter.rb

@@ -153,6 +153,9 @@ def cohort_analysis_statement(statement, period:, days:)
           when "week"
             date_sql = "CAST(DATE_FORMAT(#{time_sql} - INTERVAL ((5 + DAYOFWEEK(#{time_sql})) % 7) DAY, '%Y-%m-%d') AS DATE)"
             date_params = [tzname, tzname]
+          when "quarter"
+            date_sql = "CAST(DATE_FORMAT(CONCAT(YEAR(#{time_sql}), '-', QUARTER(#{time_sql}) * 3 - 2, '-01'), '%Y-%m-%d') AS DATE)"
+            date_params = [tzname]
           else
             date_sql = "CAST(DATE_FORMAT(#{time_sql}, '%Y-%m-01') AS DATE)"
             date_params = [tzname]

lib/generators/blazer/templates/config.yml.tt

@@ -23,7 +23,7 @@ data_sources:
 
     smart_variables:
       # zone_id: "SELECT id, name FROM zones ORDER BY name ASC"
-      # period: ["day", "week", "month"]
+      # period: ["day", "week", "month", "quarter"]
       # status: {0: "Active", 1: "Archived"}
 
     linked_columns:

test/cohort_analysis_test.rb

@@ -19,4 +19,11 @@ def test_cohort_period_default
     assert_response :success
     assert_match %{selected="selected" value="week"}, response.body
   end
+
+  def test_quarter
+    query = create_query(statement: "SELECT 1 AS user_id, NOW() AS conversion_time /* cohort analysis */")
+    get blazer.query_path(query)
+    assert_response :success
+    assert_match %{value="quarter"}, response.body
+  end
 end

test/internal/config/blazer.yml

@@ -23,7 +23,7 @@ data_sources:
 
     smart_variables:
       # zone_id: "SELECT id, name FROM zones ORDER BY name ASC"
-      period: ["day", "week", "month"]
+      period: ["day", "week", "month", "quarter"]
       # status: {0: "Active", 1: "Archived"}
 
     linked_columns: