import enum
from typing import Dict, List

from odmantic.field import Field
from odmantic.model import Model


class TreeKind(str, enum.Enum):
    BIG = "big"
    SMALL = "small"


class TreeModel(Model):
    name: str = Field(primary_key=True, default="Acacia des montagnes")
    average_size: float = Field(mongo_name="size")
    discovery_year: int
    kind: TreeKind
    genesis_continents: List[str]
    per_continent_density: Dict[str, float]

[((221, 276), 'odmantic.field.Field', 'Field', ([], {'primary_key': '(True)', 'default': '"""Acacia des montagnes"""'}), "(primary_key=True, default='Acacia des montagnes')\n", (226, 276), False, 'from odmantic.field import Field\n'), ((303, 327), 'odmantic.field.Field', 'Field', ([], {'mongo_name': '"""size"""'}), "(mongo_name='size')\n", (308, 327), False, 'from odmantic.field import Field\n')]