pygate1.py

from picamera import PiCamera
import time
from queue import Queue
import sys
import io
import sqlite3
import cv2
import RPi.GPIO as gpio
from threading import Thread
from tkinter import *

import time
from threading import Thread

alpr_path = '/usr/local/lib/python2.7/dist-packages/openalpr/'
sys.path.insert(0,alpr_path)
from openalpr import Alpr


alpr = Alpr('eu', 'openalpr.conf', 'runtime_data')
alpr.set_top_n(5)
db_path = 'alprData.db'
insertStr = 'INSERT INTO alprTable (plates, time, opened, confidence) VALUES (?, ?, ?, ?)'
selStr = 'SELECT plates FROM userTable'
url = "rtsp://admin:admin@192.168.1.119:554/0"

delay = 2 #Koliko kasni slika sa kamere [sec]
pic_dt = 3 #svakih koliko uzima sliku [sec]

Pi_Cam = False
IP_Cam = True

Qpic = Qpic_nd = Qrez = Queue(-1)

rt=Tk()
rt.geometry("700x600")
rt.wm_title("baza_korisnika")

cap = cv2.VideoCapture(url)
fps = math.floor(cap.get(5))
conn = sqlite3.connect(db_path)
cur = conn.cursor()
cur.execute(selStr)
userPlates = cur.fetchall()
userPlates = [x[0] for x in userPlates]
# gpio.setmode(gpio.BOARD)
# gpio.setup(11,gpio.OUT)

with conn:
        conn.execute("create table if not exists userTable(id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT UNIQUE, naziv TEXT, plates TEXT UNIQUE)")


def OpenPiCamera():
    print('Pi camera started')
    with PiCamera() as camera:
        camera.start_preview(alpha=200)
        while True:
                pic_stream = io.BytesIO()
                time.sleep(2)
                camera.capture(pic_stream, 'jpeg')
                Qpic.put(pic_stream.getvalue())
                pic_stream.close()

        camera.stop_preview()


def OpenIPCamera():
    i_lim = fps * pic_dt
    i = 0
    try:
        frame = cap.read()[1]
        frame = cv2.cvtColor(Qf.get(), cv2.COLOR_BGR2GRAY)
        i += 1
        if i >= i_lim:
            Qpic_nd.put(frame)
            i = 0
    
    except Exception as e:
        print(e)
        
    cap.release()
    cv2.destroyAllWindows()


def ALPR_decoder():
    print('ALPR started')
    i = 0
    while True:
        if Pi_Cam:
                alprOut = alpr.recognize_array(Qpic.get())
        if IP_Cam:
                cap = cv2.VideoCapture(url)
                fps = math.floor(cap.get(5))
                skip_frames = math.floor(delay / delta_t)
                while i<skip_frames: #Odbaci prvih nekoliko frejmova zbog kašnjenja IP stream-a
                        Qpic_nd.get()
                        i += 1
                alprOut = alpr.recognize_ndarray(Qpic_nd.get())
        else:
                print('unknown data type')

        if len(alprOut['results']):
                Qrez.put({'epoch': alprOut['epoch_time'], 'results': alprOut['results']})
                print('USPJESNOOOOO')
        else:
                print('NISTAAAAAAA')


def podizac():
        gpio.output(11,1)
        time.sleep(2)
        gpio.output(11,0)


def sqlInsert():
        print('sqlinsert started')
        conn = sqlite3.connect(db_path)
        plate1 = ''
        with conn:
                while True:
                        data = Qrez.get()
                        epoch = data['epoch']
                        rezs = data['results']
                        print('epoch:', epoch)
                        t = time.strftime('%Y-%m-%d %H:%M:%S', time.gmtime(epoch/1000))
                        for rez in rezs[0]['candidates']:
                                plate = rez['plate']
                                opened = plate in userPlates

                                if plate in userPlates:# and not(plate1 == plate):
                                        opened = 1
                                        plate1 = plate
                                        Thread(target=podizac, daemon=True).run()
                                        print('OPENED OPENED OPENED')


                                confidence = rez['confidence']
                                print(plate, t, opened, confidence)
                                conn.execute(insertStr, [plate, t, opened, confidence])

                                # if opened and not(plate1==plate):
                                #         Thread(target=podizac, daemon=True).run()
                                #         print('OPENED OPENED OPENED')
                                
                        print('PLATE111111111111:',plate1)
        
