def create_account():
if not request.json:
return jsonify({'error': 'bad request'}), 400
if 'username' not in request.json or 'password' not in request.json or 'firstname' not in request.json or 'lastname' not in request.json:
return jsonify({'error': 'bad request'}), 400
username = request.json['username']
password = request.json['password']
first = request.json['firstname']
last = request.json['lastname']
uniqueUsername = Account().username_preventduplicate(username)
if uniqueUsername == None:
new_account = Account(username=username,
balance=0,
first=first,
last=last)
new_account.set_password(password)
new_account.generate_api_key()
new_account.save()
return jsonify(
{
"Your account was created successfully.":
"You start with a balance of $0.00"
}, {"Your api key is": new_account.values['api_key']})
else:
return jsonify({"error": "bad username"})
def seed(dbpath=DBPATH):
ORM.dbpath = dbpath
default = Account(username='******',
balance=10000.00,
first='sami',
last='s',
api_key="12345678912345678902")
default.set_password('1234')
default.save()
default_buy = Trade(buy_sell='Buy',
username='******',
ticker='tsla',
price='100',
shares='10',
time=time())
default_buy.save()
default_sale = Trade(buy_sell='Sell',
username='******',
ticker='tsla',
price='110',
shares='5',
time=time())
default_sale.save()
default_position = Position(username='******', ticker='tsla', shares='5')
default_position.save()
def test_generate_api_key(self):
user = Account(username='******', balance=10000)
user.set_password('1234')
user.save()
user.generate_api_key()
reloaded = Account.login('some_user', '1234')
self.assertEqual(
user.values['api_key'], reloaded.values['api_key'],
"check that the user's api_key is the same as the one generated")
