admin.py

# Admin program
'''
The admin program will read data from a JSON file saved locally and will initialize and upload the data
to a Google Firebase Cloud Datastore (not a Firebase Realtime Database). It takes a single command-line
argument: the name of the JSON file containing the data to load.
'''
import sys
import json
from connection_authentication import db

class Country:
    def __init__(self, country, region, area, population, gdp, coastline=None):
        """
        Country constructor

        :param country: string
        :param region: string
        :param area: number
        :param population: number
        :param gdp: number
        :param coastline: number
        """
        self.country = country.strip()
        self.region = region
        self.area = area
        self.population = population
        self.gdp = gdp
        self.coastline = coastline

    @classmethod
    def from_dict(cls, source):
        """
        Convert dictionary to a country

        :param source: dictionary
        :return: Country object
        """
        return cls(source['Country'], source['Region'], source['Area'], source['Population'], source['GDP'], source.get('Coastline'))

    def to_dict(self):
        """
        Convert country to a dictionary

        :return: dictionary
        """
        dictionary = {
            'Region': self.region,
            'Area': self.area,
            'Population': self.population,
            'GDP': self.gdp,
        }

        # only input coastline if it has a value
        if self.coastline != 0:
            dictionary['Coastline'] = self.coastline

        return dictionary

    def __repr__(self):
        return f"Country(\
                country={self.country}, \
                region={self.region}, \
                area={self.area}, \
                population={self.population}, \
                gdp={self.gdp}, \
                coastline={self.coastline}\
            )"

def populate_firebase(source):
    """
    Put countries into firebase

    :param source: json file of countries
    """
    try:
        with open(source, 'r') as f:
            country_data = json.load(f)

        collection_reference = db.collection("countries")
        for country_dict in country_data:
            country = Country.from_dict(country_dict)
            collection_reference.document(country.country).set(country.to_dict())

    except FileNotFoundError:
        print("Error: File not found")

if len(sys.argv) == 2:
    populate_firebase(sys.argv[1])
else:
    print("Please provide the file path")