#!/usr/bin/python
"""
 Lightweight Motion Detection using python picamera libraries.
 Requires a Raspberry Pi computer with a picamera module.
 This code is based on a raspberry pi forum post by user utpalc
 modified by Claude Pageau for this working example.

 This project can be used for further development
 and is located on GitHub at
 https://github.com/pageauc/picamera-motion

 For a full featured program see my GitHub pi-timolo project at
 https://github.com/pageauc/pi-timolo
"""

import os, sys
import datetime
import time
import glob
import picamera
import picamera.array
import dropbox
from dropbox.exceptions import ApiError, AuthError
import smtplib
from email.mime.text import MIMEText as text


if not os.path.exists('settings.py'):
    print("ERROR : File Not Found - settings.py")
    print("        Cannot import program variables.")
    print("        To Repair Run menubox.sh UPGRADE menu pick.")
    exit(1)
try:
    from settings import *
except ImportError:
    print("ERROR : Could Not Import settings.py")
    exit(1)

PROG_VER = "ver 2.7"
SCRIPT_PATH = os.path.abspath(__file__)
# get script file name without extension
PROG_NAME = SCRIPT_PATH[SCRIPT_PATH.rfind("/")+1:SCRIPT_PATH.rfind(".")]
SCRIPT_DIR = SCRIPT_PATH[0:SCRIPT_PATH.rfind("/")+1] # get script directory
# conversion from stream coordinate to full image coordinate
X_MO_CONV = imageWidth/float(streamWidth)
Y_MO_CONV = imageHeight/float(streamHeight)
# Authorisation token
TOKEN = 'pZnyCEeK9mUAAAAAAAAAicsXxLNKWo1y1Q2CULKvnhWfPMYgvRVQ5BJv3jYNV4ck'

#------------------------------------------------------------------------------
def get_now():
    """ Get datetime and return formatted string"""
    right_now = datetime.datetime.now()
    return ("%04d%02d%02d-%02d:%02d:%02d"
            % (right_now.year, right_now.month, right_now.day,
               right_now.hour, right_now.minute, right_now.second))

#------------------------------------------------------------------------------
def check_image_dir(image_dir):
    """ if image_dir does not exist create the folder """
    if not os.path.isdir(image_dir):
        if verbose:
            print("INFO  : Creating Image Storage folder %s" % (image_dir))
        try:
            os.makedirs(image_dir)
        except OSError as err:
            print("ERROR : Could Not Create Folder %s %s" % (image_dir, err))
            exit(1)

#------------------------------------------------------------------------------
def get_file_name(image_dir, image_name_prefix, current_count):
    """
    Create a file name based on settings.py variables
    Note image numbering will not be saved but will be inferred from the
    last image name using get_last_counter() function.
    If the last image file name is not a number sequence file
    then numbering will start from imageNumStart variable and may overwrite
    previous number sequence images. This can happen if you switch between
    number sequence and datetime sequence in the same folder.
    or
    Set imageNumOn=False to save images in datetime format to
    ensure image name is unique and avoid overwriting previous image(s).

    """
    if imageNumOn:
        # you could also use os.path.join to construct image path file_path
        file_path = image_dir+ "/"+image_name_prefix+str(current_count)+".jpg"
    else:
        right_now = datetime.datetime.now()
        file_path = ("%s/%s%04d%02d%02d-%02d%02d%02d.jpg"
                     % (image_dir, image_name_prefix,
                        right_now.year, right_now.month, right_now.day,
                        right_now.hour, right_now.minute, right_now.second))
    return file_path

#------------------------------------------------------------------------------
def get_last_counter():
    """
    glob imagePath for last saved jpg file. Try to extract image counter from
    file name and convert to integer.  If it fails restart number sequence.

    Note: If the last saved jpg file name is not in number sequence name
    format (example was in date time naming format) then previous number
    sequence images will be overwritten.

    Avoid switching back and forth between datetime and number sequences
    per imageNumOn variable in settings.py
    """
    counter = imageNumStart
    if imageNumOn:
        image_ext = ".jpg"
        search_str = imagePath + "/*" + image_ext
        file_prefix_len = len(imagePath + imageNamePrefix)+1
        try:
           # Scan image folder for most recent jpg file
           # and try to extract most recent number counter from file name
            newest = max(glob.iglob(search_str), key=os.path.getctime)
            count_str = newest[file_prefix_len:newest.find(image_ext)]
            print("%s INFO  : Last Saved Image is %s Try to Convert %s"
                  % (get_now(), newest, count_str))
            counter = int(count_str)+1
            print("%s INFO  : Next Image Counter is %i"
                  % (get_now(), counter))
        except:
            print("%s WARN  : Restart Numbering at %i "
                  "WARNING: Previous Files May be Over Written."
                  % (get_now(), counter))
    return counter

#------------------------------------------------------------------------------
def take_day_image(image_path):
    """
    Take a picamera day image. Note: You may need to increase
    sleep for low light conditions
    """
    with picamera.PiCamera() as camera:
        camera.resolution = (imageWidth, imageHeight)
        # camera.rotation = cameraRotate
        # Note use imageVFlip and imageHFlip settings.py variables
        if imagePreview:
            camera.start_preview()
        camera.vflip = imageVFlip
        camera.hflip = imageHFlip
        camera.exposure_mode = 'auto'
        camera.awb_mode = 'auto'
        time.sleep(1)
        camera.capture(image_path)
        camera.close()
    return image_path