if Pi_Cam:
    Thread(target=OpenPiCamera, daemon=True).start()
if IP_Cam:
    Thread(target=OpenIPCamera, daemon=True).start()
Thread(target=ALPR_decoder, daemon=True).start()
Thread(target=sqlInsert, daemon=True).start()


def add_command():
    if(len(naziv.get())!=0 and len(plates.get())!=0):
        with conn:
                conn.execute("insert into userTable (naziv, plates) values(?,?)",(naziv.get().lower(),plates.get()))
        userPlates.append(plates.get())
        lb1.delete(0,END)
        cur.execute('select id from userTable where naziv=? and plates =?', (naziv.get().lower(),plates.get()))
        id = cur.fetchone()
        lb1.insert(END,(id, naziv.get(),plates.get()))

        

def view_command():
    lb1.delete(0,END)
    cur.execute("SELECT * FROM userTable")
    [lb1.insert(END,row) for row in cur.fetchall()]
        

def findit_command():
    lb1.delete(0,END)
    cur.execute("select * from userTable where naziv like ? or plates like ?",(naziv.get().lower(),plates.get()))
    [lb1.insert(END,row) for row in cur.fetchall()]
        

def dele_command():
        if len(plates.get())!=0:
                cur.execute("delete from userTable where naziv=? and plates=?",(naziv.get().lower(),plates.get()))
                conn.commit()
                clearit_command()
                view_command()


def clearit_command():
    e1.delete(0,END)
    e2.delete(0,END)    
    

def update_command():
        if len(plates.get())!=0:
                with conn:
                        conn.execute("update userTable set naziv=?,plates=? where id=?",(naziv.get().lower(),plates.get(), lbData[0]))
                clearit_command()
                view_command()


r=2
l0=Label(rt,text="") #nazivi labela prije unosa
l0.grid(row=r-1,column=0)

l1=Label(rt,text="Naziv",font=("Courier", 10))
l1.grid(row=r,column=3)

l2=Label(rt,text="Tablice",font=("Courier", 10))
l2.grid(row=r+1,column=3)


naziv=StringVar()


e1=Entry(rt,textvariable=naziv,font=("Courier", 15))#unosi redom ime prezime tablice
e1.grid(row=r,column=4)
plates=StringVar()
e2=Entry(rt,textvariable=plates,font=("Courier", 15))
e2.grid(row=r+1,column=4)
reg=StringVar()


lb1=Listbox(rt,height=10, width=40,font=("Courier", 20))#lista sa scrollom
lb1.grid(row=10, column=0,rowspan=8,columnspan=6)
sb=Scrollbar(rt)
sb.grid(row=7,column=5,rowspan=8)

lb1.configure(yscrollcommand=sb.set)#scroll 
sb.configure(command=lb1.yview)


def onselect(evt):
        global lbData
        lb = evt.widget
        try:
                yid = lb.curselection()[0]
                lbData = lb.get(yid)
                e1.delete(0,END)
                e1.insert(END, lbData[1])
                e2.delete(0,END)
                e2.insert(END, lbData[2])
        except:
                pass


lb1.bind('<<ListboxSelect>>',onselect)#selektovanje

b1=Button(rt,text="lista korisnika",width=20, command=view_command)
b1.grid(row=0,column=0)

b2=Button(rt,text="dodaj korisnika",width=20, command=add_command)
b2.grid(row=1,column=0)

b3=Button(rt,text="obrisi korisnika",width=20, command=dele_command)
b3.grid(row=2,column=0)

b4=Button(rt,text="pronadji korisnika",width=20, command=findit_command)
b4.grid(row=3,column=0)

b5=Button(rt,text="prepravi korisnika",width=20, command=update_command)
b5.grid(row=4,column=0)

rt.mainloop()