Script for helping set the speech-dispatcher voice and module.
This commit is contained in:
Storm Dragon
parent
0a2e9533bd
commit
bd975df912
242
tools/speechd_config.py
Executable file
242
tools/speechd_config.py
Executable file
@ -0,0 +1,242 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Voice configuration script
|
||||
# By Storm Dragon
|
||||
|
||||
import configparser
|
||||
import subprocess
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
import select
|
||||
import tempfile
|
||||
|
||||
def check_root_access():
|
||||
"""Check if script is running as root and determine available privilege escalation command"""
|
||||
if os.geteuid() == 0:
|
||||
return True, None
|
||||
sudoPath = run_command(['which', 'sudo'])
|
||||
if sudoPath:
|
||||
return False, 'sudo'
|
||||
doasPath = run_command(['which', 'doas'])
|
||||
if doasPath:
|
||||
return False, 'doas'
|
||||
return False, None
|
||||
|
||||
def run_command(cmd, needsRoot=False, privilegeCmd=None):
|
||||
"""Run a command with optional root privileges"""
|
||||
try:
|
||||
if needsRoot and privilegeCmd and os.geteuid() != 0:
|
||||
fullCmd = [privilegeCmd] + cmd
|
||||
else:
|
||||
fullCmd = cmd
|
||||
result = subprocess.run(fullCmd, capture_output=True, text=True)
|
||||
if result.returncode != 0:
|
||||
print(f"Command failed: {' '.join(fullCmd)}")
|
||||
print(f"Error: {result.stderr}")
|
||||
return None
|
||||
return result.stdout.strip()
|
||||
except Exception as e:
|
||||
print(f"Error running command {' '.join(cmd)}: {e}")
|
||||
return None
|
||||
|
||||
def run_command_with_output(cmd, needsRoot=False, privilegeCmd=None):
|
||||
"""Run a command and return its content directly"""
|
||||
try:
|
||||
if needsRoot and privilegeCmd and os.geteuid() != 0:
|
||||
cmd = [privilegeCmd] + cmd
|
||||
with subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) as process:
|
||||
output, error = process.communicate()
|
||||
if process.returncode != 0:
|
||||
print(f"Error: {error}")
|
||||
return None
|
||||
return output
|
||||
except Exception as e:
|
||||
print(f"Error running command {' '.join(cmd)}: {e}")
|
||||
return None
|
||||
|
||||
def get_speechd_modules(privilegeCmd):
|
||||
"""Get list of available speech-dispatcher modules"""
|
||||
output = run_command_with_output(['spd-say', '-O'], True, privilegeCmd)
|
||||
if output:
|
||||
lines = output.split('\n')
|
||||
if len(lines) > 1:
|
||||
return [line.strip() for line in lines[1:] if line.strip()]
|
||||
return []
|
||||
|
||||
def process_espeak_voice(voiceLine):
|
||||
"""Process espeak-ng voice line to get voice name from last two columns"""
|
||||
parts = [p for p in voiceLine.split() if p]
|
||||
if len(parts) < 2:
|
||||
return None
|
||||
langCode = parts[-2].lower()
|
||||
variant = parts[-1].lower()
|
||||
if variant and variant != 'none':
|
||||
return f"{langCode}+{variant}"
|
||||
return langCode
|
||||
|
||||
def get_module_voices(moduleName, privilegeCmd):
|
||||
"""Get available voices for a specific module"""
|
||||
output = run_command_with_output(['spd-say', '-o', moduleName, '-L'], True, privilegeCmd)
|
||||
if output:
|
||||
lines = output.split('\n')
|
||||
if len(lines) > 1:
|
||||
voices = []
|
||||
for line in lines[1:]:
|
||||
if not line.strip():
|
||||
continue
|
||||
if moduleName.lower() == 'espeak-ng':
|
||||
processedVoice = process_espeak_voice(line)
|
||||
if processedVoice:
|
||||
voices.append(processedVoice)
|
||||
else:
|
||||
voices.append(line.strip())
|
||||
return voices
|
||||
return []
|
||||
|
||||
def display_paginated_voices(voiceList):
|
||||
"""Display voices in pages of 30"""
|
||||
pageSize = 30
|
||||
totalVoices = len(voiceList)
|
||||
totalPages = (totalVoices + pageSize - 1) // pageSize
|
||||
currentPage = 0
|
||||
while currentPage < totalPages:
|
||||
start = currentPage * pageSize
|
||||
end = min(start + pageSize, totalVoices)
|
||||
print(f"\nShowing voices {start + 1}-{end} of {totalVoices} (Page {currentPage + 1} of {totalPages})")
|
||||
print("Available voices:")
|
||||
for i, voice in enumerate(voiceList[start:end], start + 1):
|
||||
print(f"{i}. {voice}")
|
||||
if currentPage < totalPages - 1:
|
||||
print("\nPress Enter for next page, 'b' for previous page, or enter a number to select a voice (0 to go back): ")
|
||||
else:
|
||||
print("\nPress 'b' for previous page, or enter a number to select a voice (0 to go back): ")
|
||||
choice = input().strip().lower()
|
||||
if choice == '':
|
||||
if currentPage < totalPages - 1:
|
||||
currentPage += 1
|
||||
elif choice == 'b':
|
||||
if currentPage > 0:
|
||||
currentPage -= 1
|
||||
elif choice.isdigit():
|
||||
choice = int(choice)
|
||||
if choice == 0:
|
||||
return None
|
||||
if 1 <= choice <= totalVoices:
|
||||
return voiceList[choice - 1]
|
||||
print("Invalid selection!")
|
||||
else:
|
||||
print("Invalid input!")
|
||||
return None
|
||||
|
||||
def test_voice_config(moduleName, voiceName, privilegeCmd):
|
||||
"""Test if the selected voice works"""
|
||||
print("\nTesting voice configuration...")
|
||||
testMessage = "If you hear this message, press Enter within 30 seconds to confirm. Otherwise, wait for timeout to return to the menu."
|
||||
try:
|
||||
cmd = ['spd-say', '-o', moduleName, '-y', voiceName, testMessage]
|
||||
if privilegeCmd and os.geteuid() != 0:
|
||||
cmd = [privilegeCmd] + cmd
|
||||
process = subprocess.Popen(cmd)
|
||||
startTime = time.time()
|
||||
while time.time() - startTime < 30:
|
||||
if select.select([sys.stdin], [], [], 0.0)[0]:
|
||||
userInput = input()
|
||||
process.terminate()
|
||||
return True
|
||||
time.sleep(0.1)
|
||||
process.terminate()
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Error testing voice: {e}")
|
||||
return False
|
||||
|
||||
def update_fenrir_config(configPath, moduleName, voiceName, privilegeCmd):
|
||||
"""Update Fenrir configuration with new speech settings"""
|
||||
config = configparser.ConfigParser()
|
||||
try:
|
||||
configContent = run_command_with_output(['cat', configPath], True, privilegeCmd)
|
||||
if not configContent:
|
||||
return False
|
||||
with tempfile.NamedTemporaryFile(mode='w', delete=False) as tempFile:
|
||||
tempPath = tempFile.name
|
||||
tempFile.write(configContent)
|
||||
config.read(tempPath)
|
||||
if 'speech' not in config:
|
||||
config['speech'] = {}
|
||||
config['speech']['driver'] = 'speechdDriver'
|
||||
config['speech']['module'] = moduleName
|
||||
config['speech']['voice'] = voiceName
|
||||
config['speech']['enabled'] = 'True'
|
||||
config['speech']['rate'] = '0.25'
|
||||
config['speech']['pitch'] = '0.5'
|
||||
config['speech']['volume'] = '1.0'
|
||||
with open(tempPath, 'w') as configFile:
|
||||
config.write(configFile)
|
||||
result = run_command(['cp', tempPath, configPath], True, privilegeCmd)
|
||||
os.unlink(tempPath)
|
||||
return result is not None
|
||||
except Exception as e:
|
||||
print(f"Error updating configuration: {e}")
|
||||
if 'tempPath' in locals():
|
||||
try:
|
||||
os.unlink(tempPath)
|
||||
except:
|
||||
pass
|
||||
return False
|
||||
|
||||
def main():
|
||||
configPath = '/etc/fenrirscreenreader/settings/settings.conf'
|
||||
isRoot, privilegeCmd = check_root_access()
|
||||
if not isRoot and not privilegeCmd:
|
||||
print("Error: This script needs root privileges. Please run with sudo or doas.")
|
||||
return
|
||||
configExists = run_command(['test', '-f', configPath], True, privilegeCmd) is not None
|
||||
if not configExists:
|
||||
print(f"Configuration file not found: {configPath}")
|
||||
return
|
||||
while True:
|
||||
print("\nFenrir Speech-Dispatcher Configuration")
|
||||
print("=====================================")
|
||||
print("\nGetting available speech-dispatcher modules...")
|
||||
moduleList = get_speechd_modules(privilegeCmd)
|
||||
if not moduleList:
|
||||
print("No speech-dispatcher modules found!")
|
||||
return
|
||||
print("\nAvailable modules:")
|
||||
for i, moduleName in enumerate(moduleList, 1):
|
||||
print(f"{i}. {moduleName}")
|
||||
while True:
|
||||
try:
|
||||
moduleChoice = int(input("\nSelect module number (or 0 to exit): "))
|
||||
if moduleChoice == 0:
|
||||
return
|
||||
if 1 <= moduleChoice <= len(moduleList):
|
||||
selectedModule = moduleList[moduleChoice - 1]
|
||||
break
|
||||
print("Invalid selection!")
|
||||
except ValueError:
|
||||
print("Please enter a number!")
|
||||
print(f"\nGetting voices for {selectedModule}...")
|
||||
voiceList = get_module_voices(selectedModule, privilegeCmd)
|
||||
if not voiceList:
|
||||
print(f"No voices found for module {selectedModule}")
|
||||
continue
|
||||
selectedVoice = display_paginated_voices(voiceList)
|
||||
if not selectedVoice:
|
||||
continue
|
||||
print("\nTesting selected voice configuration...")
|
||||
if test_voice_config(selectedModule, selectedVoice, privilegeCmd):
|
||||
print("\nVoice test successful!")
|
||||
if update_fenrir_config(configPath, selectedModule, selectedVoice, privilegeCmd):
|
||||
print("\nConfiguration updated successfully!")
|
||||
print("Please restart Fenrir for changes to take effect")
|
||||
return
|
||||
else:
|
||||
print("Failed to update configuration!")
|
||||
else:
|
||||
print("\nVoice test failed or timed out. Please try again.")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Loading…
Reference in New Issue
Block a user