From c139744886a36450450fa38daefeb682a87832ea Mon Sep 17 00:00:00 2001
From: KeyserSosa <chris@reddit.com>
Date: Thu, 16 Apr 2009 15:41:15 -0700
Subject: [PATCH] allow subreddit moderators to see their subreddit's traffic. 
 Allow sponsors to see traffic of promoted links.

---
 r2/r2/controllers/front.py           |  4 ++--
 r2/r2/lib/pages/pages.py             |  8 ++++++-
 r2/r2/lib/traffic.py                 | 16 +++++++++----
 r2/r2/templates/promotedtraffic.html | 34 ++++++++++++++++++++++++----
 r2/r2/templates/reddittraffic.html   |  6 ++---
 5 files changed, 54 insertions(+), 14 deletions(-)

diff --git a/r2/r2/controllers/front.py b/r2/r2/controllers/front.py
index 342712f48..4abd05c0f 100644
--- a/r2/r2/controllers/front.py
+++ b/r2/r2/controllers/front.py
@@ -290,7 +290,7 @@ class FrontController(RedditController):
 #            listing = LinkListing(builder)
 #            pane = listing.listing()
             pane = InfoBar(message = "There doesn't seem to be anything here.")
-        elif c.user_is_admin and location == 'traffic':
+        elif is_moderator and location == 'traffic':
             pane = RedditTraffic()
         else:
             return self.abort404()
@@ -599,7 +599,7 @@ class FrontController(RedditController):
             return self.abort404()
 
 
