A "ModelPlus" subclass i'm working on to simplify how i work with my models

[chrisdewa]

So I've been using ODMantic for a while now and I'm experimenting with subclassing the default Model.
After I've defined my engine elsewhere and made it accessible to the class...

class ModelPlus(Model):

    @classmethod
    def quary(cls, **kwargs):
        return query.and_(*(getattr(cls, kw) == v for kw, v in kwargs.items()))

    @classmethod
    def get(cls, create=False, /, **kwargs):
        """
        Searches the collection for a document with specified keys and values
        Args:
            create (bool):
                If True a new instance will be created with kwargs
                This argument is positional only, to prevent collisions with document keys.
            **kwargs: key/values to search for

        Returns:
            The document if found, else a new instance with the attributes set
        Raises:
            AttributeError: If the model does not have the specified attribute set.
            ValidationError: If an object is to be created but required fields are missing
        """
        q = cls.quary(**kwargs)
        doc = await engine.find_one(cls, q)
        return doc or cls(**kwargs) if create else None

    async def save(self):
        """
        Persists the instance to the database
        Uses Upsert method
        """
        await engine.save(self)

And I use this new Model class to interact with the database.
Each instance now gets a save method and the class itself gets some cool classmethods as well.

I don't very much fancy making my own query expressions so I made a method that does them for me.

class Item(ModelPlus, Model):  # model is here so that engine doesn't complain about not running on a subclass of Model.
    a: int
    b: str

# elsewhere
q = Item.query(a=123, b='foo')
item = await engine.find_one(Item, q)

And I often search for a document and create an instance if it doesn't exist so I simplified it like this:

item = await Item.get(True, a=123, b='foo')

If an item doesn't exist with a == 123 and b == 'foo' then I get a new instance from it.

I'm still working on this, any ideas or suggestions?