11from base64 import b64decode
22from datetime import datetime
33from math import ceil
4- from os import listdir , remove
4+ from os import remove
5+ from os .path import isfile , getmtime
56from time import time
67
78from django .contrib .auth .decorators import login_required
8- from django .shortcuts import render , redirect , HttpResponse
9+ from django .shortcuts import render , redirect
910
10- from tracker import photos_path , utility , temp_path
11+ from tracker import model_filename , temp_path
12+ from tracker .engine import predict , train as do_train
1113from tracker .forms import UserForm , ImageForm
1214from tracker .models import Attendance , User , Image
13- from tracker .recognition import predict , get_nbr_photos , train as do_train
1415
1516
1617@login_required
1718def home (request ):
18- return render (request , "home.html" ,
19- {'photos' : get_nbr_photos (),
20- 'users' : User .objects .count (),
21- 'last_training' : utility .last_training ()}, )
19+ return render (request , "home.html" , {
20+ 'photos' : Image .objects .count (),
21+ 'users' : User .objects .count (),
22+ 'last_training' : datetime .fromtimestamp (getmtime (model_filename )) if isfile (model_filename ) else None
23+ })
2224
2325
2426@login_required
2527def add_user (request ):
2628 if request .method == 'POST' :
27- form = UserForm (request .POST , request .FILES )
28- instance = form .save (commit = False )
29- instance .save ()
30- return redirect (home )
29+ UserForm (request .POST , request .FILES ).save ()
30+ return redirect ('users' )
3131 return render (request , 'userdetails.html' , {'formset' : UserForm ()})
3232
3333
@@ -62,14 +62,14 @@ def upload(request):
6262
6363
6464@login_required
65- def display_users (request ):
65+ def users (request ):
6666 return render (request , 'user.html' , {'users' : User .objects .all ()})
6767
6868
6969@login_required
7070def train (request ):
71- if not utility . are_there_photos ():
72- return redirect ('/capture/ ?status=empty' )
71+ if not Image . objects . count ():
72+ return redirect ('/capture?status=empty' )
7373 if request .method == 'GET' :
7474 return render (request , 'train.html' )
7575 start = time ()
@@ -78,51 +78,44 @@ def train(request):
7878 return render (request , 'train.html' , {'duration' : duration })
7979
8080
81- @login_required
82- def receive_images (request ):
83- label = request .POST .get ('label' )
84- photos = request .POST .getlist ('photos[]' )
85- utility .save_base64_photos (label , photos )
86- return HttpResponse ()
87-
88-
8981@login_required
9082def profile (request , user_id ):
91- user_data = User .objects .get (pk = user_id )
92- images = [filename for filename in listdir (photos_path ) if filename .split ('_' )[0 ] == str (user_id )]
93- return render (request , 'profile.html' , {'user' : user_data , 'images' : images })
83+ return render (request , 'profile.html' , {'user' : User .objects .get (pk = user_id )})
9484
9585
9686@login_required
9787def delete_user (request , user_id ):
98- User .objects .filter (id = user_id ).delete ()
99- images = [filename for filename in listdir (photos_path ) if filename .split ('_' )[0 ] == str (user_id )]
100- for image in images :
101- remove ('static/photos/' + image )
88+ user = User .objects .get (id = user_id )
89+ for image in user .image_set .all ():
90+ try :
91+ remove (image .Image .path )
92+ except (FileNotFoundError , ValueError ):
93+ pass
94+ user .delete ()
10295 return redirect ('users' )
10396
10497
10598@login_required
10699def recognize (request ):
107- if not utility . is_model_trained ( ):
100+ if not isfile ( model_filename ):
108101 return redirect ('/train/?status=untrained' )
109102 if request .method == 'POST' :
110- paths = []
103+ files = []
111104 if frame := request .POST .get ('frame' ):
112105 ext , img = frame .split (';base64,' )
113106 ext = ext .split ('/' )[- 1 ]
114107 name = f'{ temp_path } { ext }
115108 with open (name , 'wb' ) as fh :
116109 fh .write (b64decode (img ))
117- paths .append (name )
110+ files .append (name )
118111 if images := request .FILES .getlist ('images' ):
119112 for i , image in enumerate (images ):
120113 name = f'{ temp_path } { i }
121114 with open (name , 'wb' ) as fh :
122115 fh .write (image .file .read ())
123- paths .append (name )
124- if paths :
125- if predictions := predict (paths ):
116+ files .append (name )
117+ if files :
118+ if predictions := predict (files ):
126119 user_id , confidence = predictions [0 ][:2 ]
127120 user = User .objects .get (id = user_id )
128121 name = 'Unknown' if user_id in (- 1 , None ) else user .first_name + ' ' + user .last_name
0 commit comments