SinsModCheck/convert.py at master · StargateInvasion/SinsModCheck · 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
#The MIT License (MIT)
#Copyright (c) 2017 Stargate Invasion

import os
import glob
import sys
import hashlib
import subprocess
import errno
from shutil import copyfile


basegame = None
inputdir = None
output = None
fileformat = 'txt'

def mkdir_p(path):
    try:
        os.makedirs(path)
    except OSError as exc:  # Python >2.5
        if exc.errno == errno.EEXIST and os.path.isdir(path):
            pass
        else:
            raise

def getType(filename):
    ext = os.path.splitext(filename)
    return ext[1].replace(".", "").lower()

def checkBin(inputpath, directory):
    # This was designed to convert the txt files to bin in an effort to identify ones that might have an issue
    # It creates a tmp bin file and then deletes it.  If it gets stuck or throws an error - investigate checked file.
    path = os.path.join(inputpath, directory)
    osname = os.name
    #print(glob.glob(os.path.join(path, '*')))
    for filename in glob.glob(os.path.join(path, '*')):
        gamefile = os.path.basename(filename)
        datatype = getType(gamefile)
        outfile = os.path.join(output, directory, gamefile)
        if not os.path.isdir(os.path.join(output, directory)):
            os.mkdir(os.path.join(output, directory))
        if datatype == "entity" or datatype == "brushes" or datatype == "mesh" or datatype == "particle":
            print "converting " + gamefile
            if osname == "posix":
                subprocess.call(['wine', os.path.join(basegame, 'ConvertData_Rebellion.exe'), datatype, filename, outfile, fileformat])
            else:
                subprocess.call([os.path.join(basegame, 'ConvertData_Rebellion.exe'), datatype, filename, outfile, fileformat])
        else:
            print "copying " + gamefile
            try:
                copyfile(filename, outfile)
            except:
                pass

if len(sys.argv) > 1:
    i = 1
    while i < len(sys.argv):
        if sys.argv[i] == "-o" and len(sys.argv) > i:
            output = sys.argv[i+1]
        elif sys.argv[i] == "-i" and len(sys.argv) > i:
            inputdir = sys.argv[i+1]
        elif sys.argv[i] == "bin":
            fileformat = 'bin'
        i += 1
    if not inputdir and not output and len(sys.argv) > 1:
        inputdir = sys.argv[1]
        output = sys.argv[2]
    if not basegame:
        if os.path.isdir("C:\Program Files (x86)\Steam\steamapps\common\Sins of a Solar Empire Rebellion"):
            basegame = "C:\Program Files (x86)\Steam\steamapps\common\Sins of a Solar Empire Rebellion"
        elif os.path.isdir("C:\Program Files\Steam\steamapps\common\Sins of a Solar Empire Rebellion"):
            basegame = "C:\Program Files\Steam\steamapps\common\Sins of a Solar Empire Rebellion"
        elif os.path.isdir(os.path.expanduser("~/.wine/drive_c/Program Files (x86)/Steam/steamapps/common/Sins of a Solar Empire Rebellion")):
            basegame = os.path.expanduser("~/.wine/drive_c/Program Files (x86)/Steam/steamapps/common/Sins of a Solar Empire Rebellion")
    if not inputdir:
        inputdir = basegame

    if not os.path.isdir(os.path.expanduser(output)):
        mkdir_p(output)
        print "Creating dir: " + output

    print "Reading directory: " + inputdir
    for filename in glob.glob(os.path.join(inputdir, '*')):
        if os.path.isdir(filename):
            if not filename.endswith("Galaxy Forge") and not filename.endswith("Particle Forge") and not filename.endswith("font") and not filename.endswith("ConvertXSI") and not filename.endswith("_CommonRedist"):
                gamedir = os.path.basename(filename)
                checkBin(inputdir, gamedir)
        else:
            if not filename.endswith(".exe") and not filename.endswith(".dll") and not filename.endswith(".pdf"):
                gamefile = os.path.basename(filename)
                print "copying " + gamefile
                outfile = os.path.join(output, gamefile)
                copyfile(filename, outfile)
else:
    print "Usage: convert.py -i <inputdirectory> -o <outputdirectory> [txt|bin]"
    print "\tInput Directory will default to Sins basegame path if -i not specified"
    print "\tConversion type will default to txt, meaning bin->txt.  bin will convert txt->bin."
    sys.exit()