CiaranG
/
Supybot-plugins
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

WebStats: Add Python 3 compatibility (not tested).

master
Valentin Lorentz 2014-04-02 17:10:27 +00:00
parent 364cf9c8de
commit be2c4bb473
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
WebStats/plugin.py
View File

@ -36,8 +36,10 @@ import time
import urllib import urllib
import random import random
import datetime import datetime
import pygraphviz if sys.version_info[0] >= 3:
from cStringIO import StringIO from io import BytesIO
else
from cStringIO import StringIO as StringIO
import supybot.conf as conf import supybot.conf as conf
import supybot.world as world import supybot.world as world
@ -443,6 +445,7 @@ class WebStatsServerCallback(httpserver.SupyHTTPServerCallback):
return template % replacement return template % replacement
def get_links(self, urlLevel, channel, page, orderby=None): def get_links(self, urlLevel, channel, page, orderby=None):
import pygraphviz
cache = world.webStatsCacheLinks cache = world.webStatsCacheLinks
channel = '#' + channel channel = '#' + channel
items = self.db.getChanLinks(channel) items = self.db.getChanLinks(channel)
@ -456,7 +459,7 @@ class WebStatsServerCallback(httpserver.SupyHTTPServerCallback):
items = [(x,y,float(z)) for x,y,z in items] items = [(x,y,float(z)) for x,y,z in items]
if not items: if not items:
graph.add_node('No links for the moment.') graph.add_node('No links for the moment.')
buffer_ = StringIO() buffer_ = BytesIO()
graph.draw(buffer_, prog='circo', format='png') graph.draw(buffer_, prog='circo', format='png')
buffer_.seek(0) buffer_.seek(0)
output = buffer_.read() output = buffer_.read()
@ -479,7 +482,7 @@ class WebStatsServerCallback(httpserver.SupyHTTPServerCallback):
color=insertedNicks[item[1]], color=insertedNicks[item[1]],
penwidth=item[2]/divideBy, penwidth=item[2]/divideBy,
arrowsize=item[2]/divideBy/2+1) arrowsize=item[2]/divideBy/2+1)
buffer_ = StringIO() buffer_ = BytesIO()
graph.draw(buffer_, prog='circo', format='png') graph.draw(buffer_, prog='circo', format='png')
buffer_.seek(0) buffer_.seek(0)
output = buffer_.read() output = buffer_.read()