Merge pull request #45 from codeguru42/44-firebase-cloud-messaging

44 firebase cloud messaging

Author: codeguru42

.gitignore

@@ -5,3 +5,4 @@
 *.jpg
 
 /http-client.private.env.json
+*-firebase-adminsdk-*.json

go_capture/settings.py

@@ -129,3 +129,5 @@
 
 SGF_DIR = BASE_DIR / 'sgf'
 SGF_DIR.mkdir(exist_ok=True)
+
+FIREBASE_CREDENTIALS_FILE = BASE_DIR / env.str('FIREBASE_CREDENTIALS_FILE')

go_capture/tasks.py

@@ -1,8 +1,10 @@
 import os
 from pathlib import Path
 
+import firebase_admin
 from celery import Celery
 from django.conf import settings
+from firebase_admin import credentials, messaging
 
 from go_capture.sgf.process_image import process_image
 
@@ -20,16 +22,35 @@
 # Load task modules from all registered Django apps.
 app.autodiscover_tasks()
 
+cred = credentials.Certificate(settings.FIREBASE_CREDENTIALS_FILE)
+firebase_admin.initialize_app(cred)
+
 
 @app.task(bind=True)
 def debug_task(self):
     print(f'Request: {self.request!r}')
 
 
 @app.task
-def process_image_task(image_filename):
+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)
+
+    # See documentation on defining a message payload.
+    message = messaging.Message(
+        data={
+            'sgf_file': str(sgf_path),
+        },
+        token=fcm_token,
+    )
+
+    # Send a message to the device corresponding to the provided
+    # registration token.
+    response = messaging.send(message)
+    # Response is a message ID string.
+    print('Successfully sent message:', response)

go_capture/views.py

@@ -28,11 +28,13 @@ def capture(request):
 @require_POST
 def capture_async(request):
     image_file = request.FILES['image']
+    fcm_token = request.POST['fcm_registration_token']
+    print(f'token: {fcm_token}')
     filename = Path(image_file.name)
     output_path = settings.IMAGES_DIR / filename
     with output_path.open('wb') as output_file:
         output_file.write(image_file.read())
-    process_image_task.delay(str(output_path.absolute()))
+    process_image_task.delay(str(output_path.absolute()), fcm_token)
     return HttpResponse(status=201)
 
 

poetry.lock

@@ -11,6 +11,7 @@ Django = "^4.1.3"
 environs = "^9.5.0"
 gunicorn = "^20.1.0"
 celery = "^5.2.7"
+firebase-admin = "^6.1.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.2.0"

pyproject.toml

@@ -16,6 +16,12 @@ Content-Type: image/jpeg
 POST {{protocol}}{{host}}:{{port}}/capture_async/
 Content-Type: multipart/form-data; boundary=WebAppBoundary
 
+--WebAppBoundary
+Content-Disposition: form-data; name="fcm_registration_token"
+Content-Type: text/plain
+
+foobar
+
 --WebAppBoundary
 Content-Disposition: form-data; name="image"; filename="board.jpg"
 Content-Type: image/jpeg