added attach with background task
All checks were successful
checks-impure / test (pull_request) Successful in 30s
checks / test (pull_request) Successful in 3m18s

This commit is contained in:
Georg-Stahn
2023-12-03 16:56:16 +01:00
parent 09d0c28fd6
commit 9a31fbe010
3 changed files with 57 additions and 15 deletions

View File

@@ -1,6 +1,6 @@
from typing import List, Optional
from fastapi import APIRouter, Depends
from fastapi import APIRouter, BackgroundTasks, Depends
from sqlalchemy.orm import Session
from .. import sql_crud, sql_db, sql_models
@@ -139,8 +139,11 @@ def get_repository(
@router.post("/api/v1/create_entity", response_model=Entity, tags=[Tags.entities])
def create_entity(
entity: EntityCreate, db: Session = Depends(sql_db.get_db)
) -> EntityCreate:
) -> EntityCreate | str:
# todo checken ob schon da ...
if sql_crud.get_entity_by_did(db, did=entity.did):
print("did already exsists")
return "Error did already exsists in db"
return sql_crud.create_entity(db, entity)
@@ -161,3 +164,17 @@ def get_entity(
) -> Optional[sql_models.Entity]:
entity = sql_crud.get_entity_by_did(db, did=entity_did)
return entity
@router.get("/api/v1/attach", response_model=Optional[Entity], tags=[Tags.entities])
async def attach(
background_tasks: BackgroundTasks,
entity_did: str = "did:sov:test:1234",
skip: int = 0,
limit: int = 100,
db: Session = Depends(sql_db.get_db),
) -> dict[str, str]:
background_tasks.add_task(
sql_crud.check_health, db, entity_did, message="Not attched any more"
)
return {"message": "Attaching in the background"}

View File

@@ -1,3 +1,4 @@
import time
from typing import List, Optional
from sqlalchemy.orm import Session
@@ -136,3 +137,21 @@ def get_entities(
def get_entity_by_did(db: Session, did: str) -> Optional[sql_models.Entity]:
return db.query(sql_models.Entity).filter(sql_models.Entity.did == did).first()
def check_health(db: Session, entity_did: str, message: str) -> str:
# ste attached to true
db_entity = get_entity_by_did(db, entity_did)
if db_entity is not None:
# db_entity.attached = Column(True)
setattr(db_entity, "attached", True)
# save changes in db
db.add(db_entity)
db.commit()
db.refresh(db_entity)
# check every 5 secounds if status has changed
while db_entity.attached:
time.sleep(5)
return f"{entity_did} message"
else:
return f"{entity_did} not found and not attached"

View File

@@ -84,6 +84,7 @@ def test_producer(api: TestClient) -> None:
# get_request = "entity_did=did%3Asov%3Atest%3A1234"
make_test_post_and_get(api, request_body, paramter)
def test_producer2(api: TestClient) -> None:
request_body = {
"uuid": "8e285c0c-4e40-430a-a477-26b3b81e30d1",
@@ -98,6 +99,7 @@ def test_producer2(api: TestClient) -> None:
get_request = "entity_did=" + url.quote(default_entity_did2)
make_test_post_and_get(api, request_body, paramter, get_request)
def test_producer3(api: TestClient) -> None:
request_body = {
"uuid": "8e285c0c-4e40-430a-a477-26b3b81e30d2",
@@ -143,7 +145,6 @@ def test_producer5(api: TestClient) -> None:
make_test_post_and_get(api, request_body, paramter, get_request)
#########################
# #
# Consumer #
@@ -159,6 +160,7 @@ def test_consumer(api: TestClient) -> None:
# get_request = "entity_did=did%3Asov%3Atest%3A1234"
make_test_post_and_get(api, request_body, paramter)
def test_consumer2(api: TestClient) -> None:
request_body = {
"entity_did": default_entity_did2,
@@ -169,6 +171,7 @@ def test_consumer2(api: TestClient) -> None:
get_request = "entity_did=" + url.quote(default_entity_did2)
make_test_post_and_get(api, request_body, paramter, get_request)
#########################
# #
# REPOSITORY #
@@ -187,6 +190,8 @@ def test_repository(api: TestClient) -> None:
paramter = "repository"
# get_request = "entity_did=did%3Asov%3Atest%3A1234"
make_test_post_and_get(api, request_body, paramter)
def test_repository2(api: TestClient) -> None:
request_body = {
"uuid": "8e285c0c-4e40-430a-a477-26b3b81e30d1",
@@ -201,6 +206,7 @@ def test_repository2(api: TestClient) -> None:
get_request = "entity_did=" + url.quote(default_entity_did2)
make_test_post_and_get(api, request_body, paramter, get_request)
def test_repository3(api: TestClient) -> None:
request_body = {
"uuid": "8e285c0c-4e40-430a-a477-26b3b81e30d2",
@@ -246,7 +252,6 @@ def test_repository5(api: TestClient) -> None:
make_test_post_and_get(api, request_body, paramter, get_request)
#########################
# #
# Entity #
@@ -264,6 +269,7 @@ def test_entity(api: TestClient) -> None:
# get_request = "entity_did=did%3Asov%3Atest%3A1234"
make_test_post_and_get(api, request_body, paramter)
def test_entity2(api: TestClient) -> None:
request_body = {
"did": default_entity_did2,