import asyncio
import os
from enum import Enum

import databases
import ormar
import sqlalchemy

database_url = os.environ["DATABASE_URL"]

database = databases.Database(url=database_url)
metadata = sqlalchemy.MetaData()


class Country(int, Enum):
    UK = 1
    US = 2
    VIETNAM = 3
    COLOMBIA = 4


class ArtistV20(ormar.Model):
    ormar_config = ormar.OrmarConfig(
        database=database,
        metadata=metadata,
        tablename="artists_v20",
    )

    id: int = ormar.Integer(primary_key=True)
    name: str = ormar.String(max_length=100)
    country: Country = ormar.Enum(enum_class=Country)


async def main():
    if not database.is_connected:
        await database.connect()

    engine = sqlalchemy.create_engine(database_url)
    metadata.create_all(engine)

    await ArtistV20.objects.all(name="Jane Doe", country=Country.UK)

    if database.is_connected:
        await database.disconnect()


if __name__ == "__main__":
    asyncio.run(main())