Pi-Secure/handler.py at master · Flajt/Pi-Secure · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import notifier
import recognizer
import os
import cv2 as cv
import time


bad_path="/home/pi/Desktop/Pi-Secure/bad_images/"
Id=open("/home/pi/Desktop/Pi-Secure/dataset/ID.txt","r")
Idlist=[]
nameslist=[]
recognizer=recognizer.recognizer("/home/pi/opencv-3.1.0/data/haarcascades/haarcascade_frontalface_default.xml")
notifier=notifier.notyfie()

def setup():
	for ids in Id:
		spliter=ids.split(" ")
		name=spliter[0]
		ID=spliter[1].split("\n")
		Idlist.append(ID[0])
		nameslist.append(name)
	main_detection()


def main_detection():
	os.chdir("/home/pi/Desktop/Pi-Secure/")
	blackliste=[]
	persons=[]
	try:
		blacklist=open("blacklist.txt","r")
		for i in blacklist:
			ID_in_black_list=i.split(" ")[1]#check if everything is ok
			person=ID_in_black_list[0]
			persons.append(person)
			blacklist.apppend(ID_in_black_list)
	except Exception as e:
		pass

	while True:
		IDS, person_found, conf, frame=recognizer.recognize("http://192.168.178.49:8080/video")
		try:
			if conf>=80.0:
				#print(frame)
				#print(type(frame))
				curtime=time.strftime("%d.%m.%Y%H:%M:%S")
				cv.imwrite(bad_path+curtime+".jpg",frame)
				IDS="Unknown person"
				notifier.sendmail()
				notifier.instapush(curtime,unkown=True)
			for i in blackliste:
				if IDS==i:
					curtime=time.strftime("%d.%m.%Y%H:%M:%S")
					cv.imwrite(bad_path+curtime+".jpg",frame)
					notifier.instapush(blacklisted=persons.index(i-1))
					notifier.sendmail(content="A blacklisted person enterd your area! Image below")

			if name==nameslist[IDS-1]:
				curtime=time.strftime("%d.%m.%Y%H:%M:%S")
				cv.imwrite(bad_path+curtime+".jpg",frame)
				notifier.instapush(inform=True, person=name)
				notifier.sendmail(curtime,content="Your notification. A friend arrives")

		except TypeError:
			pass

			#print(IDS)
			#print("conf "+str(conf))

		for objects in Idlist:
			if str(IDS)==objects:
				#print(objects)
				name=nameslist[int(objects)-1]
				print(name)

try:
	os.chdir("/home/pi/Desktop/Pi_Secure/")
	with open("inform_me.txt","r") as f:
		name=f.read()
		f.close()

except Exception as e:
	setup()

setup()