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")