new_rfid_reader_allaccess.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from os.path import exists
import wiringpi as wiringpi2
import numpy as np
import time
import sys
from datetime import datetime
from datetime import timedelta
import csv
from time import sleep
import subprocess
import Device #this is needed for communication with servo controller
import threading #this is a built in library needed to run csv_writer and motor activation codes simultaneously

#dir_setup("/home/quailrfid/")

servo_board = Device.BoardDevice(verbose=True) # setting which board to communicate with

warn = 0
header = ['year', 'month', 'day', 'timestamp', 'PIT_tag_ID']
rfid_data = "/home/quailrfid/Data_quailforage/RFID"
#set GPIO pin
GPIO_PIN = 1

motor_data = "/home/quailrfid/Data_motor/motor"

file_extension = '.csv'

def csv_writer(data_path, datestring, header, value):
        try:
                if data_path: #if path to rfid_data exists, do this:
                        filename = datestring + '.csv'
                        full_path = data_path + "/" + filename
                        if exists(full_path):
                                file = open(full_path, 'a+')
                                tmp_writer = csv.writer(file)
                                tmp_writer.writerow(value)
                                file.close()
                        else:
                                file = open(full_path, 'w+')
                                tmp_writer = csv.writer(file)
                                tmp_writer.writerow(header)
                                tmp_writer.writerow(value)
                                file.close()
        except Exception as E:
                pass #pass does nothing


def csv_writer_motor(data_path, datestring, header, value):
        try:
                if data_path: #if path to rfid_data exists, do this:
                        filename = 'motor_' + datestring + '.csv'
                        full_path = data_path + "/" + filename
                        if exists(full_path):
                                file = open(full_path, 'a+')
                                tmp_writer = csv.writer(file)
                                tmp_writer.writerow(value)
                                file.close()
                        else:
                                file = open(full_path, 'w+')
                                tmp_writer = csv.writer(file)
                                tmp_writer.writerow(header)
                                tmp_writer.writerow(value)
                                file.close()
        except Exception as E:
                pass #pass does nothing


def activate_motor(tagid):
        try:
                time_now = datetime.now()
                global time_detected #tell python you are referring to the global time_detected variable
                        
                if "time_detected" not in globals(): #if time_detected has no value, set it to 61 seconds ago. This is needed when restarting script 
                                
                        time_detected = time_now + timedelta(seconds=-301)
                        #print('good')
                        #print(time_detected)
                        
                if time_now >= (time_detected + timedelta(seconds=300)): #if current time is at least 60 seconds after previous detection, activate motor
                                
                        time_detected = time_now #if condition is met, set new time_detected
                        #print(time_detected)
                        servo_board.device.set_target(5, 1300)
                        time.sleep(1) #length of time for motor to move before rotating back to starting position
                        servo_board.device.set_target(5, 2550)
                        
                        global motor_activation
                        motor_activation = 1
                                
        except Exception as E:
                pass 


#wait for input from pin
def WaitForCTS():
    while wiringpi2.digitalRead(GPIO_PIN):
        time.sleep(0.001)
    return

#initiate and open communication with RFID reader
def RFIDSetup():
    response = wiringpi2.wiringPiSetup()
    wiringpi2.pinMode(GPIO_PIN, 0)
    fd = wiringpi2.serialOpen('/dev/serial0', 9600)
    wiringpi2.serialFlush(fd)
    if response != 0 and fd <= 0:
        print ("Unable to Setup communications")
        sys.exit()
    return fd

#set reader mode to EM4102 compatible
def SetReaderMode(fd):
    WaitForCTS()
    wiringpi2.serialPutchar(fd, 0x76)
    wiringpi2.serialPutchar(fd, 0x03) #EM/MC2000

#set polling delay to 20 ms
def SetPollingDelay(fd):
    WaitForCTS()
    wiringpi2.serialPutchar(fd, 0x50)
    wiringpi2.serialPutchar(fd, 0x60) #262 ms

#read text from RFID reader
def ReadText(fd):
    response = ""
    i = 1
    while i <= 5:
        temp = format(wiringpi2.serialGetchar(fd), 'x').upper()
        if len(temp) == 1:
            temp = '0' + temp
        if i <= 4:
            temp = temp + "-"
        response = response + temp
        i = i + 1
    return response

#read int from RFID reader
def ReadInt(fd):
    qtydata = wiringpi2.serialDataAvail(fd)
    response = 0
    if qtydata > 0:
        response = wiringpi2.serialGetchar(fd)
    return response

def ReadTagPageZero(fd):
    try:
        while True:
            WaitForCTS()
            wiringpi2.serialPutchar(fd, 0x52)
            wiringpi2.serialPutchar(fd, 0x00)
            ans = ReadInt(fd)
            time.sleep(0.1)
            if ans == int("0xD6", 16):
                ans = ReadText(fd)
                dt = datetime.now()
                tag = ans
                if __name__ == '__main__':
                    
                    fname = f"{dt.year}_{dt.month}_{dt.day}.csv"
                    f_path = rfid_data + "/" + fname
                    if exists(f_path):
                        pass
                        #time_detected = datetime.now() + timedelta(seconds=-61)
                    else:
                        global time_detected #tell python you are defining a global time_detected variable
                        time_detected = datetime.now() + timedelta(seconds=-61) #if csv is not present, set time_detected to 61 seconds ago
                    
                    global motor_activation #using a global variable to control recording of motor activations bc csv_writer_motor did not work inside of activate_motor
                    motor_activation = 0
                    
                    t1 = threading.Thread(target=csv_writer, args=(rfid_data, f"{dt.year}_{dt.month}_{dt.day}", header, [f"{dt.year}", f"{dt.month}", f"{dt.day}", f"{dt:%H:%M:%S.%f}", ans]))
                    t2 = threading.Thread(target=activate_motor, args=(ans,))

                    t1.start()
                    t2.start()
                    
                    t1.join() # .join makes main code wait for thread to finish
                    t2.join() #motor does not finish one cycle unless I have this here
                    
                    if motor_activation == 1:
                        csv_writer_motor(motor_data, f"{dt.year}_{dt.month}_{dt.day}", header, [f"{dt.year}", f"{dt.month}", f"{dt.day}", f"{dt:%H:%M:%S.%f}", ans])
                        motor_activation = 0
                    
                time.sleep(1)
    except Exception as E:
        pass

comms = RFIDSetup()
SetReaderMode(comms)
SetPollingDelay(comms)
ReadTagPageZero(comms)

#subprocess.run(["/home/quailrfid/maestro-set-target.sh", "/dev/ttyACM0 5 10000"], shell=True)