FlatSat_student.py

"""
The Python code you will write for this module should read
acceleration data from the IMU. When a reading comes in that surpasses
an acceleration threshold (indicating a shake), your Pi should pause,
trigger the camera to take a picture, then save the image with a
descriptive filename. You may use GitHub to upload your images automatically,
but for this activity it is not required.

The provided functions are only for reference, you do not need to use them. 
You will need to complete the take_photo() function and configure the VARIABLES section
"""

#AUTHOR: Ardian Agoes, Raahat Amin, Naveed Razzaque , Tahmid Azmir, Belal Hasan
#DATE: 12/7/2024

#import libraries
import time
import board
from adafruit_lsm6ds.lsm6dsox import LSM6DSOX as LSM6DS
from adafruit_lis3mdl import LIS3MDL
from git import Repo
from picamera2 import Picamera2

#VARIABLES
THRESHOLD = 15    #Any desired value from the accelerometer
REPO_PATH = "/home/amier/Documents/FlatSatChallenge" #Your github repo path: ex. /home/pi/FlatSatChallenge
FOLDER_PATH = "Images"   #Your image folder path in your GitHub repo: ex. /Images

#imu and camera initialization
i2c = board.I2C()
accel_gyro = LSM6DS(i2c)
mag = LIS3MDL(i2c)
picam2 = Picamera2()


def git_push():
    """
    This function is complete. Stages, commits, and pushes new images to your GitHub repo.
    """
    try:
        repo = Repo(REPO_PATH)
        origin = repo.remote('origin')
        print('added remote')
        origin.pull()
        print('pulled changes')
        repo.git.add(REPO_PATH + FOLDER_PATH)
        repo.index.commit('New Photo')
        print('made the commit')
        origin.push()
        print('pushed changes')
    except Exception as e:
        print(f"Couldn't upload to git: {e}")


def img_gen(name):
    """
    This function is complete. Generates a new image name.

    Parameters:
        name (str): your name ex. MasonM
    """
    t = time.strftime("_%H%M%S")
    imgname = (f'{REPO_PATH}/{FOLDER_PATH}/{name}{t}.jpg')
    return imgname


def take_photo():
    """
    This function is NOT complete. Takes a photo when the FlatSat is shaken.
    Replace psuedocode with your own code.
    """
    while True:
        accelx, accely, accelz = accel_gyro.acceleration
        acceleration_magnitude = (accelx**2 + accely**2 + accelz**2)**0.5
        if acceleration_magnitude > THRESHOLD:
            print("Threshold Met")
            print("Picture in 3")
            time.sleep(1)
            print("Picture in 2")
            time.sleep(1)
            print("Picture in 1")
            time.sleep(1)            
            picam2.start()
            picam2.capture_file(img_gen("ArdianA"))
            picam2.start()
            print("Picture in 1")
            git_push()
        time.sleep(1)
        print("Camera Ready")

        #PAUSE


def main():
    take_photo()


if __name__ == '__main__':
    main()