from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from lib.config_lib import get_engine

Base = declarative_base()


class covidData(Base):
    __tablename__ = 'data'
    selection = Column(String(32), primary_key=True)
    selection_original = Column(String(32))
    total_cases = Column(Integer)
    new_cases = Column(Integer)
    total_deaths = Column(Integer)
    new_deaths = Column(Integer)
    total_recovered = Column(Integer)
    active_cases = Column(Integer)
    serious_critical = Column(Integer)
    total_cases_per_one_mil = Column(String(32))


engine = get_engine()
Base.metadata.create_all(engine)