# -*- coding: utf-8 -*-
"""
Created on Mon Dec 05 12:56:15 2016

@author: klaas
"""
import os #filenames
import time
from operator import truediv


folder = 'Masters' #put all the replay files here
Handle = 'IsHuman: True' #Search for this in the replay file

filenames = os.listdir(folder)
os.chdir( folder )
allies = ('germany', 'england','france')
soviets = ('russia', 'ukraine')
countries = allies + soviets
wins = [0,0,0,0,0]
loss = [0,0,0,0,0]

f2 = open('Output.txt', 'w')
f2.close()

for c, country in enumerate(countries) :
    a,b = 0,0
    for filename in filenames:
        f1 = open(filename,"rb")
        f2 = open('Output.txt', 'a') #output file
        lines = f1.readlines()
        for i, line in enumerate(lines):
            line = line.lstrip()
            if line.startswith(Handle):
                print "player found"
                #f2.write("player found")
                if country in lines[i + 3]:
                    if 'Won' in lines[i + 9]:
                        a += 1
                    else : b += 1
        #break
        print c
        f2.close()
        f1.close()
    print countries
    print c
    wins[c] = a
    loss[c] = b
    #f2 = open('Output.txt', 'a') #output file
    #f2.write(country+' - W/L :'+str(a)+'/'+str(b)+'\n')

print (", ".join(map(str, wins)))
f2 = open('Output.txt', 'a')
ratio = map(truediv, wins, loss)
ratio = [round(elem,2) for elem in ratio]


camps = ('allies','soviets')
wins_camp = [wins[0]+wins[1]+wins[2],wins[3]+wins[4]]
loss_camp = [loss[0]+loss[1]+loss[2],loss[3]+loss[4]]
ratio_camp = map(truediv, wins_camp, loss_camp)
ratio_camp = [round(elem,2) for elem in ratio_camp]

f2.write('COUNTRIES : %s' % (countries,)+'\n')
f2.write('Wins : '+", ".join(map(str, wins))+'\n')
f2.write('Losses : '+", ".join(map(str, loss))+'\n')
f2.write('W/L Ratio : '+", ".join(map(str, ratio))+'\n')
f2.write('CAMPS : %s' % (camps,)+'\n')
f2.write('Wins : '+", ".join(map(str, wins_camp))+'\n')
f2.write('Losses : '+", ".join(map(str, loss_camp))+'\n')
f2.write('W/L Ratio : '+", ".join(map(str, ratio_camp))+'\n')
f2.close()

    #time.sleep(5)

    #f2.write(lines[i - 1])
#f2.write(lines[i + 7])

#f2.write(lines[i + 3])
#f2.write(lines[i + 9])