summaryrefslogtreecommitdiff
path: root/check/commands/phash/add.py
diff options
context:
space:
mode:
Diffstat (limited to 'check/commands/phash/add.py')
-rw-r--r--check/commands/phash/add.py21
1 files changed, 21 insertions, 0 deletions
diff --git a/check/commands/phash/add.py b/check/commands/phash/add.py
new file mode 100644
index 0000000..1565202
--- /dev/null
+++ b/check/commands/phash/add.py
@@ -0,0 +1,21 @@
+"""
+Add a file to the database
+"""
+
+import click
+
+from app.models.sql_factory import add_phash_by_filename
+
+@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...')
+ add_phash_by_filename(opt_fn)