Differential D8684 Diff 40813 examples/docs_snippets/docs_snippets/concepts/logging/custom_logger.py

Changeset View

Standalone View

examples/docs_snippets/docs_snippets/concepts/logging/custom_logger.py

Show All 32 Lines	def json_console_logger(init_context):
return logger_		return logger_


@solid		@solid
def hello_logs(context):		def hello_logs(context):
context.log.info("Hello, world!")		context.log.info("Hello, world!")


@pipeline(mode_defs=[ModeDefinition(logger_defs={"my_json_logger": json_console_logger})])		@pipeline(
		mode_defs=[
		ModeDefinition(logger_defs={"my_json_logger": json_console_logger})
		]
		)
def demo_pipeline():		def demo_pipeline():
hello_logs()		hello_logs()


# end_custom_logger_marker_0		# end_custom_logger_marker_0

# start_custom_logger_testing		# start_custom_logger_testing


def test_init_json_console_logger():		def test_init_json_console_logger():
logger_ = json_console_logger(None)		logger_ = json_console_logger(None)
assert logger_.level == 20		assert logger_.level == 20
assert logger_.name == "dagster"		assert logger_.name == "dagster"


# end_custom_logger_testing		# end_custom_logger_testing

# start_custom_logger_testing_context		# start_custom_logger_testing_context
from dagster import build_init_logger_context		from dagster import build_init_logger_context


def test_init_json_console_logger_with_context():		def test_init_json_console_logger_with_context():
logger_ = json_console_logger(build_init_logger_context(logger_config={"name": "my_logger"}))		logger_ = json_console_logger(
		build_init_logger_context(logger_config={"name": "my_logger"})
		)
assert logger_.level == 20		assert logger_.level == 20
assert logger_.name == "my_logger"		assert logger_.name == "my_logger"


# end_custom_logger_testing_context		# end_custom_logger_testing_context