def test_get_limit(self):
"""
Tests the static get limit method which should return the limit on the amount of elements that should be shown/returned
"""
test_cases = {
None: 0,
'': 0,
'0': 0,
'5': 5,
'-5': 0,
'1.5': 0,
'-1.5': 0,
5: 5,
-5: 0,
1.5: 1,
-1.5: 0,
}
unicode_test_cases = {
'': 0,
'0': 0,
'5': 5,
'-5': 0,
'1.5': 0,
'-1.5': 0,
}
for tests in test_cases, unicode_test_cases:
for (action, result) in six.iteritems(tests):
self.assertEqual(History._get_limit(action), result) # pylint: disable=protected-access
def test_get_actions(self):
"""
Tests whether or not the different kinds of actions an episode can have are returned correctly
"""
test_cases = {
None: [],
'': [],
'wrong': [],
'downloaded': Quality.DOWNLOADED,
'Downloaded': Quality.DOWNLOADED,
'snatched': Quality.SNATCHED,
'Snatched': Quality.SNATCHED,
}
unicode_test_cases = {
'': [],
'wrong': [],
'downloaded': Quality.DOWNLOADED,
'Downloaded': Quality.DOWNLOADED,
'snatched': Quality.SNATCHED,
'Snatched': Quality.SNATCHED,
}
for tests in test_cases, unicode_test_cases:
for (action, result) in six.iteritems(tests):
self.assertEqual(History._get_actions(action), result) # pylint: disable=protected-access
def test_get_limit(self):
"""
Tests the static get limit method which should return the limit on the amount of elements that should be shown/returned
"""
test_cases = {
None: 0,
"": 0,
"0": 0,
"5": 5,
"-5": 0,
"1.5": 0,
"-1.5": 0,
5: 5,
-5: 0,
1.5: 1,
-1.5: 0,
}
unicode_test_cases = {
"": 0,
"0": 0,
"5": 5,
"-5": 0,
"1.5": 0,
"-1.5": 0,
}
for tests in test_cases, unicode_test_cases:
for (action, result) in tests.items():
assert History._get_limit(action) == result
def test_get_actions(self):
"""
Tests whether or not the different kinds of actions an episode can have are returned correctly
"""
test_cases = {
None: [],
"": [],
"wrong": [],
"downloaded": Quality.DOWNLOADED,
"Downloaded": Quality.DOWNLOADED,
"snatched": Quality.SNATCHED,
"Snatched": Quality.SNATCHED,
}
unicode_test_cases = {
"": [],
"wrong": [],
"downloaded": Quality.DOWNLOADED,
"Downloaded": Quality.DOWNLOADED,
"snatched": Quality.SNATCHED,
"Snatched": Quality.SNATCHED,
}
for tests in test_cases, unicode_test_cases:
for (action, result) in tests.items():
assert History._get_actions(action) == result
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.history = HistoryTool()
class History(WebRoot):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.history = HistoryTool()
def index(self, limit=None):
settings.HISTORY_LIMIT = limit = try_int(
limit or settings.HISTORY_LIMIT or 100, 100)
sickchill.start.save_config()
compact = []
data = self.history.get(limit)
for row in data:
action = {
'action': row['action'],
'provider': row['provider'],
'resource': row['resource'],
'time': row['date']
}
if not any(
(history['show_id'] == row['show_id'] and history['season'] ==
row['season'] and history['episode'] == row['episode']
and history['quality'] == row['quality'])
for history in compact):
history = {
'actions': [action],
'episode': row['episode'],
'quality': row['quality'],
'resource': row['resource'],
'season': row['season'],
'show_id': row['show_id'],
'show_name': row['show_name']
}
compact.append(history)
else:
index = [
i for i, item in enumerate(compact)
if item['show_id'] == row['show_id'] and item['season'] ==
row['season'] and item['episode'] == row['episode']
and item['quality'] == row['quality']
][0]
history = compact[index]
history['actions'].append(action)
history['actions'].sort(key=lambda x: x['time'], reverse=True)
t = PageTemplate(rh=self, filename="history.mako")
submenu = [
{
'title': _('Remove Selected'),
'path': 'history/removeHistory',
'icon': 'fa fa-eraser',
'class': 'removehistory',
'confirm': False
},
{
'title': _('Clear History'),
'path': 'history/clearHistory',
'icon': 'fa fa-trash',
'class': 'clearhistory',
'confirm': True
},
{
'title': _('Trim History'),
'path': 'history/trimHistory',
'icon': 'fa fa-scissors',
'class': 'trimhistory',
'confirm': True
},
]
return t.render(historyResults=data,
compactResults=compact,
limit=limit,
submenu=submenu,
title=_('History'),
header=_('History'),
topmenu="history",
controller="history",
action="index")
def removeHistory(self, toRemove=None):
logsToRemove = []
for logItem in toRemove.split('|'):
info = logItem.split(',')
logsToRemove.append({
'dates': info[0].split('$'),
'show_id': info[1],
'season': info[2],
'episode': info[3]
})
self.history.remove(logsToRemove)
ui.notifications.message(_('Selected history entries removed'))
return self.redirect("/history/")
def clearHistory(self):
self.history.clear()
ui.notifications.message(_('History cleared'))
return self.redirect("/history/")
def trimHistory(self):
self.history.trim()
ui.notifications.message(
_('Removed history entries older than 30 days'))
return self.redirect("/history/")
class History(WebRoot):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.history = HistoryTool()
def index(self, limit=None):
settings.HISTORY_LIMIT = limit = try_int(
limit or settings.HISTORY_LIMIT or 100, 100)
sickchill.start.save_config()
compact = []
data = self.history.get(limit)
for row in data:
action = {
"action": row["action"],
"provider": row["provider"],
"resource": row["resource"],
"time": row["date"]
}
if not any(
(history["show_id"] == row["show_id"] and history["season"] ==
row["season"] and history["episode"] == row["episode"]
and history["quality"] == row["quality"])
for history in compact):
history = {
"actions": [action],
"episode": row["episode"],
"quality": row["quality"],
"resource": row["resource"],
"season": row["season"],
"show_id": row["show_id"],
"show_name": row["show_name"],
}
compact.append(history)
else:
index = [
i for i, item in enumerate(compact)
if item["show_id"] == row["show_id"] and item["season"] ==
row["season"] and item["episode"] == row["episode"]
and item["quality"] == row["quality"]
][0]
history = compact[index]
history["actions"].append(action)
history["actions"].sort(key=lambda x: x["time"], reverse=True)
t = PageTemplate(rh=self, filename="history.mako")
submenu = [
{
"title": _("Remove Selected"),
"path": "history/removeHistory",
"icon": "fa fa-eraser",
"class": "removehistory",
"confirm": False
},
{
"title": _("Clear History"),
"path": "history/clearHistory",
"icon": "fa fa-trash",
"class": "clearhistory",
"confirm": True
},
{
"title": _("Trim History"),
"path": "history/trimHistory",
"icon": "fa fa-scissors",
"class": "trimhistory",
"confirm": True
},
]
return t.render(
historyResults=data,
compactResults=compact,
limit=limit,
submenu=submenu,
title=_("History"),
header=_("History"),
topmenu="history",
controller="history",
action="index",
)
def removeHistory(self, toRemove=None):
logsToRemove = []
for logItem in toRemove.split("|"):
info = logItem.split(",")
logsToRemove.append({
"dates": info[0].split("$"),
"show_id": info[1],
"season": info[2],
"episode": info[3]
})
self.history.remove(logsToRemove)
ui.notifications.message(_("Selected history entries removed"))
return self.redirect("/history/")
def clearHistory(self):
self.history.clear()
ui.notifications.message(_("History cleared"))
return self.redirect("/history/")
def trimHistory(self):
self.history.trim()
ui.notifications.message(
_("Removed history entries older than 30 days"))
return self.redirect("/history/")