'''
updated Mar 23  to add auto install from chrome driver
Renamed from WebandImagetoVide0.py

'''
'''
##  20 Oct  19 ,working version - modified version of  "WebToImageToVideo2.py"   modded to do html and images
###  Opens websites, copys screenshot to Image file.  Open Image File adds QRcode to all snips, creates video from images,
####  plays video every x mins
### Seleniium need correct version of ChromeDriver for chrome browser, download and replace in correct file
## chromedriver/
'''

import qrcode
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import time
from time import sleep
from datetime import datetime
#import ffmpeg
import csv
import cv2
import numpy as np
import glob
import chromedriver_autoinstaller

chromedriver_autoinstaller.install()  # Check if the current version of chromedriver exists
                                      # and if it doesn't exist, download it automatically,
                                      # then add chromedriver to path

driver = webdriver.Chrome()



options = Options()
options.headless = True
options.add_argument("window-size=1670,1000")
#options.add_argument("headless") 

UrlRun=0
ImageRun=0
       

def Url_Convert(UrlRun):
    #if (int(min_last))==0 and UrlRun==0:
    if UrlRun==0:
        UrlRun=1
        print (' start UrlConvert ....')
        #print ('run number  ' + str(count))
        driver = webdriver.Chrome(options=options)#needs correct version
        for m in range (1):
            print ('m is  ' + str(m))

            now = datetime.now()

            DirStr="./snip"
            QRStr="./QRimage"
            

            #******************************
            #  start here with loop for urls
            contents = []
            with open('urlfile.csv', 'r') as csvf: # Open file in read mode
                urls = csv.reader(csvf, delimiter='\t') #  delimiter='\t' ensure correct clsoure evenon exception
                for url in urls:
                    contents.append(url) # Add each url to list contents

            print (contents)
            if len(contents)<1:
                print(' no urls contained in file urlfile.csv')
                
            for i in range (len(contents)):
                try:
                    turl = str(contents[i])
                    turl= (turl[:-2])
                    turl=(turl[2:])
                    print(turl)
                    time.sleep(1)
                    driver.get(turl)
                    print ('got   ' + turl)
                    driver.save_screenshot((DirStr+str(i)+".png"))    # DirStr="C:\Python36\carousel\snip"
                    sleep(1)
                    print ('saved as....', (DirStr+str(i)+".png"))

                except:
                    turl = str(contents[i])
                    turl= (turl[:-2])
                    turl=(turl[2:])
                    print(turl)
                    print ('Failed to access ' + turl)
                    
                
        print ('exited m loop')
        driver.quit()
        
    sleep(1)
    print ('point 1')


def Add_QrCodes():
            # loop through all images, for each image create and add QRCode and create video
    # find and loop
    print (' start QrCode ....')
    DirStr="./snip"
    QRStr="./QRimage"
    ImgArray=[]
                
    for filename in glob.glob(DirStr +"*.png"):
        ImgArray.append(filename)
        print ('Array len is ....', len(ImgArray))
        #print (ImgArray)
    
    for i in range (len(ImgArray)):
        f_string=(ImgArray[i])
        print (f_string)
        img = cv2.imread(ImgArray[i])
        heightstr = str(np.size(img, 0))
        widthstr = str(np.size(img, 1))
        
    #*******   Create QR Code   **********
        # get time
        dt_string = now.strftime("%d/%m/%Y %H:%M")
        print("date and time =", dt_string)
        size_string=(heightstr + "," + widthstr)

           # Create qr code instance
        qr = qrcode.QRCode(
            version = 4,   # 1 is 21x21, 2 is 25x25, 4 is 33x33 @ ERRCC L gives 114 char, @ M 90 char
            error_correction = qrcode.constants.ERROR_CORRECT_L, 
            box_size = 10,
            border = 4,
        )

        # The data that you want to store
        data = f_string+ ","+ size_string + "," + dt_string  # ";" +turl +
        print ('QR String is......', data)
        qr.add_data(data)
        qr.make(fit=False)  #  make tru to autofit

        # Create an image from the QR Code instance
        qrimg = qr.make_image()

        # Save it somewhere, change the extension as needed:
        qrimg.save("TEMP.png")             #   

    #************************************
    #*********   combine images *********
        QRimg=cv2.imread("TEMP.png")
        QRimg=cv2.resize(QRimg,(250,250))

        tempimg = cv2.imread(ImgArray[i])
        #tempimg = cv2.resize(tempimg,(1650,775))
        h1, w1 = tempimg.shape[:2]
        h2, w2 = QRimg.shape[:2]

        #create empty matrix
        CombinedImage = np.zeros((max(h1, h2), w1+w2,3), np.uint8)

        #combine 2 images
        CombinedImage[:h1, :w1,:3] = tempimg
        CombinedImage[:h2, w1:w1+w2,:3] = QRimg
        SaveString = (QRStr+str(i)+".png")
        print ('SaveString is .....' + SaveString)

        cv2.imwrite ( SaveString, CombinedImage)  #
        sleep(1)
        #print (i)
        
                
        # create video file from images
def Create_Video():
    multip = 1
    img_array = []

    width=1920
    height=1000

    Filestr='./QRimage'

    out = cv2.VideoWriter('project2.avi',cv2.VideoWriter_fourcc(*'DIVX'), 1, (width,height), True)
    #for i in range(1,6):

    ImgArray=[]
                
    for filename in glob.glob(Filestr +"*.png"):
        ImgArray.append(filename)
        print ('Video Array len is ....', len(ImgArray))
    
    for i in range (len(ImgArray)): 
        stri=str(i)
        print ('File is ....', (Filestr + stri +'.png'))
        img= cv2.imread(Filestr + stri +'.png')
        img= cv2.resize(img,(width,height))

        print (Filestr + stri +'.png')
        for m in range(multip):
            out.write(img)
        if i == 1:
            out.write(img)   # extra frames to allow capture on start
            out.write(img)
    out.release()
        
    # ************************************************

    
    #************ play video evry 5 mins  **************
def Play_Video():
    print ('entering video ')
    #if (int(min_last))==5:
    #if int(min_last) % 2 > 0:
    cv2.namedWindow('frame', cv2.WINDOW_NORMAL)
    cv2.setWindowProperty('frame', cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN)

    #cv2.namedWindow('frame',cv2.WINDOW_NORMAL) # this works
    #cv2.resizeWindow('frame', width,height)  # this works
    cap = cv2.VideoCapture('project2.avi')
    while True:
        ret, frame = cap.read()
        
        if not ret:
            break
        
        cv2.imshow('frame',frame)
        if cv2.waitKey(25) & 0xFF == ord('q'):
             break
        sleep(1)

    cap.release()
    cv2.destroyAllWindows()

    
###  main loop ##
while(True):
    now = datetime.now()
    min_str = now.strftime("%M")
    print (min_str)

    min_last=min_str[1]
    print ('min last is ' + min_last)

    if int(min_last) % 2==0:  #  even
    #if int(min_last)==0:
        UrlRun=0
        ImageRun=0
        Url_Convert(UrlRun)
        Add_QrCodes()
        Create_Video()

    if int(min_last) % 2 > 0: # odd
    #if (int(min_last))==5:
        Play_Video()

    sleep(200)