fix env for celery

Author: darjeeling

pythonkr_backend/pythonkr_backend/celery.py

@@ -22,12 +22,12 @@
 
 @worker_init.connect()  
 def init_worker(*args, **kwargs):
-    logfire.configure(service_name="celery-worker", send_to_logfire='if-token-present')  
+    logfire.configure(environment='prod',service_name="celery-worker", send_to_logfire='if-token-present')  
     logfire.instrument_celery()
     logfire.instrument_pydantic_ai()
 
 @beat_init.connect()  
 def init_beat(*args, **kwargs):
-    logfire.configure(service_name="celery-beat", send_to_logfire='if-token-present')  
+    logfire.configure(environment='prod',service_name="celery-beat", send_to_logfire='if-token-present')  
     logfire.instrument_celery()
     logfire.instrument_pydantic_ai()

pythonkr_backend/pythonkr_backend/settings/prod.py

@@ -61,18 +61,11 @@
 # check WSGI environment
 IS_PRODUCTION_SERVER = os.environ.get('IS_WSGI_ENVIRONMENT', 'False') == 'True'
 
-IN_CELERY_WORKER_PROCESS = sys.argv and 'worker' in sys.argv
-
-if IN_CELERY_WORKER_PROCESS:
-    print('Im in Celery worker')
-
 # logfire settings
 if IS_PRODUCTION_SERVER:
     logfire.configure(environment='prod', service_name="web", service_version=sha_service_version)
     logfire.instrument_django()
     logfire.instrument_system_metrics()
-    #logfire.instrument_psycopg('psycopg')
-print(f"sys.argv: {sys.argv}")
 
 # celery 
 CELERY_BROKER_PASSWORD = os.environ.get("CELERY_BROKER_PASSWORD","FALSE")