diff options
Diffstat (limited to 'check/commands/imagehash/query.py')
| -rw-r--r-- | check/commands/imagehash/query.py | 34 |
1 files changed, 0 insertions, 34 deletions
diff --git a/check/commands/imagehash/query.py b/check/commands/imagehash/query.py deleted file mode 100644 index f5d3a54..0000000 --- a/check/commands/imagehash/query.py +++ /dev/null @@ -1,34 +0,0 @@ -""" -Search the database for an image -""" - -import click -import os - -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', - required=True, - help="File to search") -@click.pass_context -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) |
