def OnButtonAICarImaClick(self):
ei = facerecEscIma_tk(None)
#pathadd = filedialog.askopenfilename(initialdir = PATH + '/',title = "Escolha uma imagem",filetypes = (("Arquivos JPG","*.jpg"),("Arquivos PNG","*.png"),("all files","*.*")))
ei.focus_force()
self.withdraw()
self.wait_window(ei)
self.deiconify()
if func.conta_temp(PATH)>0:
pathadd = ei.pathadd
conn = sqlite3.connect('usuarios.db')
id = dbm.get_idDB(conn, self.selectUser.get())
conn.close()
files = os.listdir(PATH + '/temp_dir')
for f in files:
move(PATH + '/temp_dir/'+f, PATH + '/train_dir/' + str(id))
messagebox.showinfo("Notificação", "Imagem adicionada! Para que ela seja utilizada pelo sistema, treinar o classificador.")
def initialize(self):
self.grid()
self.lblDig= tkinter.Label(self, text="Escolha como adicionar sua imagem", background='#b3d0e3', font=("Arial",12, 'bold'))
self.lblCon= tkinter.Label(self, text="Imagens pendentes: " + str(func.conta_temp(PATH)), background='#b3d0e3', font="-size 10 -weight bold")
self.btnAddIma = tkinter.Button(self,text=u"Escolher\n Arquivo", command=self.OnButtonAICarImaClick)
self.btnAddWC = tkinter.Button(self,text=u"Utilizar\n Webcam", command=self.OnButtonAICarWebClick)
self.btnSalvar = tkinter.Button(self,text=u"Salvar", command=self.OnButtonSalvarClick)
self.protocol("WM_DELETE_WINDOW", self.on_closing)
self.btnAddIma.place(x=55,y=70)
self.btnAddWC.place(x=185,y=70)
self.lblDig.place(x=7, y=20)
self.lblCon.place(x=80, y=45)
self.btnSalvar.place(x=130, y=115)
pass
def OnButtonAddUClick(self):
if self.nomeuser.get()=="":
messagebox.showinfo("Notificação", "Por favor, digite seu nome.")
elif func.conta_temp(PATH)==0:
messagebox.showinfo("Notificação", "Por favor, escolha sua primeira imagem.")
else:
seed(datetime.now())
rndid=randint(1, 9999)
if not os.path.exists(PATH + '/train_dir/' + str(rndid)):
try:
dbm.inserirUserDB(rndid, self.nomeuser.get(), self.tarseg.get(), self.tarter.get(), self.tarqua.get(), self.tarqui.get(), self.tarsex.get(), self.checkad.get())
os.makedirs(PATH + '/train_dir/' + str(rndid))
files = os.listdir(PATH + '/temp_dir')
for f in files:
move(PATH + '/temp_dir/'+f, PATH + '/train_dir/' + str(rndid))
messagebox.showinfo("Notificação", "Usuário adicionado! Não se esqueça de treinar o classificador.")
self.destroy()
except sqlite3.IntegrityError:
messagebox.showinfo("Notificação", "Esse usuário já existe!")
else:
messagebox.showinfo("Notificação", "Esse usuário já existe!")
def OnButtonAICarWebClick(self):
wc = facerecWebCam_tk(self, 1)
self.wait_window(wc)
self.lblCon.config(text="Imagens pendentes: " + str(func.conta_temp(PATH)))
def OnButtonAICarImaClick(self):
copy2(filedialog.askopenfilename(initialdir = PATH + '/',title = "Escolha uma imagem",filetypes = (("Arquivos JPG","*.jpg"),("Arquivos PNG","*.png"),("all files","*.*"))), PATH + '/temp_dir')
self.lblCon.config(text="Imagens pendentes: " + str(func.conta_temp(PATH)))