From 9c1d7d1c986ffb4051340524dd218bdc7135f9b5 Mon Sep 17 00:00:00 2001
From: NikolajDanger <nikolajgade@live.dk>
Date: Tue, 24 May 2022 14:16:44 +0200
Subject: [PATCH] :broom: small cleanup

---
 aula/forms.py        |  2 +-
 aula/models.py       | 57 ++++----------------------------------------
 bank/Login/routes.py |  4 ++--
 3 files changed, 8 insertions(+), 55 deletions(-)

diff --git a/aula/forms.py b/aula/forms.py
index d00f404..aeb52b0 100644
--- a/aula/forms.py
+++ b/aula/forms.py
@@ -1,5 +1,5 @@
 from flask_wtf import FlaskForm
-from wtforms import StringField, PasswordField, SubmitField, BooleanField, IntegerField, SelectField
+from wtforms import StringField, PasswordField, SubmitField, BooleanField, IntegerField
 from wtforms.validators import DataRequired, Length
 
 class AddUserForm(FlaskForm):
diff --git a/aula/models.py b/aula/models.py
index f272aa7..cc6d999 100644
--- a/aula/models.py
+++ b/aula/models.py
@@ -65,71 +65,24 @@ class User(tuple, UserMixin):
         self.address = user_data[5]
         self.role = user_data[6]
 
-#
-# SQL
-#
-
-
-
-
-class Customers(tuple, UserMixin):
-    def __init__(self, user_data):
-        self.CPR_number = user_data[0]
-        self.risktype = False
-        self.password = user_data[2]
-        self.name = user_data[3]
-        self.address = user_data[4]
-
-    def get_id(self):
-       return (self.CPR_number)
-
-class Employees(tuple, UserMixin):
-    def __init__(self, employee_data):
-        self.id = employee_data[0]
-        self.name = employee_data[1]
-        self.password = employee_data[2]
-
-    def get_id(self):
-       return (self.id)
-
-class CheckingAccount(tuple):
-    def __init__(self, user_data):
-        self.id = user_data[0]
-        self.create_date = user_data[1]
-        self.CPR_number = user_data[2]
-        self.amount = 0
-
-class InvestmentAccount(tuple):
-    def __init__(self, user_data):
-        self.id = user_data[0]
-        self.start_date = user_data[1]
-        self.maturity_date = user_data[2]
-        self.amount = 0
-
-class Transfers(tuple):
-    def __init__(self, user_data):
-        self.id = user_data[0]
-        self.amount = user_data[1]
-        self.transfer_date = user_data[2]
-
 def insert_users(user_id, first_name, last_name, password, email, adresse, role):
     cur = conn.cursor()
-    sql = """
+    sql_call = """
     INSERT INTO Customers(user_id, first_name, last_name, password, email, adresse, role)
     VALUES (%s, %s, %s, %s, %s, %s, %s)
     """
-    cur.execute(sql, (user_id, first_name, last_name, password, email, adresse, role))
+    cur.execute(sql_call, (user_id, first_name, last_name, password, email, adresse, role))
     # Husk commit() for INSERT og UPDATE, men ikke til SELECT!
     conn.commit()
     cur.close()
 
 def select_users(user_id):
     cur = conn.cursor()
-    sql = """
+    sql_call = """
     SELECT * FROM users
     WHERE user_id = %s
     """
-    cur.execute(sql, (user_id,))
-    user = Customers(cur.fetchone()) if cur.rowcount > 0 else None;
+    cur.execute(sql_call, (user_id,))
+    user = User(cur.fetchone()) if cur.rowcount > 0 else None
     cur.close()
     return user
diff --git a/bank/Login/routes.py b/bank/Login/routes.py
index 6e0de92..f31a460 100644
--- a/bank/Login/routes.py
+++ b/bank/Login/routes.py
@@ -1,8 +1,8 @@
 from flask import render_template, url_for, flash, redirect, request, Blueprint
-from bank import app, conn, bcrypt
+from bank import bcrypt
 from bank.forms import CustomerLoginForm, EmployeeLoginForm
 from flask_login import login_user, current_user, logout_user, login_required
-from bank.models import Customers, select_Customers, select_Employees
+from bank.models import select_Customers, select_Employees
 
 Login = Blueprint('Login', __name__)