Uses current dir path for reading files

This commit is contained in:
Alicia Sykes
2023-04-16 13:06:26 +01:00
parent ed9e731e00
commit 9a764ed3c3
2 changed files with 13 additions and 7 deletions

View File

@@ -2,7 +2,11 @@ import os
import json import json
# Get list of files in external-templates # Get list of files in external-templates
files = os.listdir('../external-templates') dir = os.path.dirname(os.path.abspath(__file__))
templates_src_dir = os.path.join(dir, '../external-templates/')
template_dest_file = os.path.join(dir, '../templates.json')
files = os.listdir(templates_src_dir)
# Initialize empty list to store template objects # Initialize empty list to store template objects
templates = [] templates = []
@@ -10,7 +14,7 @@ templates = []
# For each file in external-templates # For each file in external-templates
for file in files: for file in files:
# Open the file # Open the file
with open('../external-templates/' + file) as f: with open(templates_src_dir + file) as f:
# Load the JSON into a variable # Load the JSON into a variable
data = json.load(f)['templates'] data = json.load(f)['templates']
# Append the template object to the templates list # Append the template object to the templates list
@@ -26,5 +30,5 @@ fileData = {
} }
# Open the templates.json file, and write results to it # Open the templates.json file, and write results to it
with open('../templates.json', 'w') as f: with open(template_dest_file, 'w') as f:
json.dump(fileData, f, indent=2, sort_keys=False) json.dump(fileData, f, indent=2, sort_keys=False)

View File

@@ -2,8 +2,10 @@ import os
import csv import csv
import requests import requests
destination_dir = '../external-templates' dir = os.path.dirname(os.path.abspath(__file__))
sources_list = '../sources.csv'
destination_dir = os.path.join(dir, '../external-templates')
sources_list = os.path.join(dir, '../sources.csv')
# Downloads the file from a given URL, to the local destination # Downloads the file from a given URL, to the local destination
def download(url: str, filename: str): def download(url: str, filename: str):
@@ -23,7 +25,7 @@ def download(url: str, filename: str):
# Gets list of URLs to download from CSV file # Gets list of URLs to download from CSV file
def get_source_list(): def get_source_list():
sources=[] sources=[]
with open('../sources.csv', mode='r') as file: with open(sources_list, mode='r') as file:
csvFile = csv.reader(file) csvFile = csv.reader(file)
for lines in csvFile:# for lines in csvFile:#
sources.append(lines) sources.append(lines)
@@ -33,6 +35,6 @@ def get_source_list():
if not os.path.exists(destination_dir): if not os.path.exists(destination_dir):
os.makedirs(destination_dir) os.makedirs(destination_dir)
# # For each source, download the templates JSON file # For each source, download the templates JSON file
for sourceUrl in get_source_list(): for sourceUrl in get_source_list():
download(sourceUrl[1], sourceUrl[0] + '.json') download(sourceUrl[1], sourceUrl[0] + '.json')