PySteamTimes/importSteam.py at main · NathanM0153/PySteamTimes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import time
import steamfront
import sys
#credit: https://github.com/4Kaylum/Steamfront
from openpyxl import Workbook


steamID64 = input("Enter your Steam ID64 key. This is the number at the end of your profile URL.\n")
steamAPIKey = input("Enter your Steam API Key. After registering, you can find yours at this link: \nhttps://steamcommunity.com/dev/apikey\n")
freebool = input("Include free games? Y/N\n")
steamClient = steamfront.Client(steamAPIKey)
waitTime = 1.3 #rate limit on steam api requests


def appIDList(games):
    IDList = []
    try:
        for i in games:
            IDList.append(i.appid)
    except:
        print("Internet connection required.")
        sys.exit()
    return IDList


def gameNameList(IDList):
    games = []
    errors = []
    count = 1

    for i in IDList:
        try:
            game = steamClient.getApp(appid=i)
            name = game.name
            if freebool == "Y" or freebool == "y":
                print(count, "  ", name)
                count += 1
                games.append(name)
            else:
                if not game.is_free: #exclude free games
                    print(count, "  ", name)
                    count += 1
                    games.append(name)
        except:
            print("ID not found. App ID is", i)
            errors.append(i)
            continue
        time.sleep(waitTime)
    if len(errors) > 0:
        print("Game IDs not found:")
        for i in errors:
            i = int(i)
            print(i, end="\n")

        print("You can find them at https://steamcommunity.com/app/######, inserting the relevant ID.")
        print("Most of these will be game accessories or discontinued games, but there is a possibility the script missed a game so check if you would like.")
    else:
        print("All games found successfully.")
    return games

def exportToExcel(gameList):
    file = "SteamGames.xlsx"
    wb = Workbook()
    wb.save(file)
    gameList.sort()
    gameSheet = wb.active
    for i in range(0,len(gameList)):
        gameSheet.cell(row=i+1, column=1, value=gameList[i])
    wb.save(file)
    print("File successfully saved.")
    wb.close()


def doctorOutput(gameList):
    fixedList = []
    string = ""
    bracketbool = False
    for i in gameList:
        i = i.replace("™","")
        i = i.replace("®","")
        i = i.replace("–", "-") #en dash
        i = i.replace("—", "-") #em dash
        i = i.replace(" - ", " ")
        i = i.replace("’", "'")

        string = ""
        for j in i: #removes everything in parentheses
            if j == '(':
                bracketbool = True
                #turns off adding to string
            elif j == ')':
                bracketbool = False
                #turns on adding to string
            elif not bracketbool:
                string += j
        i = string

        if i[-1] == " ":
            i = i[:-1] #cuts off extra space if there is one

        fixedList.append(i)
    return fixedList


class importSteam():

    def main():
        user = steamClient.getUser(id64=steamID64)
        games = user.apps #list of <steamfront.userapp.UserApp object at 0x00000XXXXXXXXX>
        IDList = appIDList(games)
        gameList = gameNameList(IDList)
        doctoredList = doctorOutput(gameList)
        exportToExcel(doctoredList)
        return 0

    if __name__ == "__main__":
        main()