commit

Author: sasank174

__pycache__/crypto.cpython-38.pyc

@@ -1,8 +1,10 @@
-from flask import Flask, render_template, request, redirect, session, url_for, flash, jsonify
+import cv2
+import crypto
+from flask import Flask, Response, render_template, request, redirect, session, url_for, flash, jsonify
 from flask_mail import Mail, Message
 from werkzeug.utils import secure_filename
-from facedetector import faceencodingvalues
-from flask_recaptcha import ReCaptcha
+from facedetector import faceencodingvalues, predata, detect
+import tokens
 import db
 import os
 
@@ -13,31 +15,16 @@
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 app.secret_key = os.getenv('SECRET_KEY')
 
-recaptcha = ReCaptcha(app=app)
 
-app.config.update(dict(
-    RECAPTCHA_ENABLED = True,
-    RECAPTCHA_SITE_KEY = os.getenv('RECAPTCHA_SITE_KEY'),
-    RECAPTCHA_SECRET_KEY = os.getenv('RECAPTCHA_SECRET_KEY'),
-	RECAPTCHA_SIZE = 'invisible',
-	RECAPTCHA_THEME = "dark",
-))
-
-# RECAPTCHA_DATA_ATTRS = {'bind': 'recaptcha-submit', 'callback': 'onSubmitCallback', 'size': 'invisible'}
+dbconnect = db.connect()
 
-recaptcha = ReCaptcha()
-recaptcha.init_app(app)
 
-
-dbconnect = db.connect()
+global recorded,cap
+recorded = "NO"
 
 # =============================================================================================================
 
-def mailing(tomail,username,token,no):
-	if no == 1:
-		x,y = "conformation Email","confirm_email"
-	elif no == 2:
-		x,y = "reset password","reset_password"
+def mailing(tomail,username,token):
 	try:
 		app.config['MAIL_SERVER']='smtp.gmail.com'
 		app.config['MAIL_PORT'] = 465
@@ -48,14 +35,14 @@ def mailing(tomail,username,token,no):
 		mail = Mail(app)
 		msg = Message('Hello', sender = os.getenv('EMAIL'), recipients = [tomail])
 		msg.body = "<h1>Hello Flask message sent from Flask-Mail</h1>"
-		msg.subject = x
-		link = "https://adist.herokuapp.com/{}/{}".format(y,token)
-		msg.html = "<div><h1>change password</h1><h1><a href='"+link+"'}>click me</a></h1></div>"
+		msg.subject = "Login to your account"
+		link = "http://127.0.0.1:8000/logincheck/{}".format(token)
+		msg.html = "<div><h1>Login to your account</h1><h1><a href='"+link+"'}>click here</a></h1></div>"
 		msg.html = '''<div
 		style="text-align:center;max-width:600px;background:rgba( 255, 255, 255, 0.25 );box-shadow: 0 8px 32px 0 rgba( 31, 38, 135, 0.37 );backdrop-filter: blur( 4px );border-radius: 10px;border: 1px solid rgba( 255, 255, 255, 0.18 );">
-			<h1>Adist</h1>
-			<h2>Verification mail</h2>
-			<h2>hi {} click the link below to conform your mail</h2>
+			<h1>Authenticator</h1>
+			<h2>Login to your account</h2>
+			<h2>hi {} click the link below to Login to your account</h2>
 			<h3><a href='{}' >Click Here</a></h3>
 			<p>Copy paste in browser if the above link is not working: {}</p>
 		</div>'''.format(username,link,link)
@@ -70,7 +57,10 @@ def mailing(tomail,username,token,no):
 @app.route("/home")
 def home():
 	if "user" in session:
-		return render_template("index.html",user=session["user"])
+		if "verify" in session:
+			return render_template("index.html", user = session["user"])
+		else:
+			return render_template("verify.html", user = session["user"])
 	elif "admin" in session:
 		return redirect(url_for("admin"))
 	else:
@@ -83,18 +73,69 @@ def login():
 		return render_template("login.html",RECAPTCHA_SITE_KEY = os.getenv('RECAPTCHA_SITE_KEY'))
 	if request.method == "POST":
 		email = request.form.to_dict()["email"]
-		q = "select * from tempusers where email = '{}'".format(email)
+		q = "select username,email from tempusers where email = '{}'".format(email)
 		result = db.select(q)
 		if len(result) == 0:
 			flash("Invalid email")
 			return redirect(url_for("login"))
 		elif len(result) == 1:
 			result = result[0]
