def arm(armIt): ''' Turn on and of the alarm ''' if armIt==True: steelsquid_utils.set_flag("alarm_security") else: steelsquid_utils.del_flag("alarm_security") GLOBAL.turn_off_alarm()
def enable(argument=None): ''' When this module is enabled what needs to be done (execute: steelsquid module XXX on) argument: Send data to the enable or disable method in the module Usually a string to tell the start/stop something Maybe you need create some files or enable other stuff. ''' steelsquid_utils.set_flag("auto") steelsquid_utils.del_flag("nokia") steelsquid_utils.del_flag("hdd") steelsquid_utils.del_flag("ssd")
def enable(argument=None): ''' When this module is enabled what needs to be done (execute: steelsquid module XXX on) Maybe you need create some files or enable other stuff. argument: Send data to the enable or disable method in the module Usually a string to tell the start/stop something ''' # Clear any saved settings for this module steelsquid_kiss_global.clear_modules_settings("kiss_squidrover") # Enable transeiver as client steelsquid_kiss_global.hmtrlrs_status("server") # Disable the automatic print if IP to LCD...this module will do it steelsquid_utils.set_flag("no_net_to_lcd")
def nrf24_status(status): ''' Enable nrf24 transmitter on this device Must reboot to implement status: server=Enable as server client=Enable as client master=Enable as master slave=Enable as slave None=Disable ''' if status == None: steelsquid_utils.del_flag("nrf24_server") steelsquid_utils.del_flag("nrf24_client") steelsquid_utils.del_flag("nrf24_master") steelsquid_utils.del_flag("nrf24_slave") elif status == "server": steelsquid_utils.set_flag("nrf24_server") steelsquid_utils.del_flag("nrf24_client") steelsquid_utils.del_flag("nrf24_master") steelsquid_utils.del_flag("nrf24_slave") elif status == "client": steelsquid_utils.del_flag("nrf24_server") steelsquid_utils.set_flag("nrf24_client") steelsquid_utils.del_flag("nrf24_master") steelsquid_utils.del_flag("nrf24_slave") elif status == "master": steelsquid_utils.del_flag("nrf24_server") steelsquid_utils.del_flag("nrf24_client") steelsquid_utils.set_flag("nrf24_master") steelsquid_utils.del_flag("nrf24_slave") elif status == "slave": steelsquid_utils.del_flag("nrf24_server") steelsquid_utils.del_flag("nrf24_client") steelsquid_utils.del_flag("nrf24_master") steelsquid_utils.set_flag("nrf24_slave")
def module_status(name, status, argument=None, restart=True): ''' Enable or disable a module name: Name of the mopule status: True/False argument: Send data to the enable or disable method in the module Usually a string to tell the start/stop something restart: restart the steelsquid daemon Return: Is it found ''' try: pkgpath = os.path.dirname(modules.__file__) doit = False for f in pkgutil.iter_modules([pkgpath]): if f[1] == name: doit = True if doit: try: mod = import_module('modules.' + name) if status: steelsquid_utils.set_flag("module_" + name) try: mod.enable(argument) except: steelsquid_utils.del_flag("module_" + name) steelsquid_utils.shout(string="Enable module error", is_error=True) return False else: steelsquid_utils.del_flag("module_" + name) try: mod.disable(argument) except: steelsquid_utils.set_flag("module_" + name) steelsquid_utils.shout(string="Disable module error", is_error=True) return False if restart: os.system('systemctl restart steelsquid') return True except: steelsquid_utils.shout(string="Import module error", is_error=True) return False else: return False steelsquid_utils.shout(string="Module not found") except: steelsquid_utils.shout() return False
def radio_tcp(is_the_remote, host=None): ''' Send and reseive messages with TCP (same way is with HM-TRLR-S transmitter) is_the_remote = this one send the push and request to the other is_the_remote = none, disable tcp server if host==None (this is a server, listen for connections) If host!=None this is the client (connect to server) ''' # Disable if is_the_remote == None: steelsquid_utils.del_flag("radio_tcp_server") steelsquid_utils.del_flag("radio_tcp_client") steelsquid_utils.del_flag("radio_tcp_remote") steelsquid_utils.del_parameter("radio_tcp_host") # Server elif host == None: steelsquid_utils.set_flag("radio_tcp_server") steelsquid_utils.del_flag("radio_tcp_client") if is_the_remote: steelsquid_utils.set_flag("radio_tcp_remote") else: steelsquid_utils.del_flag("radio_tcp_remote") steelsquid_utils.del_parameter("radio_tcp_host") # Client else: steelsquid_utils.del_flag("radio_tcp_server") steelsquid_utils.set_flag("radio_tcp_client") if is_the_remote: steelsquid_utils.set_flag("radio_tcp_remote") else: steelsquid_utils.del_flag("radio_tcp_remote") steelsquid_utils.set_parameter("radio_tcp_host", host)
def nrf24_status(status): ''' Enable nrf24 transmitter on this device Must reboot to implement status: server=Enable as server client=Enable as client master=Enable as master slave=Enable as slave None=Disable ''' if status==None: steelsquid_utils.del_flag("nrf24_server") steelsquid_utils.del_flag("nrf24_client") steelsquid_utils.del_flag("nrf24_master") steelsquid_utils.del_flag("nrf24_slave") elif status=="server": steelsquid_utils.set_flag("nrf24_server") steelsquid_utils.del_flag("nrf24_client") steelsquid_utils.del_flag("nrf24_master") steelsquid_utils.del_flag("nrf24_slave") elif status=="client": steelsquid_utils.del_flag("nrf24_server") steelsquid_utils.set_flag("nrf24_client") steelsquid_utils.del_flag("nrf24_master") steelsquid_utils.del_flag("nrf24_slave") elif status=="master": steelsquid_utils.del_flag("nrf24_server") steelsquid_utils.del_flag("nrf24_client") steelsquid_utils.set_flag("nrf24_master") steelsquid_utils.del_flag("nrf24_slave") elif status=="slave": steelsquid_utils.del_flag("nrf24_server") steelsquid_utils.del_flag("nrf24_client") steelsquid_utils.del_flag("nrf24_master") steelsquid_utils.set_flag("nrf24_slave")
def module_status(name, status, argument = None, restart=True): ''' Enable or disable a module name: Name of the mopule status: True/False argument: Send data to the enable or disable method in the module Usually a string to tell the start/stop something restart: restart the steelsquid daemon Return: Is it found ''' try: pkgpath = os.path.dirname(modules.__file__) doit=False for f in pkgutil.iter_modules([pkgpath]): if f[1]==name: doit=True if doit: try: mod = import_module('modules.'+name) if status: steelsquid_utils.set_flag("module_"+name) try: mod.enable(argument) except: steelsquid_utils.del_flag("module_"+name) steelsquid_utils.shout(string="Enable module error", is_error=True) return False else: steelsquid_utils.del_flag("module_"+name) try: mod.disable(argument) except: steelsquid_utils.set_flag("module_"+name) steelsquid_utils.shout(string="Disable module error", is_error=True) return False if restart: os.system('systemctl restart steelsquid') return True except: steelsquid_utils.shout(string="Import module error", is_error=True) return False else: return False steelsquid_utils.shout(string="Module not found") except: steelsquid_utils.shout() return False
def radio_hmtrlrs(is_server): ''' Send and reseive messages with HM-TRLR-S transmitter http://www.digikey.com/product-detail/en/HM-TRLR-S-868/HM-TRLR-S-868-ND/5051756 Must restart to implement is_server: None = Disable True=Enable as server False=Enable as client ''' if is_server == None: steelsquid_utils.del_flag("radio_hmtrlrs_server") steelsquid_utils.del_flag("radio_hmtrlrs_client") elif is_server: steelsquid_utils.set_flag("radio_hmtrlrs_server") steelsquid_utils.del_flag("radio_hmtrlrs_client") else: steelsquid_utils.del_flag("radio_hmtrlrs_server") steelsquid_utils.set_flag("radio_hmtrlrs_client")
def hmtrlrs_status(status): ''' Send and reseive messages with HM-TRLR-S transmitter http://www.digikey.com/product-detail/en/HM-TRLR-S-868/HM-TRLR-S-868-ND/5051756 Must reboot to implement status: server=Enable as server client=Enable as client None=Disable ''' if status==None: steelsquid_utils.del_flag("hmtrlrs_server") steelsquid_utils.del_flag("hmtrlrs_client") elif status=="server": steelsquid_utils.set_flag("hmtrlrs_server") steelsquid_utils.del_flag("hmtrlrs_client") elif status=="client": steelsquid_utils.del_flag("hmtrlrs_server") steelsquid_utils.set_flag("hmtrlrs_client")
def alarm_settings(session_id, parameters): ''' Settings of alarm ''' is_saved=False if len(parameters) > 0: if int(parameters[2]) >= int(parameters[3]): raise Exception("Alarm time must be smaller than alarm wait!") steelsquid_utils.set_parameter("alarm_security_movments", parameters[0]) steelsquid_utils.set_parameter("alarm_security_movments_seconds", parameters[1]) steelsquid_utils.set_parameter("alarm_security_seconds", parameters[2]) steelsquid_utils.set_parameter("alarm_security_wait", parameters[3]) if parameters[4]=="True": steelsquid_utils.set_flag("alarm_security_activate_siren") else: steelsquid_utils.del_flag("alarm_security_activate_siren") if parameters[5]=="True": steelsquid_utils.set_flag("alarm_security_send_mail") else: steelsquid_utils.del_flag("alarm_security_send_mail") steelsquid_utils.set_parameter("alarm_light_acivate", parameters[6]) if parameters[7]=="True": steelsquid_utils.set_flag("alarm_remote_siren") else: steelsquid_utils.del_flag("alarm_remote_siren") is_saved=True movments = steelsquid_utils.get_parameter("alarm_security_movments") movments_seconds = steelsquid_utils.get_parameter("alarm_security_movments_seconds") seconds = steelsquid_utils.get_parameter("alarm_security_seconds") wait_ = steelsquid_utils.get_parameter("alarm_security_wait") alarm_activate_siren = steelsquid_utils.get_flag("alarm_security_activate_siren") alarm_mail = steelsquid_utils.get_flag("alarm_security_send_mail") alarm_light_a = steelsquid_utils.get_parameter("alarm_light_acivate") alarm_remote_siren = steelsquid_utils.get_flag("alarm_remote_siren") return [is_saved, movments, movments_seconds, seconds, wait_, alarm_activate_siren, alarm_mail, alarm_light_a, alarm_remote_siren]
def task_event_thread(command, parameters=None): ''' execute a event ''' try: # Shutdown the system if command == "shutdown" or command == "stop": global running running = False event.set() # Enable a module elif command == "module_on": if len(parameters)>1: steelsquid_kiss_global.module_status(parameters[0], True, argument=parameters[1], restart=True) else: steelsquid_kiss_global.module_status(parameters[0], True, restart=True) # Disable a module elif command == "module_off": if len(parameters)>1: steelsquid_kiss_global.module_status(parameters[0], False, argument=parameters[1], restart=True) else: steelsquid_kiss_global.module_status(parameters[0], False, restart=True) # Set or del flag elif command == "flag": if parameters[0]=="set": steelsquid_utils.set_flag(parameters[1]) elif parameters[0]=="del": steelsquid_utils.del_flag(parameters[1]) # Set or del parameter elif command == "parameter": if parameters[0]=="set": steelsquid_utils.set_parameter(parameters[1], parameters[2]) elif parameters[0]=="del": steelsquid_utils.del_parameter(parameters[1]) # Reload modules elif command == "reload": if parameters!=None: for name, obj in steelsquid_kiss_global.loaded_modules.iteritems(): if name == parameters[0]: thread.start_new_thread(reload_file_dyn, (obj,)) else: for obj in steelsquid_kiss_global.loaded_modules.itervalues(): thread.start_new_thread(reload_file_dyn, (obj,)) # Execute a network event so the IP is shown execute_task_event("network") # Shout elif command == "shout": steelsquid_utils.shout(" ".join(parameters)) # VPN event elif command == "vpn": if parameters[0] == "up": name = "unknown" try: name = steelsquid_nm.vpn_configured() except: pass ip = steelsquid_utils.network_ip_vpn() steelsquid_kiss_global._execute_all_modules("SYSTEM", "on_vpn", (True, name, ip,)) steelsquid_utils.shout("VPN ENABLED\n"+name + "\nIP\n" + ip) else: steelsquid_utils.shout("VPN DISABLED") steelsquid_kiss_global._execute_all_modules("SYSTEM", "on_vpn", (False, None, None,)) # Network event elif command == "network": wired = steelsquid_utils.network_ip_wired() wifi = steelsquid_utils.network_ip_wifi() wan = "---" net = "---" access = "---" if wired == "---" and wifi == "---": net = "down" else: net = "up" wan = steelsquid_utils.network_ip_wan() if wifi != "---": try: access = steelsquid_nm.get_connected_access_point_name() except: pass on_network(net, wired, wifi, access, wan) elif command == "mount": the_type = parameters[0] the_remote = parameters[1] the_local = parameters[2] steelsquid_utils.shout("MOUNT %s\n%s\nTO\n%s" %(the_type, the_remote, the_local)) steelsquid_kiss_global._execute_all_modules("SYSTEM", "on_mount", (the_type, the_remote, the_local)) elif command == "umount": the_type = parameters[0] the_remote = parameters[1] the_local = parameters[2] steelsquid_utils.shout("UMOUNT %s\n%s\nFROM\n%s" %(the_type, the_remote, the_local)) steelsquid_kiss_global._execute_all_modules("SYSTEM", "on_umount", (the_type, the_remote, the_local)) # Try to broadcast the event to alla modules else: steelsquid_kiss_global._execute_all_modules("EVENTS", command, parameters) except: steelsquid_utils.shout()
def _cleanup(): ''' Clean ''' global running running = False event.set() try: for obj in steelsquid_kiss_global.loaded_modules.itervalues(): class_system = steelsquid_kiss_global._get_object(obj, "SYSTEM") if class_system!=None: method_event = steelsquid_kiss_global._get_object(class_system, "on_event_data") if method_event!=None: steelsquid_kiss_global.remove_event_data_callback(method_event) class_events = steelsquid_kiss_global._get_object(obj, "EVENTS") if class_events!=None: steelsquid_kiss_global.remove_broadcast_event_callback(class_events) if class_system!=None: steelsquid_kiss_global._exec_method_set_started(class_system, "on_stop", is_started=False) class_settings = steelsquid_kiss_global._get_object(obj, "SETTINGS") if class_settings!=None: persistent_off=False try: persistent_off = getattr(class_settings, "_persistent_off")==True except: pass if not persistent_off: members = [attr for attr in dir(class_settings) if not callable(getattr(class_settings, attr)) and not attr.startswith("_")] for var_name in members: the_var = getattr(class_settings, var_name, None) if isinstance(the_var, (bool)): if the_var: steelsquid_utils.set_flag(var_name) else: steelsquid_utils.del_flag(var_name) elif isinstance(the_var, list): steelsquid_utils.set_list(var_name, the_var) else: steelsquid_utils.set_parameter(var_name, str(the_var)) except: steelsquid_utils.shout() if steelsquid_utils.get_flag("nrf24_master") or steelsquid_utils.get_flag("nrf24_slave"): try: steelsquid_nrf24.stop() except: steelsquid_utils.shout() if steelsquid_utils.get_flag("hmtrlrs_server") or steelsquid_utils.get_flag("hmtrlrs_client"): try: steelsquid_hmtrlrs.stop() except: steelsquid_utils.shout() if steelsquid_utils.get_flag("socket_connection"): try: steelsquid_kiss_global.socket_connection.stop() except: pass if steelsquid_utils.get_flag("web"): try: steelsquid_kiss_global.http_server.stop_server() except: pass steelsquid_utils.execute_system_command_blind(['killall', 'aria2c']) steelsquid_utils.shout("Goodbye :-(")
def reload_file_dyn(obj): ''' Reload custom module ''' try: event.set() steelsquid_utils.shout("Reload module: " + obj.__name__) class_system = steelsquid_kiss_global._get_object(obj, "SYSTEM") if class_system!=None: method_event = steelsquid_kiss_global._get_object(class_system, "on_event_data") if method_event!=None: steelsquid_kiss_global.remove_event_data_callback(method_event) class_events = steelsquid_kiss_global._get_object(obj, "EVENTS") if class_events!=None: steelsquid_kiss_global.remove_broadcast_event_callback(class_events) if class_system!=None: steelsquid_kiss_global._exec_method_set_started(class_system, "on_stop", is_started=False) class_settings = steelsquid_kiss_global._get_object(obj, "SETTINGS") if class_settings!=None: persistent_off=False try: persistent_off = getattr(class_settings, "_persistent_off")==True except: pass if not persistent_off: members = [attr for attr in dir(class_settings) if not callable(getattr(class_settings, attr)) and not attr.startswith("_")] for var_name in members: the_var = getattr(class_settings, var_name, None) if isinstance(the_var, (bool)): if the_var: steelsquid_utils.set_flag(var_name) else: steelsquid_utils.del_flag(var_name) elif isinstance(the_var, list): steelsquid_utils.set_list(var_name, the_var) else: steelsquid_utils.set_parameter(var_name, str(the_var)) reload(obj) class_settings = steelsquid_kiss_global._get_object(obj, "SETTINGS") if class_settings!=None: persistent_off=False try: persistent_off = getattr(class_settings, "_persistent_off")==True except: pass if not persistent_off: members = [attr for attr in dir(class_settings) if not callable(getattr(class_settings, attr)) and not attr.startswith("_")] for var_name in members: the_var = getattr(class_settings, var_name, None) if isinstance(the_var, (bool)): the_var = steelsquid_utils.get_flag(var_name) setattr(class_settings, var_name, the_var) elif isinstance(the_var, list): the_var = steelsquid_utils.get_list(var_name, the_var) setattr(class_settings, var_name, the_var) elif isinstance(the_var, int): the_var = steelsquid_utils.get_parameter(var_name, str(the_var)) setattr(class_settings, var_name, int(the_var)) elif isinstance(the_var, float): the_var = steelsquid_utils.get_parameter(var_name, str(the_var)) setattr(class_settings, var_name, float(the_var)) else: the_var = steelsquid_utils.get_parameter(var_name, str(the_var)) setattr(class_settings, var_name, the_var) class_system = steelsquid_kiss_global._get_object(obj, "SYSTEM") if class_system!=None: steelsquid_kiss_global._exec_method_set_started(class_system, "on_start", is_started=True) method_event = steelsquid_kiss_global._get_object(class_system, "on_event_data") if method_event!=None: steelsquid_kiss_global.add_event_data_callback(method_event) class_events = steelsquid_kiss_global._get_object(obj, "EVENTS") if class_events!=None: steelsquid_kiss_global.add_broadcast_event_callback(class_events) class_loop = steelsquid_kiss_global._get_object(obj, "LOOP") event.clear() if class_loop!=None: for ob in vars(class_loop): m = getattr(class_loop, ob, None) if callable(m): thread.start_new_thread(do_on_loop, (m,)) except: steelsquid_utils.shout("Fatal error when reload module: " + obj.__name__, is_error=True)