Merge pull request #4 from Jordan-Fielding/Development

Update main.py

main.py

@@ -2,56 +2,84 @@ import RPi.GPIO as GPIO                 #Used to Import the LED
 import time                             #Used to allow wait times
 from picamera import PiCamera           #Used to import the Camera
 from gpiozero import MotionSensor       #Used to import the MotionSensor
+from datetime import datetime
+from subprocess import call
 import os
 
 pir = MotionSensor(4)
 camera = PiCamera()                     #Camera Initialization
 GPIO.setwarnings(False)
 GPIO.setmode(GPIO.BCM)                  #Setting the GPIO Mode
-GPIO.setup(22, GPIO.OUT)                #LED output pin
+GPIO.setup(22, GPIO.OUT)                #LED Flash Output
+GPIO.setup(27, GPIO.OUT)                #LED indicator output
 
 cwd = os.getcwd()                       #Sets the Current Working Directory
 
 
-print("\nProgram Running!")
-
-#To stabilize sensor and Camera
-time.sleep(2)
-while True:
-
-    #Waits for Motion from PIR Sensor
-    pir.wait_for_motion()
-
-    print("\nBug detected")
-
-    #Turns on the LED Flash
-    GPIO.output(22, 1)
-
-    #Sets Img path and filename, Saves to Working directory of Script
-    file_name = cwd + "/Pictures/Capture_" + str(time.time()) + ".jpg"
-    print("\nFile Name is: " + file_name)
 
 
-    #Used to wait for 0.1 Seconds for Camera to be ready
+def bugmotion():
-    time.sleep(0.1)
+    GPIO.output(22, 0) #Sets light off
 
-    #Saves File
+    print("\nProgram Running!")
-    camera.capture(file_name)
+    for i in range(3):
+        time.sleep(1)
+        GPIO.output(27, 1)
+        time.sleep(1)
+        GPIO.output(27, 0)
+        if i == 3:
+            break
 
-    #Sets PIR Sensor back to waiting for motion
-    pir.wait_for_no_motion()
 
-    #Used to turn off the LED after 0.2 Secs
+    #To stabilize sensor and Camera
-    time.sleep(0.2)
-
-    #Used to turn off the LED
-    GPIO.output(22, 0)
-
-    #Used to stabilize Sensor and allow the Camera to refocus
     time.sleep(2)
-    print("\nSystem Ready!")
+    while True:
-
+
-
+        #Waits for Motion from PIR Sensor
+        pir.wait_for_motion()
+
+        print("\nBug detected")
+
+        #Turns on the LED Flash
+        GPIO.output(22, 1)
+
+        #Sets Img path and filename, Saves to Working directory of Script
+        # Grab the current time
+        filePath = cwd + "/Pictures/"
+        currentTime = datetime.now()
+        picTime = currentTime.strftime("%Y-%m-%d-%H-%M-%S")
+        picName = "Capture-" + picTime + '.jpg'
+        completeFilePath = filePath + picName
+        #file_name = cwd + "/Pictures/Capture_" + str(time.time()) + ".jpg"
+        print("\nFile Name is: " + completeFilePath)
+
+
+        #Used to wait for 0.1 Seconds for Camera to be ready
+        time.sleep(0.1)
+
+        #Saves File
+        camera.capture(completeFilePath)
+
+        #Sets PIR Sensor back to waiting for motion
+        print("\nWaiting for no Motion!")
+        pir.wait_for_no_motion()
+
+        #Used to turn off the LED after 0.2 Secs
+        time.sleep(0.2)
+
+        #Used to turn off the LED
+        GPIO.output(22, 0)
+
+        #Used to stabilize Sensor and allow the Camera to refocus
+        time.sleep(2)
+        print("\nSystem Ready!")
+
+def exit_handler():
+    print('Program Shutting Down!')
+    GPIO.output(22, 0) # turns off LED
+
+bugmotion() #Starts Program
+atexit.register(exit_handler)