summaryrefslogtreecommitdiff
path: root/check/commands/imagehash
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2019-04-15 16:26:03 +0200
committerJules Laplace <julescarbon@gmail.com>2019-04-15 16:26:03 +0200
commit79f0e696f3f6067a0841a37404fb546dedaa07cb (patch)
treea064f2841dc532f81fcf04eb84300e679fda2b27 /check/commands/imagehash
parente257e83f313a2976347b0a30f58e66b7bcbc1235 (diff)
cli suite working
Diffstat (limited to 'check/commands/imagehash')
-rw-r--r--check/commands/imagehash/add.py34
-rw-r--r--check/commands/imagehash/load.py22
-rw-r--r--check/commands/imagehash/query.py34
-rw-r--r--check/commands/imagehash/test.py18
4 files changed, 0 insertions, 108 deletions
diff --git a/check/commands/imagehash/add.py b/check/commands/imagehash/add.py
deleted file mode 100644
index 73f8b69..0000000
--- a/check/commands/imagehash/add.py
+++ /dev/null
@@ -1,34 +0,0 @@
-"""
-Add a file to the database
-"""
-
-import click
-import os
-
-from app.models.sql_factory import search_by_phash, add_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 add (gif/jpg/png)")
-@click.option('-u', '--upload', 'opt_upload', is_flag=True,
- help='Whether to upload this file to S3')
-@click.pass_context
-def cli(ctx, opt_fn, opt_upload):
- """
- Add a single file
- """
- print('Adding 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)
-
- dir, fn = os.path.split(opt_fn)
- root, ext = os.path.splitext(fn)
-
- add_phash(sha256=hash, phash=phash, ext=ext)
diff --git a/check/commands/imagehash/load.py b/check/commands/imagehash/load.py
deleted file mode 100644
index e61b751..0000000
--- a/check/commands/imagehash/load.py
+++ /dev/null
@@ -1,22 +0,0 @@
-"""
-Loop over a directory of images
- - Compute their phashes
- - Optionally upload them to s3?
-"""
-
-import click
-
-from app.models.sql_factory import search_by_phash, add_phash
-
-@click.command()
-@click.option('-i', '--input', 'opt_dir_fn',
- required=True,
- help="File to add (gif/jpg/png)")
-@click.option('-u', '--upload', 'opt_upload', is_flag=True,
- help='Whether to upload this file to S3')
-@click.pass_context
-def cli(ctx, opt_dir_fn, opt_store):
- """
- Add a directory of images
- """
- print('Adding a directory...')
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)
diff --git a/check/commands/imagehash/test.py b/check/commands/imagehash/test.py
deleted file mode 100644
index b3ddbe5..0000000
--- a/check/commands/imagehash/test.py
+++ /dev/null
@@ -1,18 +0,0 @@
-"""
-Query the database with a test set
-"""
-
-import click
-
-from app.models.sql_factory import search_by_phash, add_phash
-
-@click.command()
-@click.option('-i', '--input', 'opt_fn',
- required=True,
- help="Directory to search")
-@click.pass_context
-def cli(ctx, opt_fn):
- """
- Search the database for an image
- """
- print('Query the database with a test set')