Python Vanguard.get_total_assets示例

示例#1

文件： test_vanguard.py 项目： rikonor/vanguard-api

class TestBrowser(TestCase):
    def setUp(self):
        self.v = Vanguard()

    def tearDown(self):
        self.v.close_browser()

    def test_can_login(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)

    def test_can_get_security_question(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)
        question = self.v.get_security_question()

    def test_can_answer_security_question(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)
        question = self.v.get_security_question()

        answer = tests_config.TEST_SECURITY_QUESTIONS.get(question)
        self.assertIsNotNone(answer)

        self.v.answer_security_question(answer)

    def test_can_go_to_balances_and_holdings(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)

        question = self.v.get_security_question()
        answer = tests_config.TEST_SECURITY_QUESTIONS.get(question)
        self.v.answer_security_question(answer)

        self.v.go_to_balances_and_holdings()
        self.assertEqual(self.v.browser.title, "Balances and holdings")

    def test_can_get_total_assets(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)

        question = self.v.get_security_question()
        answer = tests_config.TEST_SECURITY_QUESTIONS.get(question)
        self.v.answer_security_question(answer)

        self.v.get_total_assets()

    def test_can_get_current_holdings(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)

        question = self.v.get_security_question()
        answer = tests_config.TEST_SECURITY_QUESTIONS.get(question)
        self.v.answer_security_question(answer)

        self.v.get_current_holdings()

示例#2

class TestBrowser(TestCase):
    def setUp(self):
        self.v = Vanguard()

    def tearDown(self):
        self.v.close_browser()

    def test_can_login(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)

    def test_can_get_security_question(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)
        question = self.v.get_security_question()

    def test_can_answer_security_question(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)
        question = self.v.get_security_question()

        answer = tests_config.TEST_SECURITY_QUESTIONS.get(question)
        self.assertIsNotNone(answer)

        self.v.answer_security_question(answer)

    def test_can_go_to_balances_and_holdings(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)

        question = self.v.get_security_question()
        answer = tests_config.TEST_SECURITY_QUESTIONS.get(question)
        self.v.answer_security_question(answer)

        self.v.go_to_balances_and_holdings()
        self.assertEqual(self.v.browser.title, "Balances and holdings")

    def test_can_get_total_assets(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)

        question = self.v.get_security_question()
        answer = tests_config.TEST_SECURITY_QUESTIONS.get(question)
        self.v.answer_security_question(answer)

        self.v.get_total_assets()

    def test_can_get_current_holdings(self):
        self.v.login(tests_config.TEST_USER, tests_config.TEST_PASSWORD)

        question = self.v.get_security_question()
        answer = tests_config.TEST_SECURITY_QUESTIONS.get(question)
        self.v.answer_security_question(answer)

        self.v.get_current_holdings()

示例#3

def vanguard_total_assets():
    try:
        user = auth()
    except AuthenticationFailed:
        return rAuthError

    try:
        vanguard_user = VanguardUser(user)
    except KeyError:
        return rNotEnrolled

    res = None
    try:
        v = Vanguard()
        v.login(vanguard_user.username, vanguard_user.password)
        question = v.get_security_question()
        answer = vanguard_user.security_questions.get(question)
        v.answer_security_question(answer)

        total_assets = v.get_total_assets()
        res = dict(total=total_assets)
    finally:
        v.close_browser()
        if res is None:
            return rInternalServerError

    return jsonify(res)

示例#4

文件： api.py 项目： rikonor/vanguard-api

def vanguard_total_assets():
    try:
        user = auth()
    except AuthenticationFailed:
        return rAuthError

    try:
        vanguard_user = VanguardUser(user)
    except KeyError:
        return rNotEnrolled

    res = None
    try:
        v = Vanguard()
        v.login(vanguard_user.username, vanguard_user.password)
        question = v.get_security_question()
        answer = vanguard_user.security_questions.get(question)
        v.answer_security_question(answer)

        total_assets = v.get_total_assets()
        res = dict(total=total_assets)
    finally:
        v.close_browser()
        if res is None:
            return rInternalServerError

    return jsonify(res)