@ -1,7 +1,7 @@
#!/usr/bin/python3 #!/usr/bin/python3
""" """
SkywarnPlus v0 .2 . 0 by Mason Nelson ( N5LSN / WRKF394 ) SkywarnPlus v0 .2 . 2 by Mason Nelson ( N5LSN / WRKF394 )
== == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == == ==
SkywarnPlus is a utility that retrieves severe weather alerts from the National SkywarnPlus is a utility that retrieves severe weather alerts from the National
Weather Service and integrates these alerts with an Asterisk / app_rpt based Weather Service and integrates these alerts with an Asterisk / app_rpt based
@ -49,7 +49,9 @@ if not master_enable:
# Define tmp_dir and sounds_path # Define tmp_dir and sounds_path
tmp_dir = config . get ( " DEV " , { } ) . get ( " TmpDir " , " /tmp/SkywarnPlus " ) tmp_dir = config . get ( " DEV " , { } ) . get ( " TmpDir " , " /tmp/SkywarnPlus " )
sounds_path = config . get ( " Alerting " , { } ) . get ( " SoundsPath " , os . path . join ( baseDir , " SOUNDS " ) ) sounds_path = config . get ( " Alerting " , { } ) . get (
" SoundsPath " , os . path . join ( baseDir , " SOUNDS " )
)
# Define countyCodes # Define countyCodes
countyCodes = config . get ( " Alerting " , { } ) . get ( " CountyCodes " , [ ] ) countyCodes = config . get ( " Alerting " , { } ) . get ( " CountyCodes " , [ ] )
@ -61,18 +63,14 @@ else:
print ( " Error: tmp_dir is not set. " )
print ( " Error: tmp_dir is not set. " )
# Define Blocked events # Define Blocked events
global_blocked_events = ( global_blocked_events = config . get ( " Blocking " , { } ) . get ( " GlobalBlockedEvents " , [ ] )
config . get ( " Blocking " , { } ) . get ( " GlobalBlockedEvents " , [ ] )
)
if global_blocked_events is None : if global_blocked_events is None :
global_blocked_events = [ ]
global_blocked_events = [ ]
sayalert_blocked_events = ( sayalert_blocked_events = config . get ( " Blocking " , { } ) . get ( " SayAlertBlockedEvents " , [ ] )
config . get ( " Blocking " , { } ) . get ( " SayAlertBlockedEvents " , [ ] )
)
if sayalert_blocked_events is None : if sayalert_blocked_events is None :
sayalert_blocked_events = [ ]
sayalert_blocked_events = [ ]
tailmessage_blocked_events = ( tailmessage_blocked_events = config . get ( " Blocking " , { } ) . get (
config . get ( " Blocking " , { } ) . get ( " TailmessageBlockedEvents " , [ ] )
" TailmessageBlockedEvents " , [ ]
) )
if tailmessage_blocked_events is None : if tailmessage_blocked_events is None :
tailmessage_blocked_events = [ ]
tailmessage_blocked_events = [ ]
@ -267,20 +265,22 @@ def load_state():
Load the state from the state file if it exists , else return an initial state .
Load the state from the state file if it exists , else return an initial state .
Returns :
Returns :
dict : A dictionary containing courtesy tone ( ct ) , identifier ( id ) and alerts .
dict : A dictionary containing data .
"""
"""
if os . path . exists ( data_file ) :
if os . path . exists ( data_file ) :
with open ( data_file , " r " ) as file :
with open ( data_file , " r " ) as file :
state = json . load ( file )
state = json . load ( file )
# state["alertscript_alerts"] = set(state["alertscript_alerts"])
state [ " alertscript_alerts " ] = state [ " alertscript_alerts " ]
# state["last_alerts"] = set(state["last_alerts"])
state [ " last_alerts " ] = state [ " last_alerts " ]
state [ " last_sayalert " ] = state . get ( " last_sayalert " , [ ] )
return state
return state
else :
else :
return {
return {
" ct " : None ,
" ct " : None ,
" id " : None ,
" id " : None ,
" alertscript_alerts " : set ( ) ,
" alertscript_alerts " : [ ] ,
" last_alerts " : set ( ) ,
" last_alerts " : [ ] ,
" last_sayalert " : [ ] ,
}
}
@ -289,10 +289,11 @@ def save_state(state):
Save the state to the state file .
Save the state to the state file .
Args :
Args :
state ( dict ) : A dictionary containing courtesy tone ( ct ) , identifier ( id ) and alerts .
state ( dict ) : A dictionary containing data .
"""
"""
state [ " alertscript_alerts " ] = list ( state [ " alertscript_alerts " ] )
state [ " alertscript_alerts " ] = list ( state [ " alertscript_alerts " ] )
state [ " last_alerts " ] = list ( state [ " last_alerts " ] )
state [ " last_alerts " ] = list ( state [ " last_alerts " ] )
state [ " last_sayalert " ] = list ( state [ " last_sayalert " ] )
with open ( data_file , " w " ) as file :
with open ( data_file , " w " ) as file :
json . dump ( state , file )
json . dump ( state , file )
@ -321,9 +322,7 @@ def getAlerts(countyCodes):
# Inject alerts if DEV INJECT is enabled in the config
# Inject alerts if DEV INJECT is enabled in the config
if config . get ( " DEV " , { } ) . get ( " INJECT " , False ) :
if config . get ( " DEV " , { } ) . get ( " INJECT " , False ) :
logger . debug ( " getAlerts: DEV Alert Injection Enabled " )
logger . debug ( " getAlerts: DEV Alert Injection Enabled " )
alerts = [
alerts = [ alert . strip ( ) for alert in config [ " DEV " ] . get ( " INJECTALERTS " , [ ] ) ]
alert . strip ( ) for alert in config [ " DEV " ] . get ( " INJECTALERTS " , [ ] )
]
logger . debug ( " getAlerts: Injecting alerts: %s " , alerts )
logger . debug ( " getAlerts: Injecting alerts: %s " , alerts )
return alerts
return alerts
@ -397,6 +396,28 @@ def sayAlert(alerts):
alerts ( list ) : List of active weather alerts .
alerts ( list ) : List of active weather alerts .
"""
"""
# Define the path of the alert file
# Define the path of the alert file
state = load_state ( )
filtered_alerts = [ ]
for alert in alerts :
if any (
fnmatch . fnmatch ( alert , blocked_event )
for blocked_event in sayalert_blocked_events
) :
logger . debug ( " sayAlert: blocking %s as per configuration " , alert )
continue
filtered_alerts . append ( alert )
# Check if the filtered alerts are the same as the last run
if filtered_alerts == state [ " last_sayalert " ] :
logger . debug (
" sayAlert: The filtered alerts are the same as the last run. Not broadcasting. "
)
return
state [ " last_sayalert " ] = filtered_alerts
save_state ( state )
alert_file = " {} /alert.wav " . format ( sounds_path )
alert_file = " {} /alert.wav " . format ( sounds_path )
combined_sound = AudioSegment . from_wav (
combined_sound = AudioSegment . from_wav (
@ -407,15 +428,7 @@ def sayAlert(alerts):
)
)
alert_count = 0
alert_count = 0
for alert in filtered_alerts :
for alert in alerts :
if any (
fnmatch . fnmatch ( alert , blocked_event )
for blocked_event in sayalert_blocked_events
) :
logger . debug ( " sayAlert: blocking %s as per configuration " , alert )
continue
try :
try :
index = WS . index ( alert )
index = WS . index ( alert )
audio_file = AudioSegment . from_wav (
audio_file = AudioSegment . from_wav (
@ -437,7 +450,8 @@ def sayAlert(alerts):
if alert_count == 0 :
if alert_count == 0 :
logger . debug ( " sayAlert: All alerts were blocked, not broadcasting any alerts. " )
logger . debug ( " sayAlert: All alerts were blocked, not broadcasting any alerts. " )
else :
return
logger . debug ( " sayAlert: Exporting alert sound to %s " , alert_file )
logger . debug ( " sayAlert: Exporting alert sound to %s " , alert_file )
converted_combined_sound = convertAudio ( combined_sound )
converted_combined_sound = convertAudio ( combined_sound )
converted_combined_sound . export ( alert_file , format = " wav " )
converted_combined_sound . export ( alert_file , format = " wav " )
@ -463,6 +477,11 @@ def sayAllClear():
"""
"""
Generate and broadcast ' all clear ' message on Asterisk .
Generate and broadcast ' all clear ' message on Asterisk .
"""
"""
# Empty the last_sayalert list so that the next run will broadcast alerts
state = load_state ( )
state [ " last_sayalert " ] = [ ]
save_state ( state )
alert_clear = os . path . join ( sounds_path , " ALERTS " , " SWP96.wav " )
alert_clear = os . path . join ( sounds_path , " ALERTS " , " SWP96.wav " )
node_numbers = config . get ( " Asterisk " , { } ) . get ( " Nodes " , [ ] )
node_numbers = config . get ( " Asterisk " , { } ) . get ( " Nodes " , [ ] )
@ -553,7 +572,9 @@ def changeCT(ct):
"""
"""
state = load_state ( )
state = load_state ( )
current_ct = state [ " ct " ]
current_ct = state [ " ct " ]
tone_dir = config [ " CourtesyTones " ] . get ( " ToneDir " , os . path . join ( sounds_path , " TONES " ) )
tone_dir = config [ " CourtesyTones " ] . get (
" ToneDir " , os . path . join ( sounds_path , " TONES " )
)
ct1 = config [ " CourtesyTones " ] [ " Tones " ] [ " CT1 " ]
ct1 = config [ " CourtesyTones " ] [ " Tones " ] [ " CT1 " ]
ct2 = config [ " CourtesyTones " ] [ " Tones " ] [ " CT2 " ]
ct2 = config [ " CourtesyTones " ] [ " Tones " ] [ " CT2 " ]
wx_ct = config [ " CourtesyTones " ] [ " Tones " ] [ " WXCT " ]
wx_ct = config [ " CourtesyTones " ] [ " Tones " ] [ " WXCT " ]
@ -809,10 +830,13 @@ def change_and_log_CT_or_ID(
alert_type ,
alert_type ,
specified_alerts ,
specified_alerts ,
)
)
# Check if any alert matches specified_alerts
# Check if any alert matches specified_alerts
if set ( alerts ) . intersection ( specified_alerts ) :
# Here we replace set intersection with a list comprehension
for alert in alerts :
intersecting_alerts = [ alert for alert in alerts if alert in specified_alerts ]
if alert in specified_alerts :
if intersecting_alerts :
for alert in intersecting_alerts :
logger . debug ( " Alert %s requires a %s change " , alert , alert_type )
logger . debug ( " Alert %s requires a %s change " , alert , alert_type )
if (
if (
changeCT ( " WX " ) if alert_type == " CT " else changeID ( " WX " )
changeCT ( " WX " ) if alert_type == " CT " else changeID ( " WX " )
@ -862,6 +886,8 @@ def main():
alerts = getAlerts ( countyCodes )
alerts = getAlerts ( countyCodes )
# If new alerts differ from old ones, process new alerts
# If new alerts differ from old ones, process new alerts
logger . debug ( " Last alerts: %s " , last_alerts )
logger . debug ( " New alerts: %s " , alerts )
if last_alerts != alerts :
if last_alerts != alerts :
state [ " last_alerts " ] = alerts
state [ " last_alerts " ] = alerts
save_state ( state )
save_state ( state )
Mason10198
3 years ago