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
|
import subprocess
from argparse import Namespace
from caelestia.utils.paths import c_cache_dir
class Command:
args: Namespace
def __init__(self, args: Namespace) -> None:
self.args = args
def run(self) -> None:
if self.args.show:
# Print the ipc
self.print_ipc()
elif self.args.log:
# Print the log
self.print_log()
elif self.args.kill:
# Kill the shell
self.shell("kill")
elif self.args.message:
# Send a message
self.message(*self.args.message)
else:
# Start the shell
args = ["qs", "-c", "caelestia", "-n"]
if self.args.log_rules:
args.extend(["--log-rules", self.args.log_rules])
if self.args.daemon:
args.append("-d")
subprocess.run(args)
else:
shell = subprocess.Popen(args, stdout=subprocess.PIPE, universal_newlines=True)
for line in shell.stdout:
if self.filter_log(line):
print(line, end="")
def shell(self, *args: list[str]) -> str:
return subprocess.check_output(["qs", "-c", "caelestia", *args], text=True)
def filter_log(self, line: str) -> bool:
return f"Cannot open: file://{c_cache_dir}/imagecache/" not in line
def print_ipc(self) -> None:
print(self.shell("ipc", "show"), end="")
def print_log(self) -> None:
if self.args.log_rules:
log = self.shell("log", "-r", self.args.log_rules)
else:
log = self.shell("log")
# FIXME: remove when logging rules are added/warning is removed
for line in log.splitlines():
if self.filter_log(line):
print(line)
def message(self, *args: list[str]) -> None:
print(self.shell("ipc", "call", *args), end="")
|
