From 972c5577f4aafd820308cd79024b06b52f93ba38 Mon Sep 17 00:00:00 2001 From: StNicolay Date: Tue, 10 Jan 2023 21:00:03 +0300 Subject: [PATCH] Added validation to the database models --- src/db/models.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/src/db/models.py b/src/db/models.py index c00f13b..6b25e95 100644 --- a/src/db/models.py +++ b/src/db/models.py @@ -11,10 +11,14 @@ class MasterPass(sqlmodel.SQLModel, table=True): ) ) salt: bytes = sqlmodel.Field( - sa_column=sqlmodel.Column(sqlmodel.BINARY(64), nullable=False) + sa_column=sqlmodel.Column(sqlmodel.BINARY(64), nullable=False), + max_length=64, + min_length=64, ) password_hash: bytes = sqlmodel.Field( - sa_column=sqlmodel.Column(sqlmodel.BINARY(128), nullable=False) + sa_column=sqlmodel.Column(sqlmodel.BINARY(128), nullable=False), + max_length=128, + min_length=128, ) @@ -22,13 +26,17 @@ class Account(sqlmodel.SQLModel, table=True): __tablename__ = "accounts" __table_args__ = (sqlmodel.PrimaryKeyConstraint("user_id", "name"),) user_id: int = sqlmodel.Field() - name: str = sqlmodel.Field(max_length=255) + name: str = sqlmodel.Field(max_length=256) salt: bytes = sqlmodel.Field( - sa_column=sqlmodel.Column(sqlmodel.BINARY(64), nullable=False) + sa_column=sqlmodel.Column(sqlmodel.BINARY(64), nullable=False), + max_length=64, + min_length=64, ) enc_login: bytes = sqlmodel.Field( - sa_column=sqlmodel.Column(sqlmodel.VARBINARY(256), nullable=False) + sa_column=sqlmodel.Column(sqlmodel.VARBINARY(256), nullable=False), + max_length=256, ) enc_password: bytes = sqlmodel.Field( - sa_column=sqlmodel.Column(sqlmodel.VARBINARY(256), nullable=False) + sa_column=sqlmodel.Column(sqlmodel.VARBINARY(256), nullable=False), + max_length=256, )