-			session["user"] = email
-			return redirect(url_for("home"))
+			username,email = result[0],result[1]
+
+			token = tokens.generate_confirmation_token(email)
+			q = "update tempusers set token = '{}' where email = '{}'".format(token,email)
+			if db.insert(q):
+				if mailing(email,username,token):
+					flash("Check your email to for login link")
+					return redirect(url_for("login"))
+				else:
+					flash("Something went wrong during mailing")
+					return redirect(url_for("login"))
+			else:
+				flash("Something went wrong with our database")
+				return redirect(url_for("login"))
+
+			return redirect(url_for("login"))
+		else:
+			flash("Something went wrong")
+			return redirect(url_for("login"))
+
+# ============================================================================================================
+
+@app.route("/logincheck/<token>")
+def logincheck(token):
+	email = tokens.confirm_token(token)
+	if email == "The token has expired":
+		flash("The token has expired")
+		return redirect(url_for("login"))
+	elif email == "the token is invalid":
+		flash("the token is invalid")
+		return redirect(url_for("login"))
+	elif email:
+		q = "select username,email,token from tempusers where email = '{}'".format(email)
+		result = db.select(q)
+		if len(result) == 1:
+			result = result[0]
+			username,email = result[0],result[1]
+			if token == result[2]:
+				q = "update tempusers set token = 'no' where email = '{}'".format(email)
+				if db.insert(q):
+					session["user"] = email
+					return redirect(url_for("home"))
+
+				else:
+					flash("Something went wrong with our database")
+					return redirect(url_for("login"))
+			else:
+				flash("use the link that was last sent to your email")
+				return redirect(url_for("login"))
 		else:
 			flash("Something went wrong")
 			return redirect(url_for("login"))
+	else:
+		flash("Something went wrong")
+		return redirect(url_for("login"))
+
 
 # ============================================================================================================
 
@@ -135,7 +176,7 @@ def admin():
 					if len(facelocs)==0:
 						flash("No face detected")
 						return redirect(url_for("admin"))
-					q = "insert into tempusers(username,email,token,encodings) values('{}','{}','{}','{}')".format(username,email,"no",str(faceencodings.tolist()))
+					q = "insert into tempusers(username,email,token,encodings) values('{}','{}','{}','{}')".format(username,email,"no",crypto.encryption(str(faceencodings.tolist())))
 					db.insert(q)
 					flash("User added")
 					return redirect(url_for("admin"))
@@ -153,6 +194,35 @@ def admin():
 				return redirect(url_for("admin"))
 
 
