waybar: add wakatime custom module

This commit is contained in:
thomasabishop 2024-02-09 16:04:58 +00:00
parent 2bee33c407
commit 0122b9c321
3 changed files with 295 additions and 154 deletions

60
scripts/waybar/wakatime.py Executable file
View file

@ -0,0 +1,60 @@
#! /usr/local/bin/python3
# Retrieve status bar data from WakaTime API and present in Waybar widget
import requests
import os
import json
import textwrap
WAKATIME_API_KEY = os.getenv("WAKATIME_API_KEY")
WAKATIME_ENDPOINT = "https://wakatime.com/api/v1/users/current/status_bar/today"
def get_data(url):
response = requests.get(url)
if response.status_code == 200:
return response.json()
else:
raise Exception(
f"Failed to fetch data from API. Status code: {response.status_code}"
)
def generate_tooltip(time, languages, projects):
return textwrap.dedent(
f"""\
Time coding: {time}
Languages: {languages}
Projects: {projects}"""
)
def format_metric(metrics):
return ", ".join(
[f'{metric["name"]} ({metric["percent"]}%)' for metric in metrics[:3]]
)
def main():
output = {}
try:
data = get_data(WAKATIME_ENDPOINT + "?api_key=" + WAKATIME_API_KEY)
digital_time = data["data"]["grand_total"]["digital"]
human_time = data["data"]["grand_total"]["text"]
langs = data["data"]["languages"]
projects = data["data"]["projects"]
tooltip = generate_tooltip(
human_time, format_metric(langs), format_metric(projects)
)
output["text"] = digital_time
output["tooltip"] = tooltip
except Exception as e:
output["text"] = "Error"
print(json.dumps(output))
if __name__ == "__main__":
main()

View file

