api now fetches url

1 files changed, 6 insertions, 5 deletions

diff --git a/check/app/server/api.py b/check/app/server/api.py
index 3742b15..63de5b6 100644
--- a/check/app/server/api.py
+++ b/check/app/server/api.py
@@ -34,9 +34,9 @@ def match():
   start = time.time()
 
   try:
-    threshold = int(request.args.get('threshold') or 6)
-    limit = int(request.args.get('limit') or 1)
-    add = str(request.args.get('add') or 'true') == 'true'
+    threshold = int(request.form.get('threshold') or 6)
+    limit = int(request.form.get('limit') or 1)
+    add = str(request.form.get('add') or 'true') == 'true'
   except:
     return jsonify({
       'success': False,
@@ -61,7 +61,7 @@ def match():
 
     im = Image.open(file.stream).convert('RGB')
   else:
-    url = request.args.get('url')
+    url = request.form.get('url')
     if not url:
       return jsonify({
         'success': False,
@@ -88,7 +88,8 @@ def match():
 
   if len(results) == 0:
     if add and url:
-      hash = sha256_stream(file)
+      # hash = sha256_stream(file)
+      hash = sha256_stream(io.BytesIO(raw))
       add_phash(sha256=hash, phash=phash, ext=ext, url=url)
     match = False
   else: