@@ -325,17 +325,30 @@ impl PyEnvironmentActions {
325325/// schema. If both flavours of the same field are passed, the
326326/// snake-case form wins.
327327#[ new]
328- #[ pyo3( signature = ( * , on_enter=None , on_exit=None , onEnter=None , onExit=None ) ) ]
328+ #[ allow( clippy:: too_many_arguments) ]
329+ #[ pyo3( signature = ( * , on_enter=None , on_exit=None , on_wrap_env_enter=None , on_wrap_task_run=None , on_wrap_env_exit=None , onEnter=None , onExit=None , onWrapEnvEnter=None , onWrapTaskRun=None , onWrapEnvExit=None ) ) ]
329330 fn new (
330331 on_enter : Option < PyAction > ,
331332 on_exit : Option < PyAction > ,
333+ on_wrap_env_enter : Option < PyAction > ,
334+ on_wrap_task_run : Option < PyAction > ,
335+ on_wrap_env_exit : Option < PyAction > ,
332336 #[ allow( non_snake_case) ] onEnter : Option < PyAction > ,
333337 #[ allow( non_snake_case) ] onExit : Option < PyAction > ,
338+ #[ allow( non_snake_case) ] onWrapEnvEnter : Option < PyAction > ,
339+ #[ allow( non_snake_case) ] onWrapTaskRun : Option < PyAction > ,
340+ #[ allow( non_snake_case) ] onWrapEnvExit : Option < PyAction > ,
334341 ) -> Self {
335342 let on_enter = on_enter. or ( onEnter) ;
336343 let on_exit = on_exit. or ( onExit) ;
344+ let on_wrap_env_enter = on_wrap_env_enter. or ( onWrapEnvEnter) ;
345+ let on_wrap_task_run = on_wrap_task_run. or ( onWrapTaskRun) ;
346+ let on_wrap_env_exit = on_wrap_env_exit. or ( onWrapEnvExit) ;
337347 PyEnvironmentActions {
338348 inner : EnvironmentActions {
349+ on_wrap_env_enter : on_wrap_env_enter. map ( |a| a. inner ) ,
350+ on_wrap_task_run : on_wrap_task_run. map ( |a| a. inner ) ,
351+ on_wrap_env_exit : on_wrap_env_exit. map ( |a| a. inner ) ,
339352 on_enter : on_enter. map ( |a| a. inner ) ,
340353 on_exit : on_exit. map ( |a| a. inner ) ,
341354 } ,
@@ -370,6 +383,48 @@ impl PyEnvironmentActions {
370383 self . on_exit ( )
371384 }
372385
386+ #[ getter]
387+ fn on_wrap_env_enter ( & self ) -> Option < PyAction > {
388+ self . inner
389+ . on_wrap_env_enter
390+ . as_ref ( )
391+ . map ( |a| PyAction { inner : a. clone ( ) } )
392+ }
393+
394+ #[ getter]
395+ #[ pyo3( name = "onWrapEnvEnter" ) ]
396+ fn on_wrap_env_enter_camel ( & self ) -> Option < PyAction > {
397+ self . on_wrap_env_enter ( )
398+ }
399+
400+ #[ getter]
401+ fn on_wrap_task_run ( & self ) -> Option < PyAction > {
402+ self . inner
403+ . on_wrap_task_run
404+ . as_ref ( )
405+ . map ( |a| PyAction { inner : a. clone ( ) } )
406+ }
407+
408+ #[ getter]
409+ #[ pyo3( name = "onWrapTaskRun" ) ]
410+ fn on_wrap_task_run_camel ( & self ) -> Option < PyAction > {
411+ self . on_wrap_task_run ( )
412+ }
413+
414+ #[ getter]
415+ fn on_wrap_env_exit ( & self ) -> Option < PyAction > {
416+ self . inner
417+ . on_wrap_env_exit
418+ . as_ref ( )
419+ . map ( |a| PyAction { inner : a. clone ( ) } )
420+ }
421+
422+ #[ getter]
423+ #[ pyo3( name = "onWrapEnvExit" ) ]
424+ fn on_wrap_env_exit_camel ( & self ) -> Option < PyAction > {
425+ self . on_wrap_env_exit ( )
426+ }
427+
373428 fn __repr__ ( & self ) -> String {
374429 format ! (
375430 "EnvironmentActions(on_enter={}, on_exit={})" ,
@@ -398,6 +453,15 @@ impl PyEnvironmentActions {
398453 if let Some ( a) = slf. on_exit ( ) {
399454 kwargs. set_item ( "on_exit" , a) ?;
400455 }
456+ if let Some ( a) = slf. on_wrap_env_enter ( ) {
457+ kwargs. set_item ( "on_wrap_env_enter" , a) ?;
458+ }
459+ if let Some ( a) = slf. on_wrap_task_run ( ) {
460+ kwargs. set_item ( "on_wrap_task_run" , a) ?;
461+ }
462+ if let Some ( a) = slf. on_wrap_env_exit ( ) {
463+ kwargs. set_item ( "on_wrap_env_exit" , a) ?;
464+ }
401465 let cls = py. get_type :: < PyEnvironmentActions > ( ) ;
402466 let args = PyTuple :: new ( py, [ cls. into_any ( ) , kwargs. into_any ( ) ] ) ?;
403467 Ok ( ( helper, args. into ( ) ) )
0 commit comments