Skip to content

Commit dd988db

Browse files
committed
chore: clean up types and cleanup async task
1 parent 9ce3ed2 commit dd988db

3 files changed

Lines changed: 9 additions & 5 deletions

File tree

gitops/common/app.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -143,12 +143,12 @@ def image_prefix(self) -> str:
143143
@property
144144
def cluster(self) -> str:
145145
cluster = self.values.get("cluster", "")
146-
return str(cluster) if cluster else ""
146+
return str(cluster)
147147

148148
@property
149149
def tags(self) -> list[str]:
150150
tags = self.values.get("tags", [])
151-
return list(tags) if tags else []
151+
return list(tags)
152152

153153
@property
154154
def service_account_name(self) -> str:
@@ -166,7 +166,7 @@ def service_account_name(self) -> str:
166166
def secrets(self) -> dict[str, str]:
167167
# TODO: This should be a first class property
168168
secrets = self.values.get("secrets", {})
169-
return dict(secrets) if secrets else {}
169+
return dict(secrets)
170170

171171

172172
class Chart:

gitops_server/app.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,13 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
1717

1818
loop = asyncio.get_running_loop()
1919
deploy_queue_worker = DeployQueueWorker.get_worker()
20-
deploy_queue_worker.task = asyncio.ensure_future(deploy_queue_worker.run(), loop=loop)
20+
deploy_queue_worker_task = asyncio.ensure_future(deploy_queue_worker.run(), loop=loop)
2121
yield
22+
deploy_queue_worker_task.cancel()
23+
try:
24+
await deploy_queue_worker_task
25+
except asyncio.CancelledError:
26+
pass
2227

2328

2429
app = FastAPI(lifespan=lifespan)

gitops_server/workers/deployer/worker.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@ def __init__(self) -> None:
3333
self.queue: asyncio.Queue[Any] = asyncio.Queue()
3434
self.semaphore_manager = AppSemaphoreManager()
3535
self.active_deployments: set[asyncio.Task[None]] = set()
36-
self.task: asyncio.Task[None] | None = None
3736

3837
async def enqueue(self, work: Any) -> None:
3938
"""Enqueue an item of work for future processing.

0 commit comments

Comments
 (0)