diff --git a/scripts/33_country_code.py b/scripts/33_country_code.py index 134236c..c8b6084 100644 --- a/scripts/33_country_code.py +++ b/scripts/33_country_code.py @@ -1,51 +1,62 @@ -import csv -import sys -import json - -""" -Example usage: - -$ python 33_country_code.py 33_sample_csv.csv 33_country_codes.json -""" - - -def get_data(csv_file, json_file): - countryCodes = [] - countryNames = [] - continentNames = [] - with open(csv_file, 'rt') as file_one: - reader = csv.reader(file_one) - with open(json_file) as file_two: - json_data = json.load(file_two) - all_countries = json_data["country"] - for csv_row in reader: - for json_row in all_countries: - if csv_row[0] == json_row["countryCode"]: - countryCodes.append(json_row["countryCode"]) - countryNames.append(json_row["countryName"]) - continentNames.append(json_row["continentName"]) - - return [ - countryCodes, - countryNames, - continentNames - ] - - -def write_data(array_of_arrays): - with open('data.csv', 'wt') as csv_out: - writer = csv.writer(csv_out) - rows = zip( - array_of_arrays[0], - array_of_arrays[1], - array_of_arrays[2] - ) - for row in rows: - writer.writerow(row) - - -if __name__ == '__main__': - csv_file_name = sys.argv[1] - json_file_name = sys.argv[2] - data = get_data(csv_file_name, json_file_name) - write_data(data) +import csv, json + + +with open("33_country_codes.json","r") as file: + data=json.load(file) + print(data) + + + + + + + + +codes = {"country": [ + { + "countryCode": "AD", + "countryName": "Andorra", + "continentName": "Europe" + }, + { + "countryCode": "AE", + "countryName": "United Arab Emirates", + "continentName": "Asia" + }, + { + "countryCode": "AF", + "countryName": "Afghanistan", + "continentName": "Asia" + } + ] + } + +codesc=[] +countries=[] +continrnt=[] + + +print(codes["country"][0]["countryName"]) + +for code in data["country"]: + codesc.append(code["countryCode"]) + countries.append(code["countryName"]) + continrnt.append(code["continentName"]) + + + +print(codesc) +print(countries) +print(continrnt) + + +def save_csv(codesc, countries, continrnt): + with open("myCSV1.csv", "w", newline="") as file: + writer=csv.writer(file) + writer.writerow(["country Code","country Name","continent Name"]) + print(codesc,countries,continrnt) + for i in range(1,len(countries)): + writer.writerow([codesc[i],countries[i],continrnt[i]]) + + +save_csv(codesc,countries,continrnt)