Skip to content

Commit

Permalink
Merge pull request #2106 from ppfeister/bugfix/skip-filter-2103
Browse files Browse the repository at this point in the history
Skip content filter for explicitly chosen targets
  • Loading branch information
sdushantha committed May 9, 2024
2 parents f6bb608 + d85c073 commit 5abd758
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
6 changes: 3 additions & 3 deletions sherlock/sherlock.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down Expand Up @@ -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:
Expand Down
5 changes: 3 additions & 2 deletions sherlock/sites.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ def __init__(self, data_file_path=None):

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
Expand All @@ -190,8 +190,9 @@ def remove_nsfw_sites(self):
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
Expand Down

0 comments on commit 5abd758

Please sign in to comment.