def update(output_dir='data'):
"""
Update data from AP.
"""
client = AP(SECRETS['AP_FTP_USER'], SECRETS['AP_FTP_PASSWORD'])
ticket = client.get_topofticket('2014-11-04')
write_update(ticket, '%s/update.json' % output_dir)
write_calls(ticket, '%s/calls.json' % output_dir)
def init(output_dir='data'):
"""
Initialize data from AP.
"""
client = AP(SECRETS['AP_FTP_USER'], SECRETS['AP_FTP_PASSWORD'])
ticket = client.get_topofticket('2014-11-04')
races = []
candidates = []
for race in ticket.races:
races.append(process_race(race))
for candidate in race.candidates:
candidates.append(process_candidate(candidate, race))
with open('%s/init_races.json' % output_dir, 'w') as f:
json.dump(races, f, indent=4)
with open('%s/init_candidates.json' % output_dir, 'w') as f:
json.dump(candidates, f, indent=4)
def test_badlogin(self):
client = AP("foo", "bar")
self.assertRaises(BadCredentialsError, client.get_state, os.environ['TEST_STATE'])
def setUp(self):
self.client = AP(
os.environ['AP_USERNAME'],
os.environ['AP_PASSWORD']
)
def test_badlogin(self):
client = AP("foo", "bar")
self.assertRaises(BadCredentialsError, client.get_state, "IA")
def setUp(self):
self.client = AP(AP_USERNAME, AP_PASSWORD)
