from emod_api import schema_to_class as s2c
from emod_api.interventions import utils
from emodpy_hiv.interventions.utils import set_intervention_properties
from typing import List
[docs]def new_delay(
camp,
Bcast_Event,
Expire_Event="",
Coverage=1,
Delay=1,
Shelf_Life=36500,
Name="",
intervention_name: str = None,
disqualifying_properties: List[str] = None,
new_property_value: str = None
):
"""
Wrapper function to create and return a HIVDelayedIntervention intervention.
Args:
camp: emod_api.campaign object with schema_path set.
Returns:
ReadOnlyDict: Schema-based smart dictionary representing a new
"""
intervention = s2c.get_class_with_defaults( "HIVDelayedIntervention", camp.schema_path )
intervention.Broadcast_Event = Bcast_Event
intervention.Broadcast_On_Expiration_Event = Expire_Event
intervention.Coverage = Coverage
intervention.Delay_Period_Constant = Delay
intervention.Expiration_Period = Shelf_Life
if Name=="":
Name = "HIVDelayedIntervention"
intervention.Intervention_Name = Name
intervention.Dont_Allow_Duplicates = 1
set_intervention_properties(intervention,
intervention_name=intervention_name,
disqualifying_properties=disqualifying_properties,
new_property_value=new_property_value)
return intervention
[docs]def new_intervention_event(
camp,
bcast_event,
expire_event="",
coverage=1,
delay=1,
shelf_life=0,
intervention_name: str = None,
disqualifying_properties: List[str] = None,
new_property_value: str = None
):
"""
Delay as scheduled event.
"""
diag = new_delay( camp, bcast_event, expire_event, coverage, delay, shelf_life,
intervention_name=intervention_name, disqualifying_properties=disqualifying_properties,
new_property_value=new_property_value
)
# Coordinator
coordinator = s2c.get_class_with_defaults( "StandardEventCoordinator", camp.schema_path )
coordinator.Intervention_Config = diag
coordinator.Demographic_Coverage = coverage
# Event
event = s2c.get_class_with_defaults( "CampaignEvent", camp.schema_path )
event.Event_Coordinator_Config = coordinator
event.Start_Day = 1 # float(start_day)
event.Nodeset_Config = utils.do_nodes( camp.schema_path, [] )
from . import utils as hiv_utils
hiv_utils.declutter( event )
return event
[docs]def new_intervention_as_file( camp, start_day, filename=None ):
import emod_api.campaign as camp
camp.add( new_intervention_event( camp, bcast_event="HIVTestedPositive" ), first=True )
if filename is None:
filename = "HIVDelayedIntervention.json"
camp.save( filename )
return filename