Add files via upload

Author: peisje

scripts/33_country_code.py

@@ -1,51 +1,55 @@
-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("myCSV.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)