Add: Country column

iebank_api/models.py

@@ -8,15 +8,17 @@ class Account(db.Model):
  account_number = db.Column(db.String(20), nullable=False, unique=True)
  balance = db.Column(db.Float, nullable=False, default = 0.0)
  currency = db.Column(db.String(1), nullable=False, default="€")
+ country = db.Column(db.String(32), nullable=False, default="Spain")
  status = db.Column(db.String(10), nullable=False, default="Active")
  created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
 
  def __repr__(self):
  return '<Event %r>' % self.account_number
 
- def __init__(self, name, currency):
+ def __init__(self, name, currency, country):
  self.name = name
  self.account_number = ''.join(random.choices(string.digits, k=20))
  self.currency = currency
  self.balance = 0.0
- self.status = "Active"
+ self.status = "Active"
+ self.country = country

iebank_api/routes.py

@@ -32,8 +32,9 @@ def skull():
 def create_account():
  app.logger.debug('Route /accounts POST called')
  name = request.json['name']
+ country = request.json['country']
  currency = request.json['currency']
- account = Account(name, currency)
+ account = Account(name, currency, country)
  db.session.add(account)
  db.session.commit()
  return format_account(account)
@@ -46,21 +47,21 @@ def get_accounts():
 
 @app.route('/accounts/<int:id>', methods=['GET'])
 def get_account(id):
- app.logger.debug('Route /accounts GET called with id: ' + id)
+ app.logger.debug('Route /accounts GET called with id: ' + str(id))
  account = Account.query.get(id)
  return format_account(account)
 
 @app.route('/accounts/<int:id>', methods=['PUT'])
 def update_account(id):
- app.logger.debug('Route /accounts PUT called with id: ' + id)
+ app.logger.debug('Route /accounts PUT called with id: ' + str(id))
  account = Account.query.get(id)
  account.name = request.json['name']
  db.session.commit()
  return format_account(account)
 
 @app.route('/accounts/<int:id>', methods=['DELETE'])
 def delete_account(id):
- app.logger.debug('Route /accounts DELETE called with id: ' + id)
+ app.logger.debug('Route /accounts DELETE called with id: ' + str(id))
  account = Account.query.get(id)
  db.session.delete(account)
  db.session.commit()
@@ -73,6 +74,7 @@ def format_account(account):
  'account_number': account.account_number,
  'balance': account.balance,
  'currency': account.currency,
+ 'country': account.country,
  'status': account.status,
  'created_at': account.created_at
  }