Added custom 404 display and login redirect

ref-test/admin/models/users.py

@@ -1,4 +1,4 @@
-from flask import flash, make_response, Response
+from flask import flash, make_response, Response, session
 from flask.helpers import url_for
 from flask.json import jsonify
 from werkzeug.security import generate_password_hash, check_password_hash
@@ -61,7 +61,13 @@ class User:
             return jsonify({ 'error': f'Username {self.username} does not exist.' }), 401
         if not check_password_hash( user['password'], self.password ):
             return jsonify({ 'error': f'The password you entered is incorrect.' }), 401
-        resp = make_response(jsonify({ 'success': f'Successfully logged in user {self.username}.' }), 200)
+        response = {
+            'success': f'Successfully logged in user {self.username}.'
+        }
+        if 'prev_page' in session:
+            response['redirect_to'] = session['prev_page']
+            session.pop('prev_page')
+        resp = make_response(jsonify(response), 200)
         self._id = user['_id']
         self.start_session(resp)
         return resp