from time import sleep
import os
import cv2
import time
import pytesseract
from PIL import Image
from PIL import ImageFilter
pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe' #import tesseract from file

WorkingDirectory="D:/testing" # .py file has to be in the same disk as this folder
print(os.getcwd()) # Current working directory
os.chdir(WorkingDirectory)
print(os.listdir())

start_time = time.time()

counter = 0
# Main
def FindScore(path):
    
    cam = cv2.VideoCapture(path)

    last_frame = int(cam.get(cv2.CAP_PROP_FRAME_COUNT))
    first_frame = "1"
    framerate = int(cam.get(cv2.CAP_PROP_FPS)) # ni pomembno
    

    print("Video frames: {} ".format(last_frame) )
    print("Framerate: {} fps".format(framerate))

    #  Find first frame
    cam.set(1,2-1)
    frame = cam.read()
    name = 'frame' + str(last_frame) + '.jpg'
    cv2.imwrite(name, frame)
    myimage1 = Image.open(name)

    # Find last frame
    cam.set(1,last_frame-1)
    frame = cam.read()
    name = 'picture' + str(first_frame)+ ".jpg"
    cv2.imwrite(name, frame)
    myimage2 = Image.open(name)

    # Get score text
    text1 = pytesseract.image_to_string(myimage1)
    text2 = pytesseract.image_to_string(myimage2)

    # Get Kills
    char1 = '#'
    char2 = '/'
    kills_start = int(text1[text1.find(char1)+1 : text1.find(char2)])
    kills_end = int(text2[text2.find(char1)+1 : text2.find(char2)])
    kills_diff = kills_end - kills_start

    # Get Deaths
    dth1 = text1.split('/')
    deaths_start = int((dth1[1])) 

    dth2 = text2.split('/')
    deaths_end = int((dth2[1])) 
    deaths_diff= deaths_end - deaths_start

    # Get Assists
    ass_begin = find_nth(text1, "/", 2)   
    ass_end = find_nth(text2, "/", 2)  
    ass_diff = ass_end - ass_begin



    # Rename
    num_count= str(counter).zfill(4)
    new_name = '{}---{}-{}-{}.mp4'.format(num_count, kills_diff, deaths_diff, ass_diff)
    os.rename(path, new_name)

for subdir, dirs, files in os.walk(WorkingDirectory):
    for file in files:
        
        filepath = subdir + os.sep + file

        if filepath.endswith(".mp4"):
            print (file)

            #print ("Analyzing: "+file)
            
            FindScore(filepath)
                #FindScore(file)

            counter = counter +1
            sleep(.01)







# Get Assists Function
def find_nth(haystack, needle, n):
    start = haystack.find(needle)
    while start >= 0 and n > 1:
        start = haystack.find(needle, start+len(needle))
        n -= 1

        besd1=haystack[start+1]
        besd2=haystack[start+2]
        ass1= int(besd1)
        ass2=0
        if(besd2 != " "):
            ass2= int(besd2)

        haystack = ass1*10+ ass2

    return haystack

print("Files Analyzed: "+str(counter))
print ("Program took", time.time() - start_time, "seconds to run")

from time import sleep
import os
import cv2
import time
import pytesseract
from PIL import Image
from PIL import ImageFilter
pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe' #import tesseract from file

WorkingDirectory="D:/testing" # .py file has to be in the same disk as this folder
print(os.getcwd()) # Current working directory
os.chdir(WorkingDirectory)
print(os.listdir())

start_time = time.time()

counter = 0
# Main
def FindScore(path):
    
    cam = cv2.VideoCapture(path)

    last_frame = int(cam.get(cv2.CAP_PROP_FRAME_COUNT))
    first_frame = "1"
    framerate = int(cam.get(cv2.CAP_PROP_FPS)) # ni pomembno
    

    print("Video frames: {} ".format(last_frame))
    print("Framerate: {} fps".format(framerate))

    #  Find first frame
    cam.set(1,2-1)
    ret, frame = cam.read()
    name = '/data/frame' + str(first_frame) + '.jpg'
    cv2.imwrite(name, frame)
    myimage1 = Image.open(name)

    # Find last frame
    cam.set(1,last_frame-5)
    ret, frame = cam.read()
    name = '/data/frame' + str(last_frame) + '.jpg'
    cv2.imwrite(name, frame)
    myimage2 = Image.open(name)
  
    # Get score text
    text1 = pytesseract.image_to_string(myimage1)
    text2 = pytesseract.image_to_string(myimage2)

    # Get Kills
    char1 = '#'
    char2 = '/'
    kills_start = int(text1[text1.find(char1)+1 : text1.find(char2)])
    kills_end = int(text2[text2.find(char1)+1 : text2.find(char2)])
    kills_diff = kills_end - kills_start

    # Get Deaths
    dth1 = text1.split('/')
    deaths_start = int((dth1[1])) 

    dth2 = text2.split('/')
    deaths_end = int((dth2[1])) 
    deaths_diff= deaths_end - deaths_start

    # Get Assists
    ass_begin = find_nth(text1, "/", 2)   
    ass_end = find_nth(text2, "/", 2)  
    ass_diff = ass_end - ass_begin

    
 
    # Rename
    num_count= str(counter).zfill(4)
    new_name = '{}---{}-{}-{}.mp4'.format(num_count, kills_diff, deaths_diff, ass_diff)
    os.rename(path, new_name)

    cam.release()
    cv2.destroyAllWindows()

for subdir, dirs, files in os.walk(WorkingDirectory):
    for file in files:
        
        filepath = subdir + os.sep + file

        if filepath.endswith(".mp4"):
            print ("Analyzing: "+filepath)
            try:
                FindScore(filepath)
                
            except:
                counterino= str(counter).zfill(3)
                namerino = '{}---League of Legends.mp4'.format(counterino)
                os.rename(filepath, namerino)
                
            counter = counter +1
            sleep(.01)


# Get Assists Function
def find_nth(haystack, needle, n):
    start = haystack.find(needle)
    while start >= 0 and n > 1:
        start = haystack.find(needle, start+len(needle))
        n -= 1

        besd1=haystack[start+1]
        besd2=haystack[start+2]
        ass1= int(besd1)
        ass2=0
        if(besd2 != " "):
            ass2= int(besd2)

        haystack = ass1*10+ ass2

    return haystack

print("Files Analyzed: "+str(counter))
print ("Program took", time.time() - start_time, "seconds to run")