def test_pageview_select(self):
dmn = 'example.com'
turl = 'http://www.test.com'
PageView.create(
domain=dmn,
url=turl,
title='index',
ip='127.0.0.1',
headers='',
params=''
)
self.assertTrue(PageView.select(PageView.title).where(PageView.title=='index').count() > 0)
self.assertFalse(PageView.select(PageView.ip).where((PageView.ip=='0.0.0.9') & (PageView.title=='index')).count() > 0)
def get_paths(query, limit):
inner = (query.select(PageView.ip,
PageView.url).order_by(PageView.timestamp))
paths = (PageView.select(PageView.ip, fn.GROUP_CONCAT(PageView.url)).from_(
inner.alias('t1')).group_by(PageView.ip).order_by(
fn.COUNT(PageView.url).desc()).tuples().limit(limit))
return [(ip, urls.split(',')) for ip, urls in paths]
def get_query(start, end):
query = PageView.select()
if start and end:
query = query.where(PageView.timestamp.between(start, end))
elif start:
query = query.where(PageView.timestamp >= start)
elif end:
query = query.where(PageView.timestamp <= end)
return query
def test_pageview_insert(self):
dmn = 'www.test.com'
turl = 'http://www.test.com'
PageView.create(
domain=dmn,
url=turl,
headers='',
params=''
)
self.assertTrue(PageView.select().count() > 0)