-    @validate(VAdmin(),
+    @validate(VSponsor(),
               article = VLink('article'))
     def GET_traffic(self, article):
         res = LinkInfoPage(link = article,
diff --git a/r2/r2/lib/pages/pages.py b/r2/r2/lib/pages/pages.py
index ae539b9c8..0e8ee7a33 100644
--- a/r2/r2/lib/pages/pages.py
+++ b/r2/r2/lib/pages/pages.py
@@ -270,7 +270,6 @@ class SubredditInfoBar(Wrapped):
             buttons.append(NamedButton('edit'))
             buttons.extend([NavButton(menu.banusers, 'banned'),
                             NamedButton('spam')])
-        if c.user_is_admin:
             buttons.append(NamedButton('traffic'))
         return [NavMenu(buttons, type = "flatlist", base_path = "/about/")]
 
@@ -457,6 +456,7 @@ class LinkInfoPage(Reddit):
 
         if c.user_is_admin:
             buttons += [info_button('details')]
+        if c.user_is_sponsor:
             if self.link.promoted is not None:
                 buttons += [info_button('traffic')]
 
@@ -1245,6 +1245,12 @@ class PromotedTraffic(Traffic):
         self.traffic = load_traffic('hour', "thing", thing._fullname,
                                     start_time = d, stop_time = until)
 
+        self.totals =  load_traffic('month', "thing", thing._fullname)
+        if not self.totals:
+            self.totals = load_traffic('day', "thing", thing._fullname)
+        if self.totals:
+            self.totals = map(sum, zip(*zip(*self.totals)[1]))
+               
         imp = self.slice_traffic(self.traffic, 0, 1)
 
         if len(imp) > 2:
diff --git a/r2/r2/lib/traffic.py b/r2/r2/lib/traffic.py
index df9a64055..873093654 100644
--- a/r2/r2/lib/traffic.py
+++ b/r2/r2/lib/traffic.py
@@ -23,7 +23,7 @@ from httplib import HTTPConnection
 from urlparse import urlparse
 from cPickle import loads
 from utils import query_string
-import os, socket
+import os, socket, time, datetime
 from pylons import g
 from r2.lib.memoize import memoize, clear_memo
 
@@ -64,9 +64,17 @@ def load_traffic_uncached(interval, what, iden,
 def load_traffic(interval, what, iden, 
                  start_time = None, stop_time = None,
                  npoints = None):
-    return load_traffic_uncached(interval, what, iden, 
-                 start_time = start_time, stop_time = stop_time,
-                 npoints = npoints)
+    res = load_traffic_uncached(interval, what, iden, 
+                                start_time = start_time, stop_time = stop_time,
+                                npoints = npoints)
+
+    if res and isinstance(res[0][0], datetime.datetime):
+        res = zip(*res)
+        res[0] = [x.replace(tzinfo=None) - datetime.timedelta(0, time.timezone)
+                  for x in res[0]]
+        res = zip(*res)
+    return res
+    
 
 def load_summary(what, interval = "month", npoints = 50):
     return load_traffic(interval, "summary", what, npoints = npoints)
diff --git a/r2/r2/templates/promotedtraffic.html b/r2/r2/templates/promotedtraffic.html
index 1c4958555..79fdfabe4 100644
--- a/r2/r2/templates/promotedtraffic.html
+++ b/r2/r2/templates/promotedtraffic.html
@@ -19,6 +19,12 @@
 ## All portions of the code written by CondeNet are Copyright (c) 2006-2009
 ## CondeNet, Inc. All Rights Reserved.
 ################################################################################
+<%!
+   import locale
+   from r2.models.subreddit import DomainSR, FakeSubreddit
+   def num(x):
+      return locale.format('%d', x, True)
+ %>
 
 %if thing.traffic:
   %if thing.imp_graph:
@@ -34,6 +40,7 @@
       <th>date</th>
       <th colspan="2">Impresions</th>
       <th colspan="2">Clicks</th>
+      <th colspan="2">click-thru (%)</th>
     </tr>
     <tr>
       <th></th>
@@ -41,15 +48,34 @@
       <th>total</th>
       <th>unique</th>
       <th>total</th>
+      <th>unique</th>
+      <th>total</th>
     </tr>
     %for date, (uimp, nimp, ucli, ncli) in thing.traffic:
       <tr>
         <td>${date.strftime("%Y-%m-%d %H:%M")}</td>
-        <td>${uimp}</td>
-        <td>${nimp}</td>
-        <td>${ucli}</td>
-        <td>${ncli}</td>
+        <td>${num(uimp)}</td>
+        <td>${num(nimp)}</td>
+        <td>${num(ucli)}</td>
+        <td>${num(ncli)}</td>
+        <td>${("%.2f%%" % (float(100*ucli) / uimp)) if nimp else "--.--%"}</td>
+        <td>${("%.2f%%" % (float(100*ncli) / nimp)) if nimp else "--.--%"}</td>
       </tr>
     %endfor
+    %if thing.totals:
+      <% 
+         uimp, nimp, ucli, ncli = thing.totals
+       %>
+     <tr style="font-weight:bold">
+        <td>total (users approx)</td>
+        <td>${num(uimp)}</td>
+        <td>${num(nimp)}</td>
+        <td>${num(ucli)}</td>
+        <td>${num(ncli)}</td>
+        <td>${("%.2f%%" % (float(100*ucli) / uimp)) if nimp else "--.--%"}</td>
+        <td>${("%.2f%%" % (float(100*ncli) / nimp)) if nimp else "--.--%"}</td>
+      </tr>
+      
+    %endif
   </table>
 %endif
diff --git a/r2/r2/templates/reddittraffic.html b/r2/r2/templates/reddittraffic.html
index a6e826e63..bad853461 100644
--- a/r2/r2/templates/reddittraffic.html
+++ b/r2/r2/templates/reddittraffic.html
@@ -110,16 +110,16 @@
 
   <div style="float:left">
     <div>
-      <img class="traffic-graph" alt='hourly uniques' style="float:right;"
+      <img class="traffic-graph" alt='hourly uniques'
            src="${thing.uniques_hour}"/>
     </div>
     <div>
-      <img class="traffic-graph" alt='daily uniques' style="float:right;"
+      <img class="traffic-graph" alt='daily uniques'
            src="${thing.uniques_day}"/>
     </div>
     %if c.default_sr:
       <div>
-        <img class="traffic-graph" alt='monthly uniques' style="float:right;"
+        <img class="traffic-graph" alt='monthly uniques'
              src="${thing.uniques_month}"/>
       </div>
     %endif