Перевод обработки бд (database.py) на ORM

Есть файл database очень ужасен со всех сторон. В плане разработки тоже.
Задача хотя бы частично перевести его на ORM
Достаточно будет ну как минимум таблицу keys в модель обернуть. Не нужно прям весь файл переписывать