get_youtube_view.py

"""
Created on Thu Apr 27 16:28:36 2017
@author: barnabysandeford
"""
# Currently works for Safari, but just change to whichever
# browser you're using.

import time

# Added pafy to get video length for the user
import pafy

# Changed the method of opening the browser.
# Selenium allows for the page to be refreshed.
from selenium import webdriver

# adding ability to change number of repeats
count = int(input("Number of times to be repeated: "))
# Same as before
url = input("Enter the URL : ")

refreshrate = None

# tries to get video length using pafy
try:
    video = pafy.new(url)
    if hasattr(video, "length"):
        refreshrate = video.length
# if pafy fails to work, prints out error and asks for video length from the user
except Exception as e:
    print(e)
    print("Length of video:")
    minutes = int(input("Minutes "))
    seconds = int(input("Seconds "))
    # Calculating the refreshrate from the user input
    refreshrate = minutes * 60 + seconds

# Selecting Safari as the browser
driver = webdriver.Safari()

if url.startswith("https://"):
    driver.get(url)
else:
    driver.get("https://" + url)

for i in range(count):
    # Sets the page to refresh at the refreshrate.
    time.sleep(refreshrate)
    driver.refresh()