openbox-pipemenus/openbox-pipemenu-steam.py

99 lines
3.1 KiB
Python
Executable File

#!/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))