You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.8 KiB
55 lines
1.8 KiB
#!/usr/bin/env python3 |
|
#pylint: disable=C0111 |
|
# config example (cmd_pre and cmd_post are optional): |
|
# --- |
|
# name: test |
|
# cmd_pre: |
|
# - [/usr/bin/notify_send, 'I am precmd1!'] |
|
# - [/usr/bin/notify_send, 'I am precmd2!'] |
|
# multicmd: |
|
# selection1: |
|
# - [/usr/bin/notify-send, 'I am command1 for selection1!'] |
|
# - [/usr/bin/notify-send, 'I am command2 for selection1!'] |
|
# selection2: |
|
# - [/usr/bin/notify-send, 'I am command1 for selection2!'] |
|
# cmd_post: |
|
# - [/usr/bin/notify_send, 'I am postcmd1!'] |
|
# - [/usr/bin/notify_send, 'I am postcmd2!'] |
|
# --- |
|
import subprocess |
|
import sys |
|
import yaml |
|
|
|
def run_cmd(cmd, stdin=subprocess.PIPE, data=None): |
|
proc = subprocess.Popen( |
|
cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=stdin |
|
) |
|
stdout, stderr = (i.strip() for i in proc.communicate(data)) |
|
if proc.returncode: |
|
err_msg = ['ERROR (' + str(proc.returncode) + '): ' + ' '.join(cmd)] |
|
if stdout: |
|
err_msg.append('out:\n' + stdout.decode('UTF-8')) |
|
if stderr: |
|
err_msg.append('err:\n' + stderr.decode('UTF-8')) |
|
sys.stderr.write('\n'.join(err_msg) + '\n') |
|
return stdout |
|
|
|
def main(): |
|
config_path = sys.argv[1] |
|
with open(config_path, mode='r') as config: |
|
conf = yaml.safe_load(config.read()) |
|
dmenu_cmd = ['/usr/bin/dmenu', '-p', conf.get('name', 'N/A')] + sys.argv[2:] |
|
dmenu_opt = '\n'.join(conf['multicmd']).encode() |
|
selection = run_cmd(dmenu_cmd, data=dmenu_opt).decode('UTF-8') |
|
#selection = sys.argv[2] |
|
if not selection in conf['multicmd']: |
|
sys.exit(1) |
|
cmds =\ |
|
conf.get('cmd_pre', list()) +\ |
|
conf['multicmd'][selection] +\ |
|
conf.get('cmd_post', list()) |
|
for cmd in cmds: |
|
run_cmd(cmd) |
|
|
|
if __name__ == '__main__': |
|
main()
|
|
|