openbox-pipemenu-steam.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
#  openbox-pipemenu-steam.py
#
#  Copyright 2017 kirbylife <https://github.com/ImNotRoot>
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#  MA 02110-1301, USA.
#
#

import os

try: # Python 2
    from commands import getoutput, getstatusoutput
except: # Python 3
    from subprocess import getoutput, getstatusoutput

STEAM_URL = os.environ['HOME'] + "/.steam/steam/steamapps"
BANNED_NAMES = ("Proton", "Steamworks", "Linux Runtime")


def set_item(name, appid=None):
    if appid:
        print("<item label='" + name +
              "'><action name='Execute'><command>steam steam://rungameid/" +
              appid + "</command></action></item>")
    else:
        print("<item label='" + name + "'></item>")


def main(args):
    ### Header ###
    print("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")
    print("<openbox_pipe_menu>")
    ###        ###

    if getstatusoutput("cd " + STEAM_URL)[0] == 1:
        set_item("Steam not installed")
    else:
        folders = [STEAM_URL]
        try:
            # Search if you have games in other folders
            with open(STEAM_URL + "/libraryfolders.vdf") as extras:
                for line in extras:
                    if "/" in line:
                        folder = line.split("\t")[-1].strip()[1:-1]
                        folder = folder + "/steamapps"
                        folders.append(folder)
        except Exception:
            pass

        games = []
        for folder in folders:
            library = getoutput("ls " + folder + " | grep acf").split("\n")
            games.extend([folder + "/" + game for game in library])

        if not games:
            set_item("0 games installed")
        else:
            for game in games:
                appid = ""
                name = ""
                manifest = open(game, "r")
                data = manifest.read().split("\n")
                manifest.close()
                for f in data:
                    if '"appid"' in f:
                        appid = f.split('"')[3]
                    elif '"name"' in f:
                        name = f.split('"')[3]
                        is_valid = any(attr in name for attr in BANNED_NAMES)
                        if is_valid:
                            continue
                        set_item(name, appid)

    ### Footer ###
    print("</openbox_pipe_menu>")
    ###        ###
    return 0


if __name__ == '__main__':
    import sys
    sys.exit(main(sys.argv))