def create_application(self, module_name): """ Create new application skel Args: module_name (string): module name Raises: CommandError: if command failed """ self.__stop_watcher() cmd = self.CLI_NEW_APPLICATION_CMD % (self.CLI, module_name) self.logger.debug('Create app cmd: %s' % cmd) try: console = Console() res = console.command(cmd, 10.0) self.logger.info('Create app cmd result: %s %s' % (res['stdout'], res['stderr'])) if res['returncode'] != 0: raise CommandError( 'Error during application creation. Check Cleep logs.') # sync new app content console.command(self.CLI_SYNC_MODULE_CMD % module_name) finally: self.__start_watcher()
def tweak_power_led(self, enable): """ Tweak raspberry pi power led Note: Infos from https://www.jeffgeerling.com/blogs/jeff-geerling/controlling-pwr-act-leds-raspberry-pi Args: enable (bool): True to turn on led """ if not os.path.exists('/sys/class/leds/led1'): self.logger.info('Power led not found on this device') return raspi = Tools.raspberry_pi_infos() off_value = '0' if raspi['model'].lower().find('zero') else '1' on_value = '1' if raspi['model'].lower().find('zero') else '0' echo_value = on_value if enable else off_value console = Console() resp = console.command('echo %s > /sys/class/leds/led1/brightness' % echo_value) if resp['returncode'] != 0: raise CommandError('Error tweaking power led') # store led status self._set_config_field('enablepowerled', enable)
def download_documentation(self, module_name): """ Download documentation (html as archive tar.gz) Args: module_name (string): module name Returns: dict: archive infos:: { data: filepath filename: new filename } Raises: CommandError: command failed error """ self.logger.info('Download documentation html archive') cmd = self.CLI_DOCS_ZIP_PATH_CMD % (self.CLI, module_name) self.logger.debug('Doc zip path cmd: %s' % cmd) console = Console() res = console.command(cmd) if res['returncode'] != 0: raise CommandError(''.join(res['stdout'])) zip_path = res['stdout'][0].split('=')[1] self.logger.debug('Module "%s" docs path "%s"' % (module_name, zip_path)) return {'filepath': zip_path, 'filename': os.path.basename(zip_path)}
def build_application(self, module_name): """ Build application archive (zip format) Archive is not protected by password Args: module_name (string): module name Raises: Exception: if build failed """ cmd = self.CLI_BUILD_APP_CMD % (self.CLI, module_name) self.logger.debug('Build app cmd: %s' % cmd) console = Console() res = console.command(cmd, 60.0) self.logger.info('Build app result: %s | %s' % (res['stdout'], res['stderr'])) if res['returncode'] != 0: raise CommandError('Error building application. Check Cleep logs.') try: self.__last_application_build = json.loads(res['stdout'][0]) except Exception as error: self.logger.exception( 'Error parsing app build command "%s" output' % cmd) raise CommandError( 'Error building application. Check Cleep logs.') from error
def __cli_check(self, command, timeout=15.0): """ Execute cleep-cli check specified by command Args: command (string): cli command to execute Returns: dict: command output """ console = Console() res = console.command(command, timeout) self.logger.debug('Cli command "%s" output: %s | %s' % (command, res['stdout'], res['stderr'])) if res['returncode'] != 0: self.logger.exception('Command "%s"', command) raise CommandError('Command failed') try: return json.loads(''.join(res['stdout'])) except Exception as error: self.logger.exception('Error parsing command "%s" output' % command) raise CommandError( 'Error parsing check result. Check Cleep logs') from error
def _execute_command(self, sensor): # pragma: no cover """ Execute dht22 binary command Useful for unit testing """ console = Console() cmd = self.DHT22_CMD % sensor["gpios"][0]["pin"] self.logger.debug('Read DHT22 sensor values from command "%s"' % cmd) resp = console.command(cmd, timeout=11) self.logger.debug("Read DHT command response: %s" % resp) if resp['error'] or resp['killed']: self.logger.error("DHT22 command failed: %s" % resp) return json.loads(resp['stdout'][0])
def sync_time(): """ Synchronize device time using NTP server Note: This command may lasts some seconds Returns: bool: True if NTP sync succeed, False otherwise """ console = Console() resp = console.command("/usr/sbin/ntpdate-debian", timeout=60.0) return resp["returncode"] == 0
def tweak_activity_led(self, enable): """ Tweak raspberry pi activity led Args: enable (bool): True to turn on led """ if not os.path.exists('/sys/class/leds/led0'): self.logger.info('Activity led not found on this device') return raspi = Tools.raspberry_pi_infos() off_value = '0' if raspi['model'].lower().find('zero') else '1' on_value = '1' if raspi['model'].lower().find('zero') else '0' echo_value = on_value if enable else off_value console = Console() resp = console.command('echo %s > /sys/class/leds/led0/brightness' % echo_value) if resp['returncode'] != 0: raise CommandError('Error tweaking activity led') # store led status self._set_config_field('enableactivityled', enable)
def set_timezone(self): """ Set timezone according to coordinates Returns: bool: True if function succeed, False otherwise Raises: CommandError: if unable to save timezone """ # get position position = self._get_config_field("position") if not position["latitude"] and not position["longitude"]: self.logger.warning( "Unable to set timezone from unspecified position (%s)" % position) return False # compute timezone current_timezone = None try: # try to find timezone at position current_timezone = self.timezonefinder.timezone_at( lat=position["latitude"], lng=position["longitude"]) if current_timezone is None: # extend search to closest position # TODO increase delta_degree to extend research, careful it use more CPU ! current_timezone = self.timezonefinder.closest_timezone_at( lat=position["latitude"], lng=position["longitude"]) except ValueError: # the coordinates were out of bounds self.logger.exception("Coordinates out of bounds") except Exception: self.logger.exception( "Error occured searching timezone at position") if not current_timezone: self.logger.warning( "Unable to set device timezone because it was not found") return False # save timezone value self.logger.debug("Save new timezone: %s" % current_timezone) if not self._set_config_field("timezone", current_timezone): raise CommandError("Unable to save timezone") # configure system timezone zoneinfo = os.path.join(self.SYSTEM_ZONEINFO_DIR, current_timezone) self.logger.debug("Checking zoneinfo file: %s" % zoneinfo) if not os.path.exists(zoneinfo): raise CommandError('No system file found for "%s" timezone' % current_timezone) self.logger.debug('zoneinfo file "%s" exists' % zoneinfo) self.cleep_filesystem.rm(self.SYSTEM_LOCALTIME) self.logger.debug('Writing timezone "%s" in "%s"' % (current_timezone, self.SYSTEM_TIMEZONE)) if not self.cleep_filesystem.write_data(self.SYSTEM_TIMEZONE, "%s" % current_timezone): self.logger.error( 'Unable to write timezone data on "%s". System timezone is not configured!' % self.SYSTEM_TIMEZONE) return False # launch timezone update in background self.logger.debug("Updating system timezone") command = Console() res = command.command( "/usr/sbin/dpkg-reconfigure -f noninteractive tzdata", timeout=15.0) self.logger.debug("Timezone update command result: %s" % res) if res["returncode"] != 0: self.logger.error("Error reconfiguring system timezone: %s" % res["stderr"]) return False # TODO configure all wpa_supplicant.conf country code # propagate changes to cleep self.timezone = timezone(current_timezone) self._time_task() return True
def __kill_watchers(self): """ Kill all watchers instances """ console = Console() console.command('/usr/bin/pkill -9 -f "cleep-cli watch"')
class UsbAudioDriver(AudioDriver): """ Audio driver for USB audio devices Tested hardare: * Mini external USB stereo speaker: https://thepihut.com/collections/raspberry-pi-usb-audio/products/mini-external-usb-stereo-speaker """ VOLUME_PATTERN = ("Mono", r"\[(\d*)%\]") def __init__(self): """ Constructor """ AudioDriver.__init__(self, "USB audio device") self.asoundconf = None self.configtxt = None self.console = None self.volume_control = "" self.volume_control_numid = None def _on_audio_registered(self): """ Audio driver registered """ self.asoundconf = EtcAsoundConf(self.cleep_filesystem) self.configtxt = ConfigTxt(self.cleep_filesystem) self.console = Console() def _get_card_name(self, devices_names): """ Return card name Returns: string: card name or None if card not found """ pattern = re.compile("usb", re.IGNORECASE) for device_name in devices_names: if pattern.match(device_name["device_desc"]): return device_name["card_name"] return None def get_card_capabilities(self): """ Return card capabilities Returns: tuple: card capabilities:: ( bool: playback capability, bool: capture capability ) """ return (True, False) def _install(self, params=None): """ Install driver Args: params (dict): additional parameters """ # as the default driver and just in case, delete existing config self.asoundconf.delete() # install pulseaudio debian package resp = self.console.command( "apt update -qq && apt install -q --yes pulseaudio", timeout=300) if resp["returncode"] != 0: self.logger.error("Unable to install USB audio: %s", resp) return False # installing native audio device consists of enabling dtparam audio in /boot/config.txt if not self.configtxt.enable_audio(): raise Exception("Error enabling USB audio") return True def _uninstall(self, params=None): """ Uninstall driver Args: params (dict): additional parameters """ resp = self.console.command("apt purge --q --yes pulseaudio") if resp["returncode"] != 0: self.logger.error("Unable to uninstall USB audio: %s", resp) raise Exception('Unable to uninstall USB audio') return True def is_installed(self): """ Is driver installed Returns: bool: True if driver is installed """ resp = self.console.command("dpkg -s pulseaudio") return resp["returncode"] == 0 def enable(self, params=None): """ Enable driver """ if not self.get_card_name(): raise Exception( "No USB audio found. Please connect it before enabling it") # as the default driver and just in case, delete existing config self.asoundconf.delete() # create default /etc/asound.conf card_infos = self.get_cardid_deviceid() self.logger.debug("card_infos=%s", card_infos) if card_infos[0] is None: self.logger.error('Unable to get alsa infos for card "%s"', self.get_card_name()) return False self.logger.debug('Write to /etc/asound.conf values "%s:%s"', card_infos[0], card_infos[1]) if not self.asoundconf.save_default_file(card_infos[0], card_infos[1]): self.logger.error( 'Unable to create /etc/asound.conf for soundcard "%s"', self.get_card_name(), ) return False # force saving alsa conf (this will create asound.state if needed) self.alsa.save() return True def disable(self, params=None): """ Disable driver Args: params (dict): additional parameters """ self.logger.debug( "Delete /etc/asound.conf and /var/lib/alsa/asound.state") if not self.asoundconf.delete(): self.logger.error("Unable to delete asound.conf file") return False self.logger.debug("Driver disabled") return True def is_enabled(self): """ Is driver enabled ? Returns: bool: True if driver enabled """ card = self.is_card_enabled() asound = self.asoundconf.exists() return card and asound def _set_volumes_controls(self): """ Set controls to control volumes """ # search for appropriate volume control controls = self.alsa.get_simple_controls() self.volume_control = controls[0] if len(controls) > 0 else "" # get volume control numid self.volume_control_numid = self.get_control_numid("Volume") def get_volumes(self): """ Get volumes Returns: dict: volumes level:: { playback (float): playback volume capture (float): capture volume } """ return { "playback": self.alsa.get_volume(self.volume_control, self.VOLUME_PATTERN), "capture": None, } def set_volumes(self, playback=None, capture=None): """ Set volumes Args: playback (float): playback volume (None to disable update) capture (float): capture volume (None to disable update) Returns: dict: volumes level:: { playback (float): playback volume capture (float): capture volume } """ return { "playback": self.alsa.set_volume(self.volume_control, self.VOLUME_PATTERN, playback), "capture": None, } def require_reboot(self): """ Require reboot after install/uninstall Returns: bool: True if reboot required """ return False