Add NSFW filter to address query commands

This commit is contained in:
Martin Herkt 2017-10-27 08:36:36 +02:00
parent 7bbeb2d144
commit ebfbc6b153
No known key found for this signature in database
GPG Key ID: C24B9CD04DC6AE7F
1 changed files with 7 additions and 4 deletions

View File

@ -459,9 +459,12 @@ def queryhash(h):
query(su.enbase(f.id, 1))
@manager.command
def queryaddr(a):
def queryaddr(a, nsfw=False):
res = File.query.filter_by(addr=a)
if nsfw:
res = res.filter(File.nsfw_score > app.config["NSFW_THRESHOLD"])
for f in res:
query(su.enbase(f.id, 1))
@ -497,15 +500,15 @@ def update_nsfw():
@manager.command
def querybl():
def querybl(nsfw=False):
if os.path.isfile(app.config["FHOST_UPLOAD_BLACKLIST"]):
with open(app.config["FHOST_UPLOAD_BLACKLIST"], "r") as bl:
for l in bl.readlines():
if not l.startswith("#"):
if not ":" in l:
queryaddr("::ffff:" + l.rstrip())
queryaddr("::ffff:" + l.rstrip(), nsfw)
else:
queryaddr(l.strip())
queryaddr(l.strip(), nsfw)
if __name__ == "__main__":
manager.run()