diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-04-15 16:26:03 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-04-15 16:26:03 +0200 |
| commit | 79f0e696f3f6067a0841a37404fb546dedaa07cb (patch) | |
| tree | a064f2841dc532f81fcf04eb84300e679fda2b27 /check/commands/imagehash/query.py | |
| parent | e257e83f313a2976347b0a30f58e66b7bcbc1235 (diff) | |
cli suite working
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) |