Optimize top_downloads post analysis plugin

This commit is contained in:
Gregory Soutade 2015-01-13 18:54:22 +01:00
parent 3fb1b897ea
commit 9d3f7c05a4

View File

@ -53,19 +53,14 @@ class IWLAPostAnalysisTopDownloads(IPlugin):
def __init__(self, iwla): def __init__(self, iwla):
super(IWLAPostAnalysisTopDownloads, self).__init__(iwla) super(IWLAPostAnalysisTopDownloads, self).__init__(iwla)
self.API_VERSION = 1 self.API_VERSION = 1
self.conf_requires = ['multimedia_files', 'viewed_http_codes']
def hook(self): def hook(self):
stats = self.iwla.getCurrentVisists() stats = self.iwla.getValidVisitors()
month_stats = self.iwla.getMonthStats() month_stats = self.iwla.getMonthStats()
multimedia_files = self.iwla.getConfValue('multimedia_files')
viewed_http_codes = self.iwla.getConfValue('viewed_http_codes')
top_downloads = month_stats.get('top_downloads', {}) top_downloads = month_stats.get('top_downloads', {})
for (k, super_hit) in stats.items(): for (k, super_hit) in stats.items():
if super_hit['robot']: continue
for r in super_hit['requests'][::-1]: for r in super_hit['requests'][::-1]:
if not self.iwla.isValidForCurrentAnalysis(r): if not self.iwla.isValidForCurrentAnalysis(r):
break break
@ -75,13 +70,8 @@ class IWLAPostAnalysisTopDownloads(IPlugin):
uri = r['extract_request']['extract_uri'].lower() uri = r['extract_request']['extract_uri'].lower()
isMultimedia = False if self.iwla.isMultimediaFile(uri):
for ext in multimedia_files: continue
if uri.endswith(ext):
isMultimedia = True
break
if isMultimedia: continue
uri = "%s%s" % (r.get('server_name', ''), uri = "%s%s" % (r.get('server_name', ''),
r['extract_request']['extract_uri']) r['extract_request']['extract_uri'])