diff options
Diffstat (limited to 'real.py')
| -rw-r--r-- | real.py | 16 |
1 files changed, 8 insertions, 8 deletions
@@ -21,16 +21,16 @@ def write_data(data, site_id): f.write(data) -def write_html(data, site_id): +def write_html(data, site_id, stop_name, path): template = env.get_template('layout.html') - with open('{}.html'.format(site_id), 'w+') as f: - f.write(template.render(data=data)) + with open('{}/{}.html'.format(path, site_id), 'w+') as f: + f.write(template.render(data=data, stop_name=stop_name)) -def write_txt(data, site_id): +def write_txt(data, site_id, stop_name, path): template = env.get_template('layout.txt') - with open('{}.txt'.format(site_id), 'w+') as f: - f.write(template.render(data=data)) + with open('{}/{}.txt'.format(path, site_id), 'w+') as f: + f.write(template.render(data=data, stop_name=stop_name)) if __name__ == '__main__': @@ -53,5 +53,5 @@ if __name__ == '__main__': else: data = get_data(config['key'], config['site_id']) - write_html(data, config['site_id']) - write_txt(data, config['site_id']) + write_html(data, config['site_id'], config['stop_name'], config['output_path']) + write_txt(data, config['site_id'], config['stop_name'], config['output_path']) |