63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
import databases
|
|
import sqlalchemy
|
|
|
|
import ormar
|
|
from tests.settings import DATABASE_URL
|
|
|
|
database = databases.Database(DATABASE_URL, force_rollback=True)
|
|
metadata = sqlalchemy.MetaData()
|
|
|
|
|
|
class Company(ormar.Model):
|
|
class Meta:
|
|
tablename = "companies"
|
|
metadata = metadata
|
|
database = database
|
|
|
|
id: int = ormar.Integer(primary_key=True)
|
|
name: str = ormar.String(max_length=100)
|
|
founded: int = ormar.Integer(nullable=True)
|
|
|
|
|
|
class Car(ormar.Model):
|
|
class Meta:
|
|
tablename = "cars"
|
|
metadata = metadata
|
|
database = database
|
|
|
|
id: int = ormar.Integer(primary_key=True)
|
|
manufacturer = ormar.ForeignKey(Company)
|
|
name: str = ormar.String(max_length=100)
|
|
year: int = ormar.Integer(nullable=True)
|
|
gearbox_type: str = ormar.String(max_length=20, nullable=True)
|
|
gears: int = ormar.Integer(nullable=True)
|
|
aircon_type: str = ormar.String(max_length=20, nullable=True)
|
|
|
|
|
|
# build some sample data
|
|
toyota = await Company.objects.create(name="Toyota", founded=1937)
|
|
await Car.objects.create(
|
|
manufacturer=toyota,
|
|
name="Corolla",
|
|
year=2020,
|
|
gearbox_type="Manual",
|
|
gears=5,
|
|
aircon_type="Manual",
|
|
)
|
|
await Car.objects.create(
|
|
manufacturer=toyota,
|
|
name="Yaris",
|
|
year=2019,
|
|
gearbox_type="Manual",
|
|
gears=5,
|
|
aircon_type="Manual",
|
|
)
|
|
await Car.objects.create(
|
|
manufacturer=toyota,
|
|
name="Supreme",
|
|
year=2020,
|
|
gearbox_type="Auto",
|
|
gears=6,
|
|
aircon_type="Auto",
|
|
)
|