#------------------------------------------------------------------------------
def get_stream_array():
    """ Take a stream image and return the image data array"""
    with picamera.PiCamera() as camera:
        camera.resolution = (streamWidth, streamHeight)
        with picamera.array.PiRGBArray(camera) as stream:
            camera.vflip = imageVFlip
            camera.hflip = imageHFlip
            camera.exposure_mode = 'auto'
            camera.awb_mode = 'auto'
            camera.capture(stream, format='rgb')
            camera.close()
            return stream.array

#------------------------------------------------------------------------------
def scan_motion():
    """ Loop until motion is detected """
    data1 = get_stream_array()
    while True:
        data2 = get_stream_array()
        diff_count = 0
        for y in range(0, streamHeight):
            for x in range(0, streamWidth):
                # get pixel differences. Conversion to int
                # is required to avoid unsigned short overflow.
                diff = abs(int(data1[y][x][1]) - int(data2[y][x][1]))
                if  diff > threshold:
                    diff_count += 1
                    if diff_count > sensitivity:
                        # x,y is a very rough motion position
                        return x, y
        data1 = data2

#------------------------------------------------------------------------------

def send_mail():
    from_addr = 'raspberrypimodec@gmail.com'
    to_addr_list = 'deniturkovic@gmail.com'
    cc_addr_list = 'annamariamihel@gmail.com'
    subject = 'Detekcija pokreta'
    message = text("Otkrivena detekcija. Za pregled snimljenih fotografija posjeti http://10.42.0.145:8080/")
    login = 'raspberrypimodec'
    password = 'raspberry2018'

    server = smtplib.SMTP('smtp.gmail.com', 587)
    server.starttls()
    server.login(login,password)
    server.sendmail(from_addr, to_addr_list, message.as_string())
    server.quit()


#------------------------------------------------------------------------------
# Upload localfile to Dropbox
def uploadFile(localfile):

    # Check that access tocken added
    if (len(TOKEN) == 0):
        sys.exit("ERROR: Missing access token. "
                 "try re-generating an access token from the app console at dropbox.com.")

    # Create instance of a Dropbox class, which can make requests to API
    print("Creating a Dropbox object...")
    dbx = dropbox.Dropbox(TOKEN)

    # Check that the access token is valid
    try:
        dbx.users_get_current_account()
    except AuthError as err:
        sys.exit("ERROR: Invalid access token; try re-generating an "
                 "access token from the app console at dropbox.com.")

    # Specify upload path
    uploadPath = "/" + localfile.split('/')[-1]

    # Read in file and upload
    with open(localfile, 'rb') as f:
        print("Uploading " + localfile + " to Dropbox as " + uploadPath + "...")

        try:
            dbx.files_upload(f.read(), uploadPath)
        except ApiError as err:
            # Check user has enough Dropbox space quota
            if (err.error.is_path() and
                    err.error.get_path().error.is_insufficient_space()):
                sys.exit("ERROR: Cannot upload; insufficient space.")
            elif err.user_message_text:
                print(err.user_message_text)
                sys.exit()
            else:
                print(err)
                sys.exit()
#------------------------------------------------------------------------------
def do_motion_detection():
    """
    Loop until motion found then take an image,
    and continue motion detection. ctrl-c to exit
    """
    current_count = get_last_counter()
    if not imageNumOn:
        print("%s INFO  : File Naming by Date Time Sequence" % get_now())
    while True:
        x_pos, y_pos = scan_motion()
        file_name = get_file_name(imagePath, imageNamePrefix, current_count)
        take_day_image(file_name)
        if imageNumOn:
            current_count += 1
        # Convert xy movement location for full size image
        mo_x = x_pos * X_MO_CONV
        mo_y = y_pos * Y_MO_CONV
        if verbose:



            print("%s INFO  : Motion xy(%d,%d) Saved %s (%ix%i)"
                  % (get_now(), mo_x, mo_y, file_name,
                     imageWidth, imageHeight,))
            file = os.getcwd() + '/' + file_name
            print("11111111111111111111")
            print(file)
            uploadFile(file)
            deleteLocal(file)
            send_mail()

#------------------------------------------------------------------------------
# Delete file
def deleteLocal(file):
    os.system("rm " + file)
    print("File: " + file + " deleted ...")



# Start Main Program Logic
if __name__ == '__main__':
    print("%s %s  written by Claude Pageau" % (PROG_NAME, PROG_VER))
    print("---------------------------------------------")
    check_image_dir(imagePath)
    print("%s INFO  : Scan for Motion "
          "threshold=%i (diff)  sensitivity=%i (num px's)..."
          % (get_now(), threshold, sensitivity))
    if not verbose:
        print("%s WARN  : Messages turned off per settings.py verbose = %s"
              % (get_now(), verbose))
    try:
        do_motion_detection()

    except KeyboardInterrupt:
        print("")
        print("INFO  : User Pressed ctrl-c")
        print("        Exiting %s %s " % (PROG_NAME, PROG_VER))