Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def main(args):
- # process commandline args
- processArgs(args)
- # read config files
- loadGameKnowledge()
- conn = psycopg2.connect(database='TestServer', user='testuser', password='r4inbows', host='jgabe037.ath.cx', port='5432')
- cur = conn.cursor()
- files = [f for f in listdir("C:/Users/liado_000/Documents/mwo-screenshot-analyzer/input") if isfile(join("C:/Users/liado_000/Documents/mwo-screenshot-analyzer/input", f))]
- files.sort()
- files.reverse()
- print(files)
- if len(files)%2 != 0:
- error("Odd number of files found: %d" % len(files))
- return
- count = len(files)/2
- while len(files) != 0:
- (result, mech, status, score, kills, assists, damage, cbills, xp, psr) = (None,)*10
- file2 = files.pop()
- file1 = files.pop()
- log(file1 + " " + file2)
- match = re.search(CONFIG.fnpattern, file1)
- if not match:
- error("Screenshot filename pattern: No match")
- continue
- datetime = time.strptime(match.group(), CONFIG.dtformat)
- file1 = "C:/Users/liado_000/Documents/mwo-screenshot-analyzer/input/" + file1
- file2 = "C:/Users/liado_000/Documents/mwo-screenshot-analyzer/input/" + file2
- id = time.strftime("%Y-%m-%d %H-%M-%S", datetime)
- filedate = time.strftime("%Y-%m-%d %H:%M:%S", datetime)
- with Image.open(file1) as img:
- # load resolution information
- debug(repr(img.size))
- loadResolutionInfo(img.size)
- img = preprocess(img, id, "team")
- (dist, map) = getmap(img, id, file1)
- debug("Found Map: \"%s\" with distance=%s" % (map, repr(dist)))
- (dist, mode) = getmode(img, id, file1)
- debug("Found Mode: \"%s\" with distance=%s" % (mode, repr(dist)))
- mytime = gettime(img, id, file1)
- (result, mech, status, score, kills, assists, damage) = getplayerdata(img, id, file1)
- with Image.open(file2) as img:
- psr = getpsr(img, id, file2)
- img = preprocess(img, id, "player")
- cbills = getcbills(img, id, file2)
- xp = getxp(img, id, file2)
- debug("%s,%s,%s,%s,%s,%d,%d,%d,%d,%d,%d,%d,%s,%s" % (filedate, result, mech, map, mode, status, score, kills, assists, damage, xp, cbills, psr, mytime))
- shutil.move(file1, "C:/Users/liado_000/Documents/mwo-screenshot-analyzer/processed")
- shutil.move(file2, "C:/Users/liado_000/Documents/mwo-screenshot-analyzer/processed")
- with open("C:/Users/liado_000/Documents/mwo-screenshot-analyzer/data.csv", "a") as myfile:
- myfile.write("%s,%s,%s,%s,%s,%d,%d,%d,%d,%d,%d,%d,%s,%s\n" % (filedate, result, mech, map, mode, status, score, kills, assists, damage, xp, cbills, psr, mytime))
- cur.execute(
- """INSERT INTO mwotestdb (filedate, result, mech, map, mode, status, score, kills, assists, damage, xp, cbills, psr, mytime)
- VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s);""",
- (filedate, result, mech, map, mode, status, score, kills, assists, damage, xp, cbills, psr, mytime))
- print
- conn.commit()
- # Close communication with the database
- cur.close()
- conn.close()
- log("Finished")
- return
- if __name__ == "__main__":
- main(sys.argv[1:])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement