diff options
Diffstat (limited to 'check/commands/imagehash/query.py')
| -rw-r--r-- | check/commands/imagehash/query.py | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/check/commands/imagehash/query.py b/check/commands/imagehash/query.py index 4c51324..f5d3a54 100644 --- a/check/commands/imagehash/query.py +++ b/check/commands/imagehash/query.py @@ -3,8 +3,11 @@ Search the database for an image """ import click +import os -from app.models.sql_factory import search_by_phash, add_phash +from app.models.sql_factory import search_by_phash +from app.utils.im_utils import compute_phash_int +from app.utils.file_utils import sha256 @click.command() @click.option('-i', '--input', 'opt_fn', @@ -16,3 +19,16 @@ def cli(ctx, opt_fn): Search the database for an image """ print('Searching for a file...') + + if not os.path.exists(opt_fn): + print("File does not exist") + return + + hash = sha256(opt_fn) + phash = compute_phash_int(opt_fn) + + res = search_by_hash(hash) + print("search by hash:", res) + + res =search_by_phash(phash) + print("search by phash:", res) |