20
20
class CeleryServer (BaseServer , Celery ):
21
21
"""A server for a Celery app."""
22
22
23
- def __init__ (self , auto_run : bool = True ):
23
+ def __init__ (self , auto_run : bool = True , dump_request : bool = False ):
24
+ # pylint: disable=line-too-long
24
25
"""Initialize a Celery app.
25
26
26
27
Examples:
@@ -35,10 +36,12 @@ def __init__(self, auto_run: bool = True):
35
36
36
37
Args:
37
38
auto_run: A flag designating whether to auto-run the server.
39
+ dump_request: A flag designating whether to add the dump_request task (useful for debugging).
38
40
39
41
Raises:
40
42
EnvironmentError: If "DJANGO_SETTINGS_MODULE" is not in os.environ.
41
43
"""
44
+ # pylint: enable=line-too-long
42
45
43
46
call_django_command ("check" )
44
47
@@ -58,14 +61,14 @@ def __init__(self, auto_run: bool = True):
58
61
# Load task modules from all registered Django apps.
59
62
self .autodiscover_tasks ()
60
63
61
- if self . django_dev_server_is_running () :
64
+ if dump_request :
62
65
63
66
@self .task (
64
67
name = f"{ settings .SERVICE_NAME } .dump_request" ,
65
68
bind = True ,
66
69
ignore_result = True ,
67
70
)
68
- def dump_request (self , * args , ** kwargs ):
71
+ def _dump_request (self , * args , ** kwargs ):
69
72
"""Dumps its own request information."""
70
73
71
74
logging .info ("Request: %s" , self .request )
0 commit comments