# -*- coding: iso-8859-1 -*-

 """

     MoinMoin - Hitcount Statistics

     This macro creates a hitcount chart from the data in "event.log".

     TODO: refactor to use a class, this code is ugly.

           A lot of code here is duplicated in stats.useragents.

           Maybe both can use same base class, maybe some parts are useful to other code.

     @copyright: 2002-2004 Juergen Hermann <jh@web.de>,

                 2007 MoinMoin:ThomasWaldmann

     @license: GNU GPL, see COPYING for details.

 """

 _debug = 0

 import time

 from MoinMoin import caching, wikiutil, logfile

 from MoinMoin.Page import Page

 from MoinMoin.logfile import eventlog

 # this is a CONSTANT used for on-disk caching, it must NOT be configurable and

 # not depend on request.user!

 DATE_FMT = '%04d-%02d-%02d' # % (y, m, d)

 def linkto(pagename, request, params=''):

     _ = request.getText

     if not request.cfg.chart_options:

         return text(pagename, request, params)

     if _debug:

         return draw(pagename, request)

     page = Page(request, pagename)

     # Create escaped query string from dict and params

     querystr = {'action': 'chart', 'type': 'hitcounts'}

     querystr = wikiutil.makeQueryString(querystr)

     querystr = wikiutil.escape(querystr)

     if params:

         querystr += '&' + params

     data = {'url': page.url(request, querystr)}

     data.update(request.cfg.chart_options)

     result = ('<img src="%(url)s" width="%(width)d" height="%(height)d"'

               ' alt="hitcounts chart">') % data

     return result

 def get_data(pagename, request, filterpage=None):

     cache_days, cache_views, cache_edits = [], [], []

     cache_date = 0

     # Get results from cache

     if filterpage:

         arena = Page(request, pagename)

         cache = caching.CacheEntry(request, arena, 'hitcounts', scope='item', use_pickle=True)

     else:

         arena = 'charts'

         cache = caching.CacheEntry(request, arena, 'hitcounts', scope='wiki', use_pickle=True)

     if cache.exists():

         try:

             cache_date, cache_days, cache_views, cache_edits = cache.content()

         except:

             cache.remove() # cache gone bad

     # Get new results from the log

     log = eventlog.EventLog(request)

     try:

         new_date = log.date()

     except logfile.LogMissing:

         new_date = None

     # prepare data

     days = []

     views = []

     edits = []

     ratchet_day = None

     ratchet_time = None

     if new_date is not None:

         log.set_filter(['VIEWPAGE', 'SAVEPAGE'])

         latest = None

         for event in log.reverse():

             # don't use event_log.date()

             if latest is None:

                 latest = event[0]

             event_usecs = event[0]

             if event_usecs <= cache_date:

                 break

             eventpage = event[2].get('pagename', '')

             if filterpage and eventpage != filterpage:

                 continue

             event_secs = wikiutil.version2timestamp(event_usecs)

             time_tuple = time.gmtime(event_secs) # must be UTC

             day = tuple(time_tuple[0:3])

             if day != ratchet_day:

                 # new day

                 while ratchet_time:

                     ratchet_time -= 86400 # seconds per day

                     rday = tuple(time.gmtime(ratchet_time)[0:3]) # must be UTC

                     if rday <= day:

                         break

                     days.append(DATE_FMT % rday)

                     views.append(0)

                     edits.append(0)

                 days.append(DATE_FMT % day)

                 views.append(0)

                 edits.append(0)

                 ratchet_day = day

                 ratchet_time = event_secs

             if event[1] == 'VIEWPAGE':

                 views[-1] += 1

             elif event[1] == 'SAVEPAGE':

                 edits[-1] += 1

         days.reverse()

         views.reverse()

         edits.reverse()

     # merge the day on the end of the cache

     if cache_days and days and days[0] == cache_days[-1]:

         cache_edits[-1] += edits[0]

         cache_views[-1] += views[0]

         days, views, edits = days[1:], views[1:], edits[1:]

     # Update and save the cache

     cache_days.extend(days)

     cache_views.extend(views)

     cache_edits.extend(edits)

     if new_date is not None:

         cache.update((latest, cache_days, cache_views, cache_edits))

     return cache_days, cache_views, cache_edits

 def text(pagename, request, params=''):

     from MoinMoin.util.dataset import TupleDataset, Column

     from MoinMoin.widget.browser import DataBrowserWidget

     _ = request.getText

     # check params

     filterpage = None

     if params.startswith('page='):

         filterpage = wikiutil.url_unquote(params[len('page='):])

     if request and request.values and 'page' in request.values:

         filterpage = request.values['page']

     days, views, edits = get_data(pagename, request, filterpage)

     hits = TupleDataset()

     hits.columns = [Column('day', label=_("Date"), align='left'),

                     Column('views', label=_("Views/day"), align='right'),

                     Column('edits', label=_("Edits/day"), align='right'),

                     ]

     maxentries = 30

     if maxentries < len(days):

         step = float(len(days))/ maxentries

     else:

         step = 1

     sv = 0.0

     se = 0.0

     sd = 0.0

     cnt = 0

     for i in xrange(len(days)-1, -1, -1):

         d, v, e = days[i], views[i], edits[i]

         # sum up views and edits to step days

         sd += 1

         cnt += 1

         sv += v

         se += e

         if cnt >= step:

             cnt -= step

             hits.addRow((d, "%.1f" % (sv/sd), "%.1f" % (se/sd)))

             sv = 0.0

             se = 0.0

             sd = 0.0

     table = DataBrowserWidget(request)

     table.setData(hits)

     return table.render(method="GET")

 def draw(pagename, request):

     import shutil, cStringIO

     from MoinMoin.stats.chart import Chart, ChartData, Color

     _ = request.getText

     # check params

     filterpage = None

     if request and request.values and 'page' in request.values:

         filterpage = request.values['page']

     days, views, edits = get_data(pagename, request, filterpage)

     import math

     try:

         scalefactor = float(max(views))/max(edits)

     except (ZeroDivisionError, ValueError):

         scalefactor = 1.0

     else:

         scalefactor = int(10 ** math.floor(math.log10(scalefactor)))

     # scale edits up

     edits = [x * scalefactor for x in edits]

     # create image

     image = cStringIO.StringIO()

     c = Chart()

     c.addData(ChartData(views, color='green'))

     c.addData(ChartData(edits, color='red'))

     chart_title = ''

     if request.cfg.sitename:

         chart_title = "%s: " % request.cfg.sitename

     chart_title = chart_title + _('Page hits and edits')

     if filterpage:

         chart_title = _("%(chart_title)s for %(filterpage)s") % {

             'chart_title': chart_title,

             'filterpage': filterpage,

         }

     chart_title = "%s\n%sx%d" % (chart_title, _("green=view\nred=edit"), scalefactor)

     c.option(

         title=chart_title.encode('iso-8859-1', 'replace'), # gdchart can't do utf-8

         xtitle=(_('date') + ' (Server)').encode('iso-8859-1', 'replace'),

         ytitle=_('# of hits').encode('iso-8859-1', 'replace'),

         title_font=c.GDC_GIANT,

         #thumblabel = 'THUMB', thumbnail = 1, thumbval = 10,

         #ytitle_color = Color('green'),

         #yaxis2 = 1,

         #ytitle2 = '# of edits',

         #ytitle2_color = Color('red'),

         #ylabel2_color = Color('black'),

         #interpolations = 0,

         threed_depth=1.0,

         requested_yinterval=1.0,

         stack_type=c.GDC_STACK_BESIDE

     )

     c.draw(c.GDC_LINE,

         (request.cfg.chart_options['width'], request.cfg.chart_options['height']),

         image, days)

     request.content_type = 'image/gif'

     request.content_length = len(image.getvalue())

     # copy the image

     image.reset()

     shutil.copyfileobj(image, request, 8192)