def test_8percent_login():
username = os.environ.get('_8PERCENT_USERNAME')
password = os.environ.get('_8PERCENT_PASSWORD')
provider = _8Percent()
resp = provider.login(username, password)
assert 200 == resp.status_code
def test_8percent_fetch_data():
provider = _8Percent()
resp = provider.fetch_data(829)
assert resp.status_code in [200, 404]
if resp.status_code == 200:
sample_file = os.path.join(PROJECT_PATH, 'tests', 'data',
'8percent-829.html')
with open(sample_file, 'w') as fout:
fout.write(resp.text)
def test_8percent_parse_data():
sample_file = os.path.join(PROJECT_PATH, 'tests', 'data',
'8percent-829.html')
with open(sample_file) as fin:
raw = fin.read()
stored_data = [
('2016-04-11', 1694, 613, 160, 340),
('2016-05-11', 1916, 390, 90, 0),
('2016-06-13', 1920, 386, 90, 0),
('2016-07-11', 1982, 324, 80, 0),
('2016-08-10', 1963, 343, 80, 0),
('2016-09-12', 1979, 327, 80, 0),
('2016-10-10', 2005, 301, 70, 0),
('2016-11-14', 1992, 314, 70, 0),
('2016-12-12', 2054, 252, 60, 0),
('2017-01-10', 2044, 262, 60, 0),
('2017-02-13', 2053, 253, 60, 0),
('2017-03-13', 2099, 207, 50, 0),
('2017-04-10', 2101, 205, 50, 0),
('2017-05-15', 2098, 208, 50, 0),
('2017-06-12', 2145, 161, 40, 0),
('2017-07-10', 2151, 155, 30, 0),
('2017-08-14', 2153, 153, 30, 0),
('2017-09-11', 2188, 118, 20, 0),
('2017-10-11', 2198, 108, 20, 0),
('2017-11-13', 2216, 90, 20, 0),
('2017-12-11', 2238, 68, 10, 0),
('2018-01-10', 2251, 55, 10, 0),
('2018-02-12', 2270, 36, 0, 0),
('2018-03-12', 2290, 16, 0, 0),
]
provider = _8Percent()
parsed_data = provider.parse_data(raw)
assert parsed_data['name']
assert parsed_data['grade']
assert isinstance(parsed_data['duration'], int)
assert isinstance(parsed_data['annual_percentage_yield'], float)
assert 0.0 < parsed_data['annual_percentage_yield'] <= 0.3
assert isinstance(parsed_data['amount'], int)
assert 0 < parsed_data['amount']
flag = True
for expected, actual in zip(stored_data, parsed_data['records']):
assert len(expected) == len(actual)
expected = list(expected)
expected[0] = parse_date(expected[0])
for exp, act in zip(expected, actual):
flag = False
assert exp == act
if flag:
pytest.fail('parse_8percent_data() did not return any data')
def fetch_8percent(filename):
"""
:param filename: A file containing bond IDs
"""
with open(filename) as fin:
raw = fin.read()
bond_ids = [int(x) for x in re.findall(r'/my/repayment_detail/(\d+)', raw)]
provider = _8Percent()
provider.login()
for bond_id in bond_ids:
log.info('Fetching bond ID = {}', bond_id)
target_path = os.path.join(BASE_PATH, 'sample-data',
'8percent-{}.html'.format(bond_id))
resp = provider.fetch_data(bond_id)
with open(target_path, 'w') as fout:
fout.write(resp.text)
def import_8percent(filename):
"""Imports a single file."""
app = create_app(__name__)
provider = _8Percent()
with app.app_context():
with open(filename) as fin:
raw = fin.read()
account_8p = Account.query.filter(Account.name == '8퍼센트').first()
account_checking = Account.query.filter(
Account.name == '신한은행 입출금').first()
asset_krw = Asset.query.filter(Asset.name == 'KRW').first()
parsed_data = provider.parse_data(raw)
import_8percent_data(parsed_data,
account_checking=account_checking,
account_8p=account_8p,
asset_krw=asset_krw)