@ -1,11 +1,24 @@
{ {
"layer": "top", // Waybar at top layer "layer": "top",
"height": 30, // Waybar height (to be removed for auto height) "height": 30,
"spacing": 4, // Gaps between modules (4px) "spacing": 4,
// Choose the order of the modules "modules-left": [
"modules-left": ["custom/os", "hyprland/workspaces"], "hyprland/workspaces"
"modules-right": ["custom/spotify", "network", "bluetooth", "cpu",
"memory", "disk", "temperature", "keyboard-state", "battery", "clock" ], ],
"modules-right": [
"custom/spotify",
"network",
"bluetooth",
"cpu",
"memory",
"disk",
"temperature",
"backlight",
"custom/wakatime",
"battery",
"clock"
],
"keyboard-state": { "keyboard-state": {
"numlock": true, "numlock": true,
"capslock": true, "capslock": true,
@ -49,11 +62,9 @@
} }
}, },
"tray": { "tray": {
// "icon-size": 21,
"spacing": 10 "spacing": 10
}, },
"clock": { "clock": {
// "timezone": "America/New_York",
"format": "{: %H:%M  %d/%m/%Y}", "format": "{: %H:%M  %d/%m/%Y}",
"tooltip-format": "<big>{:%Y %B}</big>\n<tt><small>{calendar}</small></tt>", "tooltip-format": "<big>{:%Y %B}</big>\n<tt><small>{calendar}</small></tt>",
"format-alt": "{:%Y-%m-%d}" "format-alt": "{:%Y-%m-%d}"
@ -66,21 +77,31 @@
"format": " {}%" "format": " {}%"
}, },
"temperature": { "temperature": {
// "thermal-zone": 2,
// "hwmon-path": "/sys/class/hwmon/hwmon2/temp1_input",
"critical-threshold": 80, "critical-threshold": 80,
// "format-critical": "{temperatureC}°C {icon}",
"format": "{icon} {temperatureC}°C", "format": "{icon} {temperatureC}°C",
"format-icons": ["", "", ""] "format-icons": [
"",
"",
""
]
}, },
"backlight": { "backlight": {
"device": "acpi_video1", "device": "acpi_video1",
"format": "{percent}% {icon}", "format": "{percent}% {icon}",
"format-icons": ["", "", "", "", "", "", "", "", ""] "format-icons": [
"",
"",
"",
"",
"",
"",
"",
"",
""
]
}, },
"battery": { "battery": {
"states": { "states": {
// "good": 95,
"warning": 30, "warning": 30,
"critical": 15 "critical": 15
}, },
@ -88,15 +109,18 @@
"format-charging": " {capacity}%", "format-charging": " {capacity}%",
"format-plugged": " {capacity}%", "format-plugged": " {capacity}%",
"format-alt": "{time} {icon}", "format-alt": "{time} {icon}",
// "format-good": "", // An empty format will hide the module "format-icons": [
// "format-full": "", "",
"format-icons": ["", "", "", "", ""] "",
"",
"",
""
]
}, },
"battery#bat2": { "battery#bat2": {
"bat": "BAT2" "bat": "BAT2"
}, },
"network": { "network": {
// "interface": "wlp2*", // (Optional) To force the use of this interface
"format-wifi": " {essid} ({signalStrength}%)", "format-wifi": " {essid} ({signalStrength}%)",
"format-ethernet": "{ipaddr}/{cidr} ", "format-ethernet": "{ipaddr}/{cidr} ",
"tooltip-format": "{ifname} via {gwaddr} ", "tooltip-format": "{ifname} via {gwaddr} ",
@ -105,7 +129,6 @@
"format-alt": "{ifname}: {ipaddr}/{cidr}" "format-alt": "{ifname}: {ipaddr}/{cidr}"
}, },
"pulseaudio": { "pulseaudio": {
// "scroll-step": 1, // %, can be a float
"format": "{icon} {volume}%", "format": "{icon} {volume}%",
"format-bluetooth": "{icon} {volume}% {format_source}", "format-bluetooth": "{icon} {volume}% {format_source}",
"format-bluetooth-muted": " {icon} {format_source}", "format-bluetooth-muted": " {icon} {format_source}",
@ -115,14 +138,32 @@
"format-icons": { "format-icons": {
"headphone": "", "headphone": "",
"hands-free": "", "hands-free": "",
"headset": "",
"phone": "", "phone": "",
"portable": "", "portable": "",
"car": "", "car": "",
"default": ["", "", ""] "default": [
"",
"",
""
]
}, },
"on-click": "pavucontrol" "on-click": "pavucontrol"
}, },
"disk": {
"interval": 30,
"format": " {percentage_used}%",
"path": "/home"
},
"bluetooth": {
"controller": "bluetoothctl",
"format": " {status}",
"format-connected": " {device_alias}",
"format-connected-battery": " {device_alias} {device_battery_percentage}% ",
"tooltip-format": "{controller_alias}\t{controller_address}\n\n{num_connections} connected",
"tooltip-format-connected": "{controller_alias}\t{controller_address}\n\n{num_connections} connected\n\n{device_enumerate}",
"tooltip-format-enumerate-connected": "{device_alias}\t{device_address}",
"tooltip-format-enumerate-connected-battery": "{device_alias}\t{device_address}\t{device_battery_percentage}%"
},
"custom/media": { "custom/media": {
"format": "{icon} {}", "format": "{icon} {}",
"return-type": "json", "return-type": "json",
@ -132,19 +173,11 @@
"default": "🎜" "default": "🎜"
}, },
"escape": true, "escape": true,
"exec": "$HOME/.config/waybar/mediaplayer.py 2> /dev/null" // Script in resources folder "exec": "$HOME/.config/waybar/mediaplayer.py 2> /dev/null"
// "exec": "$HOME/.config/waybar/mediaplayer.py --player spotify 2> /dev/null" // Filter player based on name
}, },
"custom/os": { "custom/os": {
"format": " archbish", "format": " archbish"
}, },
"disk": {
"interval": 30,
"format": " {percentage_used}%",
"path": "/home",
},
"custom/spotify": { "custom/spotify": {
"exec": "/usr/bin/python3 $HOME/.config/waybar/resources/custom_modules/mediaplayer.py --player spotify", "exec": "/usr/bin/python3 $HOME/.config/waybar/resources/custom_modules/mediaplayer.py --player spotify",
"format": "{} ", "format": "{} ",
@ -153,19 +186,10 @@
"on-scroll-up": "playerctl next", "on-scroll-up": "playerctl next",
"on-scroll-down": "playerctl previous" "on-scroll-down": "playerctl previous"
}, },
"custom/wakatime": {
"bluetooth": { "exec": "source $HOME/dotfiles/.env && python3 $HOME/.config/waybar/resources/custom_modules/wakatime.py",
"controller": "bluetoothctl", // specify the alias of the controller if there are more than 1 on the system "format": " {}",
"format": " {status}", "return-type": "json",
"format-connected": " {device_alias}", "interval": 600
"format-connected-battery": " {device_alias} {device_battery_percentage}% ",
// "format-device-preference": [ "device1", "device2" ], // preference list deciding the displayed device
"tooltip-format": "{controller_alias}\t{controller_address}\n\n{num_connections} connected",
"tooltip-format-connected": "{controller_alias}\t{controller_address}\n\n{num_connections} connected\n\n{device_enumerate}",
"tooltip-format-enumerate-connected": "{device_alias}\t{device_address}",
"tooltip-format-enumerate-connected-battery": "{device_alias}\t{device_address}\t{device_battery_percentage}%"
} }
} }

View file

@ -0,0 +1,57 @@
#! /usr/local/bin/python3
import requests
import os
import json
import textwrap
WAKATIME_API_KEY = os.getenv("WAKATIME_API_KEY")
WAKATIME_ENDPOINT = "https://wakatime.com/api/v1/users/current/status_bar/today"
def get_data(url):
response = requests.get(url)
if response.status_code == 200:
return response.json()
else:
raise Exception(
f"Failed to fetch data from API. Status code: {response.status_code}"
)
def generate_tooltip(time, languages, projects):
return textwrap.dedent(
f"""\
Time coding: {time}
Languages: {languages}
Projects: {projects}"""
)
def format_metric(metrics):
return ", ".join(
[f'{metric["name"]} ({metric["percent"]}%)' for metric in metrics[:3]]
)
def main():
output = {}
try:
data = get_data(WAKATIME_ENDPOINT + "?api_key=" + WAKATIME_API_KEY)
digital_time = data["data"]["grand_total"]["digital"]
human_time = data["data"]["grand_total"]["text"]
langs = data["data"]["languages"]
projects = data["data"]["projects"]
tooltip = generate_tooltip(
human_time, format_metric(langs), format_metric(projects)
)
output["text"] = digital_time
output["tooltip"] = tooltip
except Exception as e:
output["text"] = "Error"
print(json.dumps(output))
if __name__ == "__main__":
main()