[update]: schema in models

app/models/account.py

@@ -2,6 +2,9 @@ from datetime import datetime
 from app import db
 
 class Account(db.Model):
+    __tablename__ = 'accounts'
+    __table_args__ = {'schema': 'flask_app'} 
+
     id = db.Column(db.String(50), primary_key=True)  
     customer_id = db.Column(db.String(50), db.ForeignKey('customer.id'), nullable=False)
     account_type = db.Column(db.String(50))

app/models/customer.py

@@ -2,6 +2,9 @@ from datetime import datetime
 from app import db
 
 class Customer(db.Model):
+    __tablename__ = 'customers'
+    __table_args__ = {'schema': 'flask_app'} 
+
     id = db.Column(db.String(50), primary_key=True) 
     msisdn = db.Column(db.String(20), unique=True, nullable=False)
     country_code = db.Column(db.String(3), nullable=False)

app/models/loan.py

@@ -3,6 +3,9 @@ from app import db
 
 
 class Loan(db.Model):
+    __tablename__ = 'loans'
+    __table_args__ = {'schema': 'flask_app'} 
+
     id = db.Column(db.String(50), primary_key=True) 
     customer_id = db.Column(db.String(50), db.ForeignKey('customer.id'), nullable=False)
     account_id = db.Column(db.String(50), db.ForeignKey('account.id'), nullable=False)

app/models/offer.py

@@ -2,6 +2,9 @@ from datetime import datetime
 from app import db
 
 class Offer(db.Model):
+    __tablename__ = 'offers'
+    __table_args__ = {'schema': 'flask_app'} 
+
     id = db.Column(db.Integer, primary_key=True)
     amount = db.Column(db.Float, nullable=False)
     interest_rate = db.Column(db.Float, nullable=False)

app/models/transaction.py

@@ -2,6 +2,9 @@ from datetime import datetime
 from app import db
 
 class Transaction(db.Model):
+    __tablename__ = 'transactions'
+    __table_args__ = {'schema': 'flask_app'} 
+
     id = db.Column(db.String(50), primary_key=True)
     account_id = db.Column(db.String(50), db.ForeignKey('account.id'), nullable=False)
     type = db.Column(db.String(50), nullable=False)