Merge pull request #47 from codeguru42/46-sgf-files

46 sgf files

Author: codeguru42

go_capture/settings.py

@@ -124,10 +124,14 @@
 
 DEFAULT_AUTO_FIELD = 'django.db.models.BigAutoField'
 
-IMAGES_DIR = BASE_DIR / 'images'
+MEDIA_URL = 'media/'
+MEDIA_ROOT = BASE_DIR / 'media'
+MEDIA_ROOT.mkdir(exist_ok=True)
+
+IMAGES_DIR = MEDIA_ROOT / 'images'
 IMAGES_DIR.mkdir(exist_ok=True)
 
-SGF_DIR = BASE_DIR / 'sgf'
+SGF_DIR = MEDIA_ROOT / 'sgf'
 SGF_DIR.mkdir(exist_ok=True)
 
 FIREBASE_CREDENTIALS_FILE = BASE_DIR / env.str('FIREBASE_CREDENTIALS_FILE')

go_capture/tasks.py

@@ -1,5 +1,5 @@
+import io
 import os
-from pathlib import Path
 
 import firebase_admin
 from celery import Celery
@@ -34,17 +34,17 @@ def debug_task(self):
 @app.task
 def process_image_task(image_filename, fcm_token):
     print(f'token: {fcm_token}')
-    sgf_filename = f'{Path(image_filename).stem}.sgf'
-    sgf_path = settings.SGF_DIR / sgf_filename
     print(f'Processing {image_filename}')
     with open(image_filename, 'rb') as image_file:
-        with open(sgf_path, 'w') as sgf_file:
-            process_image(image_file, sgf_file)
+        sgf_buffer = io.StringIO()
+        process_image(image_file, sgf_buffer)
 
-    # See documentation on defining a message payload.
+    sgf_buffer.seek(0)
+    sgf_data = sgf_buffer.read()
+    print(sgf_data)
     message = messaging.Message(
         data={
-            'sgf_file': str(sgf_path),
+            'sgf': sgf_data,
         },
         token=fcm_token,
     )

go_capture/views.py

@@ -31,6 +31,7 @@ def capture_async(request):
     fcm_token = request.POST['fcm_registration_token']
     print(f'token: {fcm_token}')
     filename = Path(image_file.name)
+    print(filename)
     output_path = settings.IMAGES_DIR / filename
     with output_path.open('wb') as output_file:
         output_file.write(image_file.read())