chore: 🔨 lvl2 refactor

Author: codeperfectplus

.gitignore

@@ -57,7 +57,7 @@ cover/
 
 # Django stuff:
 *.log
-local_settings.py
+local_.py
 db.sqlite3
 db.sqlite3-journal
 
@@ -130,7 +130,7 @@ ENV/
 env.bak/
 venv.bak/
 
-# Spyder project settings
+# Spyder project 
 .spyderproject
 .spyproject
 

app.py

@@ -0,0 +1,28 @@
+from flask import render_template, Blueprint
+
+
+additional_bp = Blueprint('additional', __name__)
+
+@additional_bp.route('/pricing')
+def pricing() -> str:
+    return render_template('pricing.html')
+
+@additional_bp.route('/features')
+def features() -> str:
+    return render_template('features.html')
+
+@additional_bp.route('/about')
+def about() -> str:
+    return render_template('about.html')
+
+
+@additional_bp.route('/permission-denied')
+def permission_denied():
+    return render_template('permission_denied.html')
+
+@additional_bp.errorhandler(404)
+def page_not_found(e):
+    return render_template('404.html'), 404
+
+
+

static/css/audit_log.css src/__init__.py

@@ -0,0 +1,116 @@
+import psutil
+
+from flask import render_template, request, session, redirect, url_for, flash, Blueprint
+
+from flask_login import current_user, login_required
+from flask_login import current_user, login_required
+
+from src.config import db
+from src.models import User, AuditLog, Settings, log_action
+
+admin_bp = Blueprint('admin', __name__)
+
+@admin_bp.route('/admin/manage_users', methods=['GET', 'POST'])
+@login_required
+def manage_users():
+    if current_user.user_level != 'admin':
+        return redirect(url_for('permission_denied'))
+
+    users = User.query.all()
+
+    if request.method == 'POST':
+        user_id = request.form.get('user_id')
+        new_level = request.form.get('user_level')
+        
+        user = User.query.get(user_id)
+        if user:
+            user.user_level = new_level
+            db.session.commit()
+            flash(f"User {user.username}'s level updated to {new_level}.", 'success')
+        else:
+            flash("User not found.", 'error')
+
+    return render_template('manage_users.html', users=users)
+
+@admin_bp.route('/settings', methods=['GET', 'POST'])
+@login_required
+def settings():
+    if current_user.user_level != 'admin':
+        return redirect(url_for('home'))
+
+    if request.method == 'POST':
+        currency = request.form.get('currency')
+        theme = request.form.get('theme')
+        language = request.form.get('language')
+        is_gold_jewellers_sidebar = int(request.form.get('is_gold_jewellers_sidebar'))
+        is_gold_calculator_enabled = int(request.form.get('is_gold_calculator_enabled'))
+        is_silver_calculator_enabled = int(request.form.get('is_silver_calculator_enabled'))
+
+        # Save the settings (you might save them to a database or a config file)
+        # Assuming you have a Settings model or similar logic to save settings
+        settings = Settings.query.first()
+        if not settings:
+            settings = Settings(currency=currency, theme=theme, language=language,
+                                is_gold_jewellers_sidebar=is_gold_jewellers_sidebar,
+                                is_gold_calculator_enabled=is_gold_calculator_enabled,
+                                is_silver_calculator_enabled=is_silver_calculator_enabled)
+        else:
+            settings.currency = currency
+            settings.theme = theme
+            settings.language = language
+            settings.is_gold_jewellers_sidebar = is_gold_jewellers_sidebar
+            settings.is_gold_calculator_enabled = is_gold_calculator_enabled
+            settings.is_silver_calculator_enabled = is_silver_calculator_enabled
+
+        db.session.add(settings)
+        db.session.commit()
+        
+        log_action(current_user.id, current_user.username, 'System Settings Change', details=f"Currency set to {currency}, Theme set to {theme}")
+        return redirect(url_for('settings'))
+    
+    settings = Settings.query.first()
+    return render_template('settings.html', settings=settings)
+
+
+@admin_bp.route('/admin/audit_log')
+@login_required
+def audit_log():
+    if current_user.user_level != 'admin':
+        return redirect(url_for('home'))
+
+    logs = AuditLog.query.order_by(AuditLog.timestamp.desc()).all()
+    return render_template('audit_log.html', logs=logs)
+
+
+@admin_bp.route("/dashboard")
+@login_required
+def dashboard():
+    if current_user.user_level == 'admin':
+        total_users = User.query.count()  # Count total users
+        active_sessions = len(session)  # This is a basic approach. You may want to track sessions differently.
+        
+        # Example: If you store active users in the session, you might track like this:
+        active_sessions = session.get('active_users', 0)
+        
+        system_health = "Good"  # This can be determined by your own logic
+        cpu_core = psutil.cpu_count()
+        cpu_utilization = psutil.cpu_percent(interval=1)
+
+        if cpu_utilization > 80:
+            system_health = "Warning"
+        else:
+            system_health = "Good"
+
+        audit_logs = AuditLog.query.filter_by(user_id=current_user.id).all()
+        return render_template('admin_dashboard.html', 
+                            total_users=total_users, 
+                            active_sessions=active_sessions, 
+                            system_health=system_health,
+                            cpu_core=cpu_core,
+                            cpu_util=cpu_utilization,
+                            audit_logs=audit_logs)
+    elif current_user.user_level == 'customer':
+        audit_logs = AuditLog.query.filter_by(user_id=current_user.id).all()
+        return render_template('customer_dashboard.html', audit_logs=audit_logs)
+ 
+

