Merge pull request #2106 from ppfeister/bugfix/skip-filter-2103
Skip content filter for explicitly chosen targets
This commit is contained in:
@@ -575,7 +575,7 @@ def main():
|
||||
action="append",
|
||||
metavar="SITE_NAME",
|
||||
dest="site_list",
|
||||
default=None,
|
||||
default=[],
|
||||
help="Limit analysis to just the listed sites. Add multiple options to specify more than one site.",
|
||||
)
|
||||
parser.add_argument(
|
||||
@@ -725,13 +725,13 @@ def main():
|
||||
sys.exit(1)
|
||||
|
||||
if not args.nsfw:
|
||||
sites.remove_nsfw_sites()
|
||||
sites.remove_nsfw_sites(do_not_remove=args.site_list)
|
||||
|
||||
# Create original dictionary from SitesInformation() object.
|
||||
# Eventually, the rest of the code will be updated to use the new object
|
||||
# directly, but this will glue the two pieces together.
|
||||
site_data_all = {site.name: site.information for site in sites}
|
||||
if args.site_list is None:
|
||||
if args.site_list == []:
|
||||
# Not desired to look at a sub-set of sites
|
||||
site_data = site_data_all
|
||||
else:
|
||||
|
||||
+3
-2
@@ -179,7 +179,7 @@ class SitesInformation:
|
||||
|
||||
return
|
||||
|
||||
def remove_nsfw_sites(self):
|
||||
def remove_nsfw_sites(self, do_not_remove: list = []):
|
||||
"""
|
||||
Remove NSFW sites from the sites, if isNSFW flag is true for site
|
||||
|
||||
@@ -190,8 +190,9 @@ class SitesInformation:
|
||||
None
|
||||
"""
|
||||
sites = {}
|
||||
do_not_remove = [site.casefold() for site in do_not_remove]
|
||||
for site in self.sites:
|
||||
if self.sites[site].is_nsfw:
|
||||
if self.sites[site].is_nsfw and site.casefold() not in do_not_remove:
|
||||
continue
|
||||
sites[site] = self.sites[site]
|
||||
self.sites = sites
|
||||
|
||||
Reference in New Issue
Block a user