query api

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)