Add cod_unidade_lotacao_participante to PlanoTrabalho

src/crud.py

@@ -148,6 +148,7 @@ async def create_plano_trabalho(
             .filter_by(origem_unidade=plano_trabalho.origem_unidade)
             .filter_by(cod_unidade_autorizadora=plano_trabalho.cod_unidade_autorizadora)
             .filter_by(matricula_siape=plano_trabalho.matricula_siape)
+            .filter_by(cod_unidade_lotacao=plano_trabalho.cod_unidade_lotacao_participante)
         )
         result = await session.execute(query)
         db_participante = result.scalars().unique().one_or_none()
@@ -156,7 +157,8 @@ async def create_plano_trabalho(
                 "Plano de Trabalho faz referência a participante inexistente. "
                 f"origem_unidade: {plano_trabalho.origem_unidade} "
                 f"cod_unidade_autorizadora: {plano_trabalho.cod_unidade_autorizadora} "
-                f"matricula_siape: {plano_trabalho.matricula_siape}"
+                f"matricula_siape: {plano_trabalho.matricula_siape} "
+                f"cod_unidade_lotacao: {plano_trabalho.cod_unidade_lotacao_participante}"
             )
         db_plano_trabalho.participante = db_participante
 

src/models.py

@@ -332,6 +332,15 @@ class PlanoTrabalho(Base):
         comment="Número da matrícula do participante no Sistema Integrado de "
         "Administração de Recursos Humanos (SIAPE).",
     )
+    cod_unidade_lotacao_participante = Column(
+        Integer,
+        nullable=False,
+        comment="Código da unidade organizacional (UORG) no Sistema Integrado de "
+        "Administração de Recursos Humanos (SIAPE) corresponde à unidade de "
+        "lotação do participante.\n\n"
+        "Obs: (1) A instituição que não esteja no SIAPE pode usar o código SIORG.\n"
+        "Obs: (2) Pode ser o mesmo ou diferente do `cod_unidade_executora`.",
+    )
     data_inicio = Column(
         Date,
         nullable=False,
@@ -378,14 +387,14 @@ class PlanoTrabalho(Base):
             [
                 origem_unidade,
                 cod_unidade_autorizadora,
-                cod_unidade_executora,
                 matricula_siape,
+                cod_unidade_lotacao_participante
             ],
             [
                 "participante.origem_unidade",
                 "participante.cod_unidade_autorizadora",
-                "participante.cod_unidade_lotacao",
                 "participante.matricula_siape",
+                "participante.cod_unidade_lotacao"
             ],
         ),
         UniqueConstraint(

src/schemas.py

@@ -249,6 +249,10 @@ class PlanoTrabalhoSchema(BaseModel):
         title="Matrícula SIAPE do participante",
         description=PlanoTrabalho.matricula_siape.comment,
     )
+    cod_unidade_lotacao_participante: int = Field(
+        title="Código da unidade lotacao participante",
+        description=PlanoTrabalho.cod_unidade_lotacao_participante.comment,
+    )
     data_inicio: date = Field(
         title="Data de início do plano de trabalho",
         description=PlanoTrabalho.data_inicio.comment,

tests/conftest.py

@@ -311,7 +311,7 @@ def example_part(client: httpx.Client, input_part: dict, header_admin: dict):
 @pytest.fixture()
 def example_part_2(client: httpx.Client, input_part: dict, header_admin: dict):
     """Cria um exemplo de status de participante com diferente SIAPE e lotação"""
-    input_part["cod_unidade_lotacao"] = 100
+    input_part["cod_unidade_lotacao"] = 99
     input_part["matricula_siape"] = "1234567"
     client.put(
         f"/organizacao/{input_part['origem_unidade']}"

tests/data/plano_trabalho.json

@@ -6,6 +6,7 @@
     "cod_unidade_executora": 99,
     "cpf_participante": "64635210600",
     "matricula_siape": "1237654",
+    "cod_unidade_lotacao_participante": 99,
     "data_inicio": "2023-01-01",
     "data_termino": "2023-01-15",
     "carga_horaria_disponivel": 80,

tests/plano_trabalho/core_test.py

@@ -27,6 +27,7 @@
         ["cod_unidade_executora"],
         ["cpf_participante"],
         ["matricula_siape"],
+        ["cod_unidade_lotacao_participante"],
         ["data_inicio"],
         ["data_termino"],
         ["carga_horaria_disponivel"],