src/blueprints/__init__.py

@@ -0,0 +1,51 @@
+
+from flask import render_template, request, redirect, url_for,  Blueprint
+
+from flask_login import login_user, logout_user
+
+from src.config import app, db, bcrypt, login_manager
+from src.models import User, log_action
+
+auth_bp = Blueprint('auth', __name__)
+
+@auth_bp.route("/register", methods=['GET', 'POST'])
+def register():
+    if request.method == 'POST':
+        fname = request.form.get('fname')
+        lname = request.form.get('lname')
+        username = request.form.get('username')
+        email = request.form.get('email')
+        password = bcrypt.generate_password_hash(request.form.get('password')).decode('utf-8')
+        # Check if this is the first user
+        if User.query.count() == 0:
+            user_level = 'admin'  # Make the first user an admin
+        else:
+            user_level = 'customer'  # Default to 'customer' for all other users
+        user = User(fname=fname, lname=lname, username=username, email=email, password=password, user_level=user_level)
+        db.session.add(user)
+        db.session.commit()
+        return redirect(url_for('login'))
+    return render_template('register.html')
+
+@auth_bp.route("/login", methods=['GET', 'POST'])
+def login():
+    if request.method == 'POST':
+        email = request.form.get('email')
+        password = request.form.get('password')
+        user = User.query.filter_by(email=email).first()
+        if user and bcrypt.check_password_hash(user.password, password):
+            login_user(user)
+            log_action(user.id, user.username, 'Login', f'User {user.username} logged in.')
+            return redirect(url_for('admin.dashboard'))
+    return render_template('login.html')
+
+@auth_bp.route("/logout")
+def logout():
+    logout_user()
+    return redirect(url_for('auth.login'))
+
+@login_manager.user_loader
+def load_user(user_id):
+    return User.query.get(int(user_id))
+
+

src/blueprints/additional.py

@@ -0,0 +1,33 @@
+
+from flask import render_template, request, redirect, url_for, Blueprint
+
+from src.config import db
+from src.models import JewellerDetails
+
+config_bp = Blueprint('config', __name__)
+
+@config_bp.route('/config', methods=['GET', 'POST'])
+def update_jeweller_details():
+    if request.method == 'POST':
+        jeweller = JewellerDetails.query.first()
+
+        jeweller.jeweller_name = request.form['jeweller_name']
+        jeweller.jeweller_address = request.form['jeweller_address']
+        jeweller.jeweller_contact = request.form['jeweller_contact']
+        jeweller.jeweller_email = request.form['jeweller_email']
+        jeweller.jeweller_website = request.form['jeweller_website']
+        jeweller.jeweller_gstin = request.form['jeweller_gstin']
+        jeweller.gold_price_per_gram = float(request.form['gold_price_per_gram'])
+
+        if 'jeweller_logo' in request.files:
+            logo = request.files['jeweller_logo']
+            logo.save(f'static/images/{logo.filename}')
+            jeweller.jeweller_logo = f'images/{logo.filename}'
+
+        db.session.commit()
+        return redirect(url_for('update_jeweller_details'))
+
+    jeweller = JewellerDetails.query.first()
+    return render_template('config.html', jeweller=jeweller)
+
+

