from arlo import Arlo

from datetime import timedelta, date
import datetime
import sys, os
sys.path.append('..')
#import json

USERNAME = 'user@example.com'
PASSWORD = 'supersecretpassword'

videopath = 'videos'

try:
    # Instantiating the Arlo object automatically calls Login(), which returns an oAuth token that gets cached.
    # Subsequent successful calls to login will update the oAuth token.
    arlo = Arlo(USERNAME, PASSWORD)
    # At this point you're logged into Arlo.

    today = (date.today() - timedelta(days=0)).strftime("%Y%m%d")
    seven_days_ago = (date.today() - timedelta(days=7)).strftime("%Y%m%d")

    # Get all of the recordings for a date range.
    library = arlo.GetLibrary(seven_days_ago, today)

    # Check if videos folder already exists
    if not os.path.exists(videopath):
        os.makedirs(videopath)

    # Iterate through the recordings in the library.
    for recording in library:
        # Get video as a chunked stream; this function returns a generator.
        stream = arlo.StreamRecording(recording['presignedContentUrl'])
        videofilename = datetime.datetime.fromtimestamp(
            int(recording['name']) // 1000).strftime(
                '%Y-%m-%d %H-%M-%S') + ' ' + recording['uniqueId'] + '.mp4'
        with open(videopath + '/' + videofilename, 'wb') as f:
            for chunk in stream:
                f.write(chunk)
            f.close()

        print('Downloaded video ' + videofilename + ' from ' +
              recording['createdDate'] + '.')

        # Use the following line to print all the data we got for the recording.
        #print(json.dumps(recording, indent = 4))

    # Delete all of the videos you just downloaded from the Arlo library.
    # Notice that you can pass the "library" object we got back from the GetLibrary() call.
    #result = arlo.BatchDeleteRecordings(library)

    # If we made it here without an exception, then the videos were successfully deleted.
    #print ('Batch deletion of videos completed successfully.')

    arlo.Logout()
    print('Logged out')

except Exception as e:
    print(e)