+# ============================================================================================================
+
+cap = cv2.VideoCapture(0)
+# =============================for attendence recording============================================================================================
+def gen_frames(email):
+	global recorded,cap
+	cap = cv2.VideoCapture(0)
+	while True:
+		sucess,img = cap.read()
+		(frame,ans) = detect(img,email)
+		recorded = ans
+		if recorded == "YES":
+			break
+
+		yield(b'--frame\r\n'
+					b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
+	print("yield condition exit")
+	cap.release()
+
+
+
+@app.route("/video_feed")
+def video_feed():
+	predata(session["user"])
+	return Response(gen_frames(session["user"]),
+					mimetype='multipart/x-mixed-replace; boundary=frame')
+
+
+
 # ============================================================================================================
 
 @app.route("/logout")
@@ -162,7 +232,8 @@ def logout():
 
 @app.errorhandler(404)
 def page_not_found(e):
-    return redirect("/")
+	flash("Page not found")
+	return redirect("/")
 
 
 if __name__ == '__main__':

__pycache__/facedetector.cpython-38.pyc

@@ -0,0 +1,26 @@
+from Crypto.Cipher import AES
+from Crypto.Util.Padding import pad, unpad
+from Crypto.Random import get_random_bytes
+
+key = b'\xbc\x96(WA\x82\n\xfbh\x80\xcb\xb2<\x1d\xfea'
+iv = b"\xef\xd1\xf4G'T\xb8\x10\xa3w<L\x08\xa0;\x96"
+
+def encryption(plaintext):
+    plaintext = bytes(plaintext, 'utf-8')
+    cipher = AES.new(key,AES.MODE_CBC,iv)
+    text = cipher.encrypt(pad(plaintext,16))
+    return text.hex()
+
+def decryption(ciphertext):
+    ciphertext = bytes.fromhex(ciphertext)
+    decipher = AES.new(key,AES.MODE_CBC,iv)
+    text = unpad(decipher.decrypt(ciphertext),block_size=16)
+    return text.decode('utf-8') 
+
+# text = 'sasank world'
+
+# encrypted = encryption(text)
+# print(encrypted)
+
+# decrypted = decryption(encrypted)
+# print(decrypted)

__pycache__/tokens.cpython-38.pyc

@@ -1,6 +1,16 @@
+from base64 import encode
 import face_recognition
 import numpy as np
 import cv2
+import db
+import crypto
+
+dbconnect = db.connect()
+
+global knownencodings,count,sub
+knownencodings =[]
+count = 0
+sub=""
 
 def faceencodingvalues(img):
 	print("===============start=====================================================")
@@ -16,4 +26,69 @@ def faceencodingvalues(img):
 	print("==================encodeimg==================================================")
 	# print(encodeimg)
 
-	return (encodeimg,faceloc)
+	return (encodeimg,faceloc)
+
+
+
+def predata(email):
+	encodinglist = []
+	emaillist = []
+	q = "select email,encodings from tempusers"
+	result = db.select(q)
+	for i in result:
+		emaillist.append(i[0])
+		arr = crypto.decryption(i[1])
+		arr = np.array(eval(arr))
+		encodinglist.append(arr)
+	global knownencodings,count
+	knownencodings = encodinglist
+	count = emaillist.index(email)
+
+
+def detect(img,email):
+	imgS = cv2.resize(img,(0,0),None,0.25,0.25)
+	imgS = cv2.cvtColor(imgS,cv2.COLOR_BGR2RGB)
+	facesS = face_recognition.face_locations(imgS)
+	encodeS = face_recognition.face_encodings(imgS,facesS)
+
+	for encodeFace,faceLoc in zip(encodeS,facesS):
+		matches = face_recognition.compare_faces(knownencodings,encodeFace)
+		faceDis = face_recognition.face_distance(knownencodings,encodeFace)
+		matchindex = np.argmin(faceDis)
+		print("=======matchindex========================================")
+		print(matchindex)
+		print(count)
+
+
+
+
+	# print("===============start=====================================================")
+	# imgload = face_recognition.load_image_file(img)
+	# imgload = cv2.cvtColor(imgload,cv2.COLOR_BGR2RGB)
+	# try:
+	# 	faceloc = face_recognition.face_locations(imgload)[0]  # (260, 825, 528, 557)
+	# except:
+	# 	return (img, "NO")
+	# encodeimg = face_recognition.face_encodings(imgload)[0]
+	# print("===============faceloc=====================================================")
+	# # print(faceloc)
+	# print("==================encodeimg==================================================")
+	# # print(encodeimg)
+	# global knownencodings,count,sub
+	# distances = face_recognition.face_distance(knownencodings,encodeimg)
+	# print("===============distances=====================================================")
+	# # print(distances)
+	# if len(distances) == 0:
+	# 	return (img, "NO")
+	# else:
+	# 	distances = np.array(distances)
+	# 	min_value = np.min(distances)
+	# 	index = np.argmin(distances)
+	# 	# print(index)
+	# 	if min_value < 0.6:
+	# 		if email == emaillist[index]:
+	# 			return (img, "YES")
+	# 		else:
+	# 			return (img, "NO")
+	# 	else:
+	# 		return (img, "NO")

app.py

@@ -0,0 +1,32 @@
+<!-- <div class="jumbotron text-center">
+        <div class="container">
+            <h1>{{ subjectname }}</h1>
+            <div>
+                <img src="{{ url_for('video_feed') }}" height="60%">
+            </div>
+            <div>
+                <h1 id="test">not yet recorded</h1>
+            </div>
+            <script src="//ajax.googleapis.com/ajax/libs/jquery/1.9.1/jquery.min.js"></script>
+            <script>
+                $(document).ready(function () {
+                    refresh();
+                });
+
+                function refresh() {
+                    setTimeout(function () {
+                        $.getJSON('/recorded', function (data) {
+                            if (data == "YES") {
+                                window.location.href = "http://127.0.0.1:7890/recorddone";
+                            }
+                            $('#test').text(data);
+                        });
+                        refresh();
+                    }, 1000);
+                }
+            </script>
+            <hr>
+
+
+        </div>
+    </div> -->