src/blueprints/admin.py

@@ -0,0 +1,86 @@
+
+import logging 
+
+from flask import render_template, request, session, redirect, url_for, flash, Blueprint
+from flask_login import current_user
+from calculators import GoldCalculator
+
+from src.config import db
+from src.models import Settings, GoldTransaction, JewellerDetails, log_action
+from src.blueprints.helper import get_currency_symbol
+
+gold_calculator_bp = Blueprint('gold_calculator', __name__)
+
+# Gold calculator route
+@gold_calculator_bp.route('/gold-calculator', methods=['GET', 'POST'])
+def gold_calculator():
+    system_settings = Settings.query.first()
+    jeweller_details = JewellerDetails.query.first()
+    if not system_settings.is_gold_calculator_enabled:
+        return redirect(url_for('permission_denied'))
+    if request.method == 'POST':
+        try:
+            weight = float(request.form['weight'])
+            gold_price_per_gram = float(request.form['price_per_gram'])
+            gold_service_charge = float(request.form['service_charge'])
+            gold_tax = float(request.form['tax'])
+            purity = request.form['purity']
+
+            # Save price per gram to session
+            session['gold_price_per_gram'] = gold_price_per_gram
+            session['gold_service_charge'] = gold_service_charge
+            session['gold_tax'] = gold_tax
+
+            # Calculate gold price
+            gold_item = GoldCalculator(weight, gold_price_per_gram, gold_service_charge, gold_tax)
+            bill_details = gold_item.calculate_price()
+
+            # Save to database
+            transaction = GoldTransaction(
+                weight=weight,
+                price_per_gram=gold_price_per_gram,
+                purity=purity,
+                service_charge=gold_service_charge,
+                tax=gold_tax,
+                total=bill_details['Final Price'],
+                currency=system_settings.currency,
+            )
+            db.session.add(transaction)
+            db.session.commit()
+
+            # if user is logged in, log the transaction
+            if current_user.is_authenticated:
+                log_action(user_id=current_user.id, username=current_user.username,
+                           action='Gold Calculator', details=f"Calculated gold price for weight {weight} grams")
+            else:
+                log_action(user_id="-1", username="Anonymous",
+                           action='Gold Calculator', details=f"Calculated gold price for weight {weight} grams")
+                
+            return render_template('gold_bill.html',
+                                   bill=bill_details,
+                                   weight=weight,
+                                   price_per_gram=gold_price_per_gram,
+                                   purity=purity,
+                                   settings=system_settings,
+                                   jeweller_details=jeweller_details,
+                                   currency_symbol=get_currency_symbol(system_settings.currency))
+        
+        except ValueError as e:
+            logging.error(f"ValueError in gold calculator: {str(e)}")
+            flash(f"Input error: {str(e)}", 'error')
+            return redirect(url_for('gold_calculator'))
+
+    # Use session-stored price per gram or a default value
+    gold_price_per_gram = session.get('gold_price_per_gram', 0)
+    gold_service_charge = session.get('gold_service_charge', 0)
+    gold_tax = session.get('gold_tax', 0)
+
+    return render_template('gold_calculator.html',
+                           price_per_gram=gold_price_per_gram,
+                           service_charge=gold_service_charge,
+                           tax=gold_tax,
+                           settings=system_settings,
+                           jeweller_details=jeweller_details,
+                           currency_symbol=get_currency_symbol(system_settings.currency))
+                           
+

src/blueprints/auth.py

@@ -0,0 +1,15 @@
+from datetime import datetime
+
+from flask import render_template, Blueprint
+
+health_bp = Blueprint('health', __name__)
+
+@health_bp.route('/health')
+def health():
+    health_info = {
+        "status": "healthy",
+        "message": "The server is up and running.",
+        "version": "1.0.0",
+        "timestamp": datetime.now().isoformat()
+    }
+    return render_template('health.html', **health_info)

src/blueprints/config.py

@@ -0,0 +1,10 @@
+def get_currency_symbol(currency):
+    currency_to_symbol_dict = {
+        "INR" : "₹",
+        "USD" : "$ ",
+        "EUR" : "€ ",
+        "GBP" : "£ ",
+        "JPY" : "¥ ",
+        "AUD" : "A$ ",
+    }
+    return currency_to_symbol_dict.get(currency, currency)