From 5072dbb2d66ab0933afdb90050f70c62423c7daa Mon Sep 17 00:00:00 2001 From: James Richardson Date: Sat, 10 Feb 2024 14:16:58 +0000 Subject: [PATCH] typing --- gopro_overlay/widgets/msi.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/gopro_overlay/widgets/msi.py b/gopro_overlay/widgets/msi.py index fde7c11..465d6b5 100644 --- a/gopro_overlay/widgets/msi.py +++ b/gopro_overlay/widgets/msi.py @@ -1,3 +1,5 @@ +from typing import Callable + from PIL import Image, ImageDraw from .asi import roundup, scale, Arc @@ -6,7 +8,8 @@ class MotorspeedIndicator(Widget): - def __init__(self, size, font, needle, reading, green, yellow, end, rotate=180, outline=2): + def __init__(self, size: int, font, needle: int, reading: Callable[[], float], green: int, yellow: int, end: int, + rotate: int = 180, outline: int = 2): self.end = end self.yellow = yellow self.green = green @@ -115,7 +118,8 @@ def draw(self, image: Image, draw: ImageDraw): class MotorspeedIndicator2(Widget): - def __init__(self, size, font, reading, green, yellow, end, rotate=180, outline=2): + def __init__(self, size: int, font, reading: Callable[[], float], green: int, yellow: int, end: int, + rotate: int = 180, outline: int = 2): self.end = end self.yellow = yellow self.green = green