コード例 #1
0
ファイル: seed.py プロジェクト: Ssawalha/Deployable-TTrader
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()
コード例 #2
0
ファイル: routes.py プロジェクト: Ssawalha/W3-TTrader
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"})
コード例 #3
0
    def test_encode_auth_token(self):
        account = Account(email="*****@*****.**", password="******")

        account.save()
        auth_token = account.encode_auth_token()

        assert isinstance(auth_token, bytes)
コード例 #4
0
 def post(self):
     # get the post data
     post_data = request.get_json()
     # check if account already exists
     account = Account.query.filter_by(
         email=post_data.get('email')).one_or_none()
     if not account:
         try:
             account = Account(email=post_data.get('email'),
                               password=post_data.get('password'))
             account.save()
             # generate auth token
             aut_token = account.encode_auth_token()
             response = {
                 'status': 'success',
                 'message': 'Successfully registered.',
                 'auth_token': aut_token.decode()
             }
             return make_response(jsonify(response)), 201
         except Exception as e:
             response = {
                 'status': 'fail',
                 'message': 'Some error occurred. Please try again.'
             }
             return make_response(jsonify(response)), 401
     else:
         response = {
             'status': 'fail',
             'message': 'User already exists. Please log in.'
         }
         return make_response(jsonify(response)), 202
コード例 #5
0
 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")
コード例 #6
0
    def test_register_with_already_registered_account(self):
        """Test registration with already registered email."""

        account = Account(
            email='*****@*****.**',
            password='******'
        )
        account.save()

        with self.client:
            response = register_account(self, '*****@*****.**', '123456')
            assert response.content_type == 'application/json'
            assert response.status_code == 202

            data = json.loads(response.data.decode())
            assert data['status'] == 'fail'
            assert data['message'] == 'User already exists. Please log in.'
コード例 #7
0
    def test_save_and_pk_load(self):
        user = Account(username="******")
        user.save()
        self.assertIsInstance(user.values['pk'], int, 'save sets pk')

        pk = user.values['pk']
        same_user = Account.one_from_pk(pk)

        self.assertIsInstance(same_user, Account,
                              "one_from_pk loads an Account object")

        self.assertEqual(same_user.values['username'], "Greg",
                         "save creates database row")
        same_user.values['username'] = "******"
        same_user.save()

        same_again = Account.one_from_pk(pk)

        self.assertEqual(same_again.values['username'], "Gregory",
                         "save updates an existing row")
コード例 #8
0
    def test_save_and_pk_load(self):
        user = Account(username='******')
        user.save()
        self.assertIsInstance(
            user.values['pk'], int, 'save sets pk'
        )  #assert will always return a boolean - if we get false back, something is broken

        pk = user.values['pk']
        same_user = Account.one_from_pk(pk)

        self.assertIsInstance(same_user, Account,
                              'one_from_pk loads an Account object')

        self.assertEqual(same_user.values['username'], 'Sami',
                         'save creates database row')
        same_user.values['username'] = '******'
        same_user.save()
        same_again = Account.one_from_pk(pk)

        self.assertEqual(same_again.values['username'], 'Gregory',
                         'save updates an exisiting row')
コード例 #9
0
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 'first_name' not in request.json or 'last_name' not in request.json:
        return jsonify({'error': 'bad request'}), 400
    username = request.json['username']
    password = request.json['password']
    first = request.json['first_name']
    last = request.json['last_name']
    pword_hash = hash_password(password)
    new_account = Account(username=username,
                          password_hash=pword_hash,
                          balance=10000,
                          first=first,
                          last=last)
    new_account.generate_api_key
    new_account.save()
    return jsonify(
        {
            "Your account was created successfully.":
            "You start with a balance of $10,000."
        }, {"Your api key is": new_account.values['api_key']})