def seed(dbpath=DBPATH): ORM.dbpath = dbpath mike_bloom = Account(username='******', balance=10000.00) mike_bloom.set_password('password') mike_bloom.api_key = '00000' mike_bloom.save() buy_trade = Trade(accounts_pk=mike_bloom.pk, ticker='tsla', volume=10, price=100.0) sell_trade = Trade(accounts_pk=mike_bloom.pk, ticker='tsla', volume=-5, price=200.0) buy_trade.save() sell_trade.save() tsla_position = Position(ticker='tsla', shares=5, accounts_pk=mike_bloom.pk) appl_position = Position(ticker='aapl', shares=5, accounts_pk=mike_bloom.pk) tsla_position.save() appl_position.save()
def seed(dbpath=DBPATH): ORM.dbpath = dbpath mike_bloom = Account(username='******', balance=10000.00) mike_bloom.set_password('password') mike_bloom.save() # trade for a purchase of 10 shares yesterday # trade for a sell of 5 shares today tsla_position = Position(ticker='tsla', shares=5, accounts_pk=mike_bloom.pk) tsla_position.save() ms_position = Position(ticker='ms', shares=10, accounts_pk=mike_bloom.pk) ms_position.save() tsla_trade = Trade( ticker='tsla', volume=5, price=95.20, time=time.time(), accounts_pk=mike_bloom.pk, ) tsla_trade.save() ms_trade = Trade( ticker='ms', volume=10, price=25.50, time=time.time(), accounts_pk=mike_bloom.pk, ) ms_trade.save()
def createaccount(): if not request.json or 'username' not in request.json or 'password' not in request.json: return jsonify(BADREQUEST), 400 account = Account() r = request.json account.username = r['username'] account.set_password(r['password']) account.set_apikey() account.balance = 0.00 account.save() return jsonify({"username": account.username, "apikey": account.api_key})
def test_key_creation(self): #Create new account record incl. api_key and save to DB new_user = Account(username="******") new_user.set_password("test_password") new_user.create_api_key() new_user.save() #Retrieve new account record and evaluate api_key pop'd pk = new_user.pk same_user = Account.one_from_pk(pk) self.assertEqual(len(same_user.api_key), 15, "api key populated")
def seed(dbpath=DBPATH): ORM.dbpath = dbpath print("DATABASE: ", dbpath) mike_bloom = Account(username='******', balance=10000.00) mike_bloom.set_password("password") print(mike_bloom.dbpath) mike_bloom.save() tsla_position = Position(ticker='tsla', shares=5, account_pk=mike_bloom.pk) tsla_position.save() fake_trade = Trade(ticker='tsla', quantity=2, type=1) fake_trade.save()
def create(): if not request.json or 'username' not in request.json or 'password_hash' not in request.json: return jsonify(BAD_REQUEST), 401 account = Account(username=request.json['username'], password_hash=request.json['password_hash']) hashed_pw = Account.set_password(account, request.json['password_hash']) account.set_api_key() account.save() token = encodeAuthToken(account.pk) return jsonify({ 'status': 'success', 'auth_token': str(token), 'api_key': account.api_key })