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
98
99
100
101
102
|
import asyncio
import argparse
import html
from ferdium_dbus import Client
async def toggle_window(client, args):
"""Toggle window visibility"""
await client.toggle_window()
async def toggle_mute(client, args):
"""Toggle mute status"""
await client.toggle_mute()
async def unread(client, args):
"""Get unread messages count"""
def callback():
"""Print unread count(s)"""
# For each service
counts = {}
for service in client.unread_services:
name, direct, indirect = service
safe_name = html.escape(name)
# If it's exactly the service we're looking for, just return the count
if safe_name == args.services:
count = direct
if not args.direct:
count += indirect
print(count)
return
# If the service in included in the services we're looking for
if args.services in ("total", "all") or safe_name in args.services:
counts[safe_name] = direct
if not args.direct:
counts[safe_name] += indirect
# Get total notifications
if args.services == "total":
print(sum(counts.values()))
return
# Finally, print each service notifications on a different line
print(
"\n".join(
f"{name}: {count}"
for name, count in counts.items()
)
)
# Do print counts and keep running if tail mode enabled
callback()
if args.tail:
client.on_change(callback)
await asyncio.get_running_loop().create_future()
async def main():
"""Main cli interface"""
# Define commands
commands = {
"unread": unread,
"toggle-mute": toggle_mute,
"toggle-window": toggle_window,
}
# Arguments parser
argparser = argparse.ArgumentParser(description="Script to interact with Ferdium on your bar")
subparsers = argparser.add_subparsers(dest="command", required=True)
# Unread command
argparser_unread = subparsers.add_parser("unread", help=unread.__doc__)
argparser_unread.add_argument("-s", "--services", default="total", help="Which services to get notifications from {total, all, <name>} (the name can be a comma-separated list)")
argparser_unread.add_argument("-d", "--direct", action="store_true", default=False, help="Get only direct (mentions or DM) messages")
argparser_unread.add_argument("-t", "--tail", action="store_true", default=False, help="Keep running and print on change")
# Toggle mute and toggle window commands
argparser_toggle_mute = subparsers.add_parser("toggle-mute", help=toggle_mute.__doc__)
argparser_toggle_window = subparsers.add_parser("toggle-window", help=toggle_window.__doc__)
# Get args
args = argparser.parse_args()
# Initialise ferdium client
client = Client()
await client.connect()
if not client.running:
print("not running")
return
# Execute command
await commands[args.command](client, args)
if __name__ == "__main__":
asyncio.run(main())
|
