from models.dbmodel.demo_model import IRecord, Record, Session, Depends, get_db

class Demo(object):
    def getalldemodata(db: Session):
        users = db.query(Record).all()
        db.close()
        return users

    async def savedemodata(db: Session, data: IRecord):
        rec = Record(date = data.date, country = data.country, cases = data.cases, deaths = data.deaths, recoveries = data.recoveries)
        db.add(rec)
        db.commit()
        db.refresh(rec)
        return rec