1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
#!/usr/bin/python
# -*- coding: utf-8 -*-
# Copyright (C) René 'Necoro' Neumann
#
# Some helper functions for managing workspaces in i3.
#
import sys
from os.path import realpath, dirname, join, basename
cwd = realpath(dirname(__file__))
sys.path.insert(1, join(cwd, "libs"))
import i3
import sh
from functools import partial
DEFAULT = "switch"
def nag(msg, **kwargs):
sh.i3_nagbar(m = "%s: %s" % (basename(sys.argv[0]), msg), **kwargs)
switch = i3.workspace
move = partial(i3.move, "container to workspace")
def new_ws(cmd, args):
"""Create a new workspace by using the first free number > 0."""
nums = (w["num"] for w in i3.get_workspaces())
nums = filter(lambda n: n is not None and n >= 0, nums)
try:
exe = args[args.index("--exec")+1]
except (IndexError, ValueError):
exe = None
i = -1 # fallback if `nums` happens to be empty
for i,n in enumerate(sorted(nums)):
if i != n:
cmd(str(i))
break
else:
cmd(str(i+1))
if exe:
# We use i3.exec_ here instead of sh.Command, as we do not want the
# exe to be a child of this script's process
# Also we get startup notification support for free :-)
if sh.which(exe): # i3 exec always yields 'success'
i3.exec_(exe)
else:
nag("Command '%s' not found!" % exe)
def to_ws(cmd, prompt, args):
"""Use `dmenu` to switch or move to a workspace."""
ws = sorted(w["name"] for w in i3.get_workspaces())
try:
sel = sh.dmenu(b = True, p = prompt, _in = "\n".join(ws)).strip()
except sh.ErrorReturnCode:
sel = None
if sel:
cmd(sel)
def rename(args):
cur_ws = i3.filter(i3.get_workspaces(), focused = True)[0]
input = sh.i3_input.bake(P = "Rename workspace: ")
if "--keep-num" in args and cur_ws["num"]:
input(F = ('rename workspace to "%d: %%s"' % cur_ws["num"]))
else:
input(F = 'rename workspace to "%s"')
if __name__ == "__main__":
try:
arg = sys.argv[1]
args = sys.argv[1:]
except IndexError:
arg = DEFAULT
args = []
if arg == "switch":
to_ws(switch, "Switch to:", args)
elif arg == "move":
to_ws(move, "Move to:", args)
elif arg == "new":
new_ws(switch, args)
elif arg == "move_new":
new_ws(move, args)
elif arg == "rename":
rename(args)
else:
nag("Unknown arg: %s" % arg)
sys.exit(1)
|
