Revert "Update StatsViewModel.kt"

This reverts commit 7c02ba5.
wordpress-mobile · Feb 28, 2024 · 5fe88d0 · 5fe88d0
1 parent 7c02ba5
commit 5fe88d0
Showing 1 changed file with 1 addition and 14 deletions.
diff --git a/WordPress/src/main/java/org/wordpress/android/ui/stats/refresh/StatsViewModel.kt b/WordPress/src/main/java/org/wordpress/android/ui/stats/refresh/StatsViewModel.kt
@@ -140,20 +140,7 @@ class StatsViewModel
     }
 
     private fun getInitialTimeFrame(intent: Intent): StatsSection? {
-        val timeframe = intent.getSerializableExtraCompat<StatsTimeframe>(StatsActivity.ARG_DESIRED_TIMEFRAME)
-        if (statsTrafficTabFeatureConfig.isEnabled()) {
-            if (timeframe == DAY || timeframe == WEEK || timeframe == MONTH || timeframe == YEAR) {
-                when (timeframe) {
-                    DAY -> selectedTrafficGranularityManager.setSelectedTrafficGranularity(StatsGranularity.DAYS)
-                    WEEK -> selectedTrafficGranularityManager.setSelectedTrafficGranularity(StatsGranularity.WEEKS)
-                    MONTH -> selectedTrafficGranularityManager.setSelectedTrafficGranularity(StatsGranularity.MONTHS)
-                    YEAR -> selectedTrafficGranularityManager.setSelectedTrafficGranularity(StatsGranularity.YEARS)
-                    else -> { /* Do nothing */}
-                }
-            }
-        }
-
-        return when (timeframe) {
+        return when (intent.getSerializableExtraCompat<Serializable>(StatsActivity.ARG_DESIRED_TIMEFRAME)) {
             StatsTimeframe.INSIGHTS -> StatsSection.INSIGHTS
             DAY -> StatsSection.DAYS
             WEEK -> StatsSection.WEEKS