From ac3268e6c8f368ad420324b6152ca27371fbfe04 Mon Sep 17 00:00:00 2001
From: ewandor <ewandor@dorfsvald.net>
Date: Wed, 8 Mar 2023 15:07:39 +0100
Subject: [PATCH] Endding entity_id to user model

---
 back/app/user/models.py  | 3 ++-
 back/app/user/schemas.py | 7 ++-----
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/back/app/user/models.py b/back/app/user/models.py
index 81fa0dbe..c2b74116 100644
--- a/back/app/user/models.py
+++ b/back/app/user/models.py
@@ -1,6 +1,6 @@
 from typing import Optional, TypeVar
 from datetime import datetime
-from pydantic import BaseModel, Field
+from pydantic import Field
 from beanie import PydanticObjectId
 
 from fastapi_users.db import BeanieBaseUser, BeanieUserDatabase
@@ -15,6 +15,7 @@ class AccessToken(BeanieBaseAccessToken[PydanticObjectId]):
 
 class User(BeanieBaseUser[PydanticObjectId]):
     login: str
+    entity_id: str
     created_at: datetime = Field(default=datetime.utcnow(), nullable=False)
     updated_at: datetime = Field(default_factory=datetime.utcnow, nullable=False)
 
diff --git a/back/app/user/schemas.py b/back/app/user/schemas.py
index 5c6e25d5..62943b8b 100644
--- a/back/app/user/schemas.py
+++ b/back/app/user/schemas.py
@@ -1,10 +1,6 @@
-import uuid
-from typing import TypeVar
-
-from pydantic import BaseModel, Field
+from pydantic import BaseModel
 from fastapi_users import schemas
 
-from ..core.schemas import Reader
 from .models import User
 
 
@@ -24,6 +20,7 @@ class UserCreate(UserBase):
     login: str
     password: str
     email: str
+    entity_id: str
 
 
 class UserUpdate(UserBase):