From 4b76634aa4ba4cf1ddce5c3cd1cd97229d2aaff0 Mon Sep 17 00:00:00 2001
From: bsimpson63 <bsimpson63@gmail.com>
Date: Tue, 28 May 2013 14:05:25 -0400
Subject: [PATCH] Move get_traffic_dates from trafficpages to promote.

---
 r2/r2/lib/pages/trafficpages.py | 15 +++------------
 r2/r2/lib/promote.py            |  8 ++++++++
 2 files changed, 11 insertions(+), 12 deletions(-)

diff --git a/r2/r2/lib/pages/trafficpages.py b/r2/r2/lib/pages/trafficpages.py
index 0de659f7c..ba6e0e44a 100644
--- a/r2/r2/lib/pages/trafficpages.py
+++ b/r2/r2/lib/pages/trafficpages.py
@@ -479,15 +479,6 @@ def _is_promo_preliminary(end_date):
     return end_date + datetime.timedelta(days=1) > now
 
 
-def get_traffic_dates(thing):
-    """Retrieve the start and end of a Promoted Link or PromoCampaign."""
-    now = datetime.datetime.now(g.tz).replace(minute=0, second=0,
-                                              microsecond=0)
-    start, end = promote.get_total_run(thing)
-    end = min(now, end)
-    return start, end
-
-
 def get_promo_traffic(thing, start, end):
     """Get traffic for a Promoted Link or PromoCampaign"""
     if isinstance(thing, Link):
@@ -511,7 +502,7 @@ def get_promo_traffic(thing, start, end):
 
 def get_billable_traffic(campaign):
     """Get traffic for dates when PromoCampaign is active."""
-    start, end = get_traffic_dates(campaign)
+    start, end = promote.get_traffic_dates(campaign)
     return get_promo_traffic(campaign, start, end)
 
 
@@ -638,7 +629,7 @@ class PromotedLinkTraffic(Templated):
 
     def check_dates(self, thing):
         """Shorten range for display and add next/prev buttons."""
-        start, end = get_traffic_dates(thing)
+        start, end = promote.get_traffic_dates(thing)
 
         if self.period:
             display_start = self.after
@@ -721,7 +712,7 @@ class PromotedLinkTraffic(Templated):
         import csv
         import cStringIO
 
-        start, end = get_traffic_dates(thing)
+        start, end = promote.get_traffic_dates(thing)
         history = cls.get_hourly_traffic(thing, start, end)
 
         out = cStringIO.StringIO()
diff --git a/r2/r2/lib/promote.py b/r2/r2/lib/promote.py
index b39768e34..3a6aeb0e8 100644
--- a/r2/r2/lib/promote.py
+++ b/r2/r2/lib/promote.py
@@ -826,6 +826,14 @@ def get_total_run(thing):
     return earliest, latest
 
 
+def get_traffic_dates(thing):
+    """Retrieve the start and end of a Promoted Link or PromoCampaign."""
+    now = datetime.now(g.tz).replace(minute=0, second=0, microsecond=0)
+    start, end = get_total_run(thing)
+    end = min(now, end)
+    return start, end
+
+
 def Run(offset=0, verbose=True):
     """reddit-job-update_promos: Intended to be run hourly to pull in
     scheduled changes to ads