fix: celery, asgi 각각 초기화

QRTaxi · Aug 17, 2023 · 9161f3f · 9161f3f
1 parent b84c7f3
commit 9161f3f
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/call/models/assign_signals.py b/call/models/assign_signals.py
@@ -1,4 +1,5 @@
 from firebase_admin import messaging
+import firebase_admin
 
 def websocket_message(instance, created: bool):
  from . import Assign
@@ -55,5 +56,10 @@ def send_push_notification(instance):
  },
  token=registration_token,
  )
- response = messaging.send(message)
+ if instance.status == "failed":
+ app = firebase_admin.get_app('celery_app')
+ else:
+ app = firebase_admin.get_app('asgi_app')
+
+ response = messaging.send(message, app=app)
  print('Successfully sent message:', response)
diff --git a/hackathon/asgi.py b/hackathon/asgi.py
@@ -18,8 +18,7 @@
 
 cred_path = config("FIRE_BASE_JSON_KEY_PATH")
 cred = credentials.Certificate(cred_path)
-if not firebase_admin._apps:
- firebase_admin.initialize_app(cred)
+firebase_admin.initialize_app(cred, name='asgi_app')
 
 django_asgi_app = get_asgi_application()
 

diff --git a/hackathon/celery.py b/hackathon/celery.py
@@ -13,8 +13,7 @@
 
 cred_path = config("FIRE_BASE_JSON_KEY_PATH")
 cred = credentials.Certificate(cred_path)
-if not firebase_admin._apps:
- firebase_admin.initialize_app(cred)
+firebase_admin.initialize_app(cred, name='celery_app')
 
 #테스트할 때만 주석해제하고 실제로 배포할 때는 주석처리하기
 #@app.task(bind=True, ignore_result=True)