Exemple #1
0
    pw=POSTGRESQL_PASSWORD,
    url=POSTGRESQL_URL,
    db=POSTGRESQL_DATABASE)
db = create_engine(db_string)
base = declarative_base()
print("db_string", db_string)
Session = sessionmaker(db)
session = Session()

countries = list(master_data_json.keys())

with open('/Users/sai/workspace/work/covid-api/data/country_name_to_iso.json',
          'r') as fp:
    country_name_to_code = json.loads(fp.read())

for country in countries:
    for everyday in master_data_json[country]:
        record = Records()
        record.uuid = uuid.uuid4()
        record.country_name = country
        record.country_iso = country_name_to_code.get(country)
        record.date = everyday["date"]
        record.confirmed = everyday["confirmed"]
        record.deaths = everyday["deaths"]
        record.recovered = everyday["recovered"]
        session.add(record)
        print("Record Object", record)
        session.commit()
        print(record)
        print(f"Successfully added record for {country}")