Fixed lint errors

Author: mfateev

updatable_timer/starter.py

@@ -23,7 +23,9 @@ async def main(client: Optional[Client] = None):
         )
         logging.info(f"Workflow started: run_id={handle.result_run_id}")
     except exceptions.WorkflowAlreadyStartedError as e:
-        logging.info(f"Workflow already running: workflow_id={e.workflow_id}, run_id={e.run_id}")
+        logging.info(
+            f"Workflow already running: workflow_id={e.workflow_id}, run_id={e.run_id}"
+        )
 
 
 if __name__ == "__main__":

updatable_timer/updatable_timer_lib.py

@@ -5,7 +5,6 @@
 
 
 class UpdatableTimer:
-
     def __init__(self, wake_up_time: datetime) -> None:
         self.wake_up_time = wake_up_time
         self.wake_up_time_updated = False

updatable_timer/wake_up_time_updater.py

@@ -1,6 +1,6 @@
 import asyncio
 import logging
-from datetime import timedelta, datetime
+from datetime import datetime, timedelta
 from typing import Optional
 
 from temporalio.client import Client
@@ -12,11 +12,12 @@ async def main(client: Optional[Client] = None):
     logging.basicConfig(level=logging.INFO)
 
     client = client or await Client.connect("localhost:7233")
-    handle = client.get_workflow_handle(
-        workflow_id="updatable-timer-workflow"
-    )
+    handle = client.get_workflow_handle(workflow_id="updatable-timer-workflow")
     # signal workflow about the wake up time change
-    await handle.signal(Workflow.update_wake_up_time, (datetime.now() + timedelta(seconds=10)).timestamp())
+    await handle.signal(
+        Workflow.update_wake_up_time,
+        (datetime.now() + timedelta(seconds=10)).timestamp(),
+    )
 
     logging.info("Updated wake up time to 10 seconds from now")
 

updatable_timer/worker.py

@@ -15,9 +15,9 @@ async def main():
 
     client = await Client.connect("localhost:7233")
     async with Worker(
-            client,
-            task_queue=TASK_QUEUE,
-            workflows=[Workflow],
+        client,
+        task_queue=TASK_QUEUE,
+        workflows=[Workflow],
     ):
         logging.info("Worker started, ctrl+c to exit")
         # Wait until interrupted

updatable_timer/workflow.py

@@ -1,4 +1,5 @@
 from datetime import datetime, timezone
+from typing import Optional
 
 from temporalio import workflow
 
@@ -7,13 +8,14 @@
 
 @workflow.defn
 class Workflow:
-
     def __init__(self):
-        self.timer = None
+        self.timer: Optional[UpdatableTimer] = None
 
     @workflow.run
     async def run(self, wake_up_time: float):
-        self.timer = UpdatableTimer(datetime.fromtimestamp(wake_up_time, tz=timezone.utc))
+        self.timer = UpdatableTimer(
+            datetime.fromtimestamp(wake_up_time, tz=timezone.utc)
+        )
         await self.timer.sleep()
 
     @workflow.signal
@@ -22,8 +24,12 @@ async def update_wake_up_time(self, wake_up_time: float):
         # This happens when a workflow task is executed after a signal is received
         # or when a workflow is started using the signal-with-start.
         await workflow.wait_condition(lambda: self.timer is not None)
-        self.timer.update_wake_up_time(datetime.fromtimestamp(wake_up_time, tz=timezone.utc))
+        assert self.timer is not None  # for mypy
+        self.timer.update_wake_up_time(
+            datetime.fromtimestamp(wake_up_time, tz=timezone.utc)
+        )
 
     @workflow.query
     def get_wake_up_time(self):
+        assert self.timer is not None  # for mypy
         return self.timer.get_wake_up_time()