mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-12 14:13:38 +00:00
212 lines
7.5 KiB
Python
212 lines
7.5 KiB
Python
# Author: Dennis Lutter <lad1337@gmail.com>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of Sick Beard.
|
|
#
|
|
# Sick Beard is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Sick Beard is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Sick Beard. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from sickbeard import db, logger
|
|
|
|
class BlackAndWhiteList(object):
|
|
_tableBlack = "blacklist"
|
|
_tableWhite = "whitelist"
|
|
blackList = []
|
|
whiteList = []
|
|
blackDict = {}
|
|
whiteDict = {}
|
|
|
|
last_black_valid_result = None
|
|
last_white_valid_result = None
|
|
|
|
def __init__(self, show_id):
|
|
if not show_id:
|
|
raise BlackWhitelistNoShowIDException()
|
|
self.show_id = show_id
|
|
self.refresh()
|
|
|
|
def refresh(self):
|
|
logger.log(u"Building black and white list for " + str(self.show_id), logger.DEBUG)
|
|
|
|
(self.blackList, self.blackDict) = self.load_blacklist()
|
|
(self.whiteList, self.whiteDict) = self.load_whitelist()
|
|
|
|
def load_blacklist(self):
|
|
return self._load_list(self._tableBlack)
|
|
|
|
def load_whitelist(self):
|
|
return self._load_list(self._tableWhite)
|
|
|
|
def get_black_keywords_for(self, range):
|
|
if range in self.blackDict:
|
|
return self.blackDict[range]
|
|
else:
|
|
return []
|
|
|
|
def get_white_keywords_for(self, range):
|
|
if range in self.whiteDict:
|
|
return self.whiteDict[range]
|
|
else:
|
|
return []
|
|
|
|
def set_black_keywords(self, range, values):
|
|
self._del_all_black_keywords()
|
|
self._add_keywords(self._tableBlack, range, values)
|
|
|
|
def set_white_keywords(self, range, values):
|
|
self._del_all_white_keywords()
|
|
self._add_keywords(self._tableWhite, range, values)
|
|
|
|
def set_black_keywords_for(self, range, values):
|
|
self._del_all_black_keywords_for(range)
|
|
self._add_keywords(self._tableBlack, range, values)
|
|
|
|
def set_white_keywords_for(self, range, values):
|
|
self._del_all_white_keywords_for(range)
|
|
self._add_keywords(self._tableWhite, range, values)
|
|
|
|
def add_black_keyword(self, range, value):
|
|
self._add_keywords(self._tableBlack, range, [value])
|
|
|
|
def add_white_keyword(self, range, value):
|
|
self._add_keywords(self._tableWhite, range, [value])
|
|
|
|
def get_last_result_msg(self):
|
|
blackResult = whiteResult = "Untested"
|
|
if self.last_black_valid_result == True:
|
|
blackResult = "Valid"
|
|
elif self.last_black_valid_result == False:
|
|
blackResult = "Invalid"
|
|
|
|
if self.last_white_valid_result == True:
|
|
whiteResult = "Valid"
|
|
elif self.last_white_valid_result == False:
|
|
whiteResult = "Invalid"
|
|
|
|
return "Blacklist: " + blackResult + ", Whitelist: " + whiteResult
|
|
|
|
def _add_keywords(self, table, range, values):
|
|
with db.DBConnection() as myDB:
|
|
for value in values:
|
|
myDB.action("INSERT INTO " + table + " (show_id, range , keyword) VALUES (?,?,?)", [self.show_id, range, value])
|
|
self.refresh()
|
|
|
|
def _del_all_black_keywords(self):
|
|
self._del_all_keywords(self._tableBlack)
|
|
|
|
def _del_all_white_keywords(self):
|
|
self._del_all_keywords(self._tableWhite)
|
|
|
|
def _del_all_black_keywords_for(self, range):
|
|
self._del_all_keywords_for(self._tableBlack, range)
|
|
|
|
def _del_all_white_keywords_for(self, range):
|
|
self._del_all_keywords_for(self._tableWhite, range)
|
|
|
|
def _del_all_keywords(self, table):
|
|
logger.log(u"Deleting all " + table + " keywords for " + str(self.show_id), logger.DEBUG)
|
|
with db.DBConnection() as myDB:
|
|
myDB.action("DELETE FROM " + table + " WHERE show_id = ?", [self.show_id])
|
|
self.refresh()
|
|
|
|
def _del_all_keywords_for(self, table, range):
|
|
logger.log(u"Deleting all " + range + " " + table + " keywords for " + str(self.show_id), logger.DEBUG)
|
|
with db.DBConnection() as myDB:
|
|
myDB.action("DELETE FROM " + table + " WHERE show_id = ? and range = ?", [self.show_id, range])
|
|
self.refresh()
|
|
|
|
def _load_list(self, table):
|
|
with db.DBConnection() as myDB:
|
|
sqlResults = myDB.select("SELECT range,keyword FROM " + table + " WHERE show_id = ? ", [self.show_id])
|
|
if not sqlResults or not len(sqlResults):
|
|
return ([], {})
|
|
|
|
list, dict = self._build_keyword_dict(sqlResults)
|
|
logger.log("BWL: " + str(self.show_id) + " loaded keywords from " + table + ": " + str(dict), logger.DEBUG)
|
|
return list, dict
|
|
|
|
def _build_keyword_dict(self, sql_result):
|
|
list = []
|
|
dict = {}
|
|
for row in sql_result:
|
|
list.append(row["keyword"])
|
|
if row["range"] in dict:
|
|
dict[row["range"]].append(row["keyword"])
|
|
else:
|
|
dict[row["range"]] = [row["keyword"]]
|
|
|
|
return (list, dict)
|
|
|
|
def is_valid_for_black(self, haystack):
|
|
logger.log(u"BWL: " + str(self.show_id) + " is valid black", logger.DEBUG)
|
|
result = self._is_valid_for(self.blackDict, False, haystack)
|
|
self.last_black_valid_result = result
|
|
return result
|
|
|
|
def is_valid_for_white(self, haystack):
|
|
logger.log(u"BWL: " + str(self.show_id) + " is valid white", logger.DEBUG)
|
|
result = self._is_valid_for(self.whiteDict, True, haystack)
|
|
self.last_white_valid_result = result
|
|
return result
|
|
|
|
def is_valid(self, haystack):
|
|
return self.is_valid_for_black(haystack) and self.is_valid_for_white(haystack)
|
|
|
|
def _is_valid_for(self, list, mood, haystack):
|
|
if not len(list):
|
|
return True
|
|
|
|
results = []
|
|
for range in list:
|
|
for keyword in list[range]:
|
|
string = None
|
|
if range == "global":
|
|
string = haystack.name
|
|
elif range in haystack.__dict__:
|
|
string = haystack.__dict__[range]
|
|
elif not range in haystack.__dict__:
|
|
results.append((not mood))
|
|
else:
|
|
results.append(False)
|
|
|
|
if string:
|
|
results.append(self._is_keyword_in_string(string, keyword) == mood)
|
|
|
|
# black: mood = False
|
|
# white: mood = True
|
|
if mood in results:
|
|
return mood
|
|
else:
|
|
return (not mood)
|
|
|
|
def _is_keyword_in_string(self, fromPost, fromBWList):
|
|
"""
|
|
will return true if fromBWList is found in fromPost
|
|
for now a basic find is used
|
|
"""
|
|
fromPost = fromPost.lower()
|
|
fromBWList = fromBWList.lower()
|
|
logger.log(u"BWL: " + str(self.show_id) + " comparing fromPost: " + fromPost + " vs fromBWlist: " + fromBWList, logger.DEBUG)
|
|
return (fromPost.find(fromBWList) >= 0)
|
|
|
|
class BlackWhiteKeyword(object):
|
|
range = ""
|
|
value = []
|
|
|
|
def __init__(self, range, values):
|
|
self.range = range # "global" or a parser group
|
|
self.value = values # a list of values may contain only one item (still a list)
|
|
|
|
|
|
class BlackWhitelistNoShowIDException(Exception):
|
|
"No show_id was given"
|