send push notification

Author: feijooso

app/main.py

@@ -6,10 +6,13 @@
 from service.calculator_service import CalculatorService
 from router import api_router
 
+
 app = FastAPI()
 service = CalculatorService()
 controller = CalculatorController(service)
 
+
+
 logger = logging.getLogger("measurements")
 logger.setLevel("DEBUG")
 

app/main_rabbitmq.py

@@ -1,9 +1,12 @@
 import os
 import logging
 from service.rabbitmq.consumer import Consumer
-
+import firebase_admin
+from firebase_admin import credentials
 
 def main():
+    cred = credentials.Certificate("firebase_credentials.json")
+    firebase_admin.initialize_app(cred)
     logger = logging.getLogger("rabbitmq_consumer")
     # DEBUG, INFO, WARNING, ERROR, CRITICAL
     logging_level = os.environ.get("LOGGING_LEVEL")

app/schemas/measurement.py

@@ -23,3 +23,4 @@ class MeasurementReadingSchema(BaseModel):
     watering: Optional[int]
     id_device: str
     time_stamp: str
+    device_token: Optional[str]

app/service/rabbitmq/consumer.py

@@ -17,6 +17,7 @@
 from database.database import SQLAlchemyClient
 from resources.parser import apply_rules
 from os import environ
+from firebase_admin import messaging
 
 
 Base = declarative_base(
@@ -50,6 +51,7 @@ def obtain_device_plant(self, measurement_from_rabbit):
             logger.error(f"{err} - {type(err)}")
             raise RowNotFoundError(measurement_from_rabbit.id_device, "DEVICE_PLANT")
 
+
     def check_package(self, measurement):
         empty_values = []
         if measurement.temperature is None:
@@ -69,11 +71,16 @@ def check_package(self, measurement):
 
     def send_notification(self, id_user, measurement, error, details):
         logger.info(LoggerMessages.USER_NOTIFIED.format(id_user))
-        logger.info(
-            "New notification: {}: {} sent on {}".format(
-                error, details, measurement.time_stamp
-            )
-        )
+
+        print(f"details: {details}")
+        print(f"error: {error}")
+
+        if measurement.device_token is not None:
+                    message = messaging.Message(
+                        notification=messaging.Notification(title="Estado de tu planta", body="details"),
+                        token=measurement.device_token,
+                    )
+                    messaging.send(message)
 
     def apply_rules(self, measurement,  device_plant):
         deviated_parameters = apply_rules(measurement, device_plant.plant_type)

requirements.txt

@@ -5,6 +5,7 @@ flake8
 psycopg2
 pandas
 httpx
+firebase-admin
 
 ### RabbitMQ ###
 pika