1#!/usr/bin/env python3
2"""
3Set the title of an iTerm 2 tab.
4"""
5
6import sys
7
8import iterm2
9
10
11def set_title(title: str):
12    async def main(connection: iterm2.connection.Connection) -> None:
13        app = await iterm2.async_get_app(connection)
14
15        window = app.current_window
16        if window is None:
17            return
18
19        tab = window.current_tab
20        if tab is None:
21            return
22
23        await tab.async_set_title(title)
24
25    iterm2.run_until_complete(main)
26
27
28if __name__ == "__main__":
29    try:
30        title = sys.argv[1]
31    except IndexError:
32        sys.exit(f"Usage: {__file__} TITLE")
33
34    set_title(title)