Displays the battery level of a SteelSeries Arctis 7 headset.

arctis7.py

#!/usr/bin/env python3

"""
This script displays the battery level of a SteelSeries Acrtis 7 headset.


USING
-----

To use this script you must install hidapi (https://github.com/trezor/cython-hidapi):

    pip3 install hidapi

and then running it using Python 3:

    python3 ./arctis7.py


ON LINUX
--------

On Linux, you will have to run this script as root or to add new udev rules.
To add the udev rules, just create the following file:

    /etc/udev/rules.d/99-steelseries-arctis7.rules

with the following content:

    SUBSYSTEM=="hidraw", ATTRS{idVendor}=="1038", ATTRS{idProduct}=="12ad", MODE="0666"
    SUBSYSTEM=="usb", ATTRS{idVendor}=="1038", ATTRS{idProduct}=="12ad", MODE="0666"

Finally run the following command to update udev rules:

    sudo udevadm trigger


LICENSE
-------

        DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
                    Version 2, December 2004

 Copyright (C) 2004 Sam Hocevar <[email protected]>

 Everyone is permitted to copy and distribute verbatim or modified
 copies of this license document, and changing it is allowed as long
 as the name is changed.

            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
   TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION

  0. You just DO WHAT THE FUCK YOU WANT TO.

"""  # noqa


import hid


VENDOR_ID = 0x1038
PRODUCT_ID = 0x12ad
ENDPOINT = 5


def open_device(vendor_id, product_id, endpoint):
    """Opens and returns the HID device

    .. NOTE::

       Cherry-picked from https://github.com/flozz/rivalcfg/

    :param int vendor_id: The vendor id of the device (e.g. ``0x1038``)).
    :param int product_id: The product id of the device (e.g. ``0x1710``).
    :param int endpoint: The number of the endpoint to open on the device (e.g.
                         ``0``).

    :raise DeviceNotFound: The requested device is not plugged to the computer
                           or it does not provide the requested endpoint.
    :raise IOError: The device or its interface cannot be opened (permission
                    issue, device busy,...).

    :rtype: hid.device
    """
    path = None
    device = hid.device()

    # Search the device
    for interface in hid.enumerate(vendor_id, product_id):
        if interface["interface_number"] == endpoint:
            path = interface["path"]
            break

    # Open the found device. This can raise an IOError.
    if path:
        device.open_path(path)
        return device

    # No matching device found
    raise Exception("Requested device or endpoint not found: %04x:%04x:%02x" % (  # noqa
        vendor_id, product_id, endpoint))


def get_status():
    status = {
        "transmitter_connected": False,
        "headset_connected": False,
        "headset_battery": 0,
        }

    try:
        device = open_device(VENDOR_ID, PRODUCT_ID, ENDPOINT)
    except Exception:
        return status

    status["transmitter_connected"] = True

    # Is headset powered on?
    device.write(b"\x06\x14")
    data = device.read(31)
    if data[2] == 0x03:
        status["headset_connected"] = True

    # Get battery level
    device.write(b"\x06\x18")
    data = device.read(31)
    status["headset_battery"] = data[2]

    device.close()

    return status


if __name__ == "__main__":
    status = get_status()
    if not status["transmitter_connected"]:
        print("The transmitter is not connected or cannot be openend")
    elif not status["headset_connected"]:
        print("The headset is powered off")
    else:
        print("Headset is powered on")

    battery = status["headset_battery"]
    print("Battery [%-10s] %02i%%" % (
        "=" * round(battery / 10),
        battery
        ))

Do you maybe know how to add custom devices or track their battery? I have custom built keyboard that also have battery and 2.4GHz transmitter, and app on Windows that shows that ugly 3 bar battery charge like steelseries gg do.

Sorry I cannot help for your keyboard. Each device is different and use its own protocol. For SS mice and the Arctis 7 I had to do reverse engineering to find how they work. :)

Yeah, I have read this https://aarol.dev/posts/arctis-hid/, so I assume that my keyboard should work in almost the same way, but the problem is I don't know how to write program that will work, that one from this tut wasn't working for me lol