From c00be9512faba05afe5314048b017432499818a6 Mon Sep 17 00:00:00 2001 From: Norman Hooper Date: Tue, 19 Nov 2024 19:43:07 +0000 Subject: [PATCH] Fix type hints --- corehq/apps/app_manager/xform_builder.py | 12 ++++++++++-- .../tests/test_elastic_sync_multiplexed_command.py | 2 +- corehq/apps/userreports/models.py | 2 ++ .../userreports/tests/test_data_source_config.py | 2 +- corehq/messaging/smsbackends/twilio/models.py | 5 ++++- 5 files changed, 18 insertions(+), 5 deletions(-) diff --git a/corehq/apps/app_manager/xform_builder.py b/corehq/apps/app_manager/xform_builder.py index 4ef430ef10b8..eb441cf8252e 100644 --- a/corehq/apps/app_manager/xform_builder.py +++ b/corehq/apps/app_manager/xform_builder.py @@ -18,8 +18,11 @@ """ +from __future__ import annotations + import re import uuid +from typing import Optional from lxml import etree from lxml.builder import E @@ -424,13 +427,18 @@ def get_input_question_node(name_, groups_=None, choices_=None, **params_): class Question(object): - def __init__(self, name, xform, groups=None): + def __init__( + self, + name: str, + xform: XFormBuilder, + groups: Optional[list[str]] = None, + ) -> None: self.name = name self.xform = xform self.groups = groups @property - def form(self) -> "XFormBuilder": + def form(self) -> XFormBuilder: return self.xform diff --git a/corehq/apps/es/tests/test_elastic_sync_multiplexed_command.py b/corehq/apps/es/tests/test_elastic_sync_multiplexed_command.py index b5cb876df069..5a5c7a018ed1 100644 --- a/corehq/apps/es/tests/test_elastic_sync_multiplexed_command.py +++ b/corehq/apps/es/tests/test_elastic_sync_multiplexed_command.py @@ -95,7 +95,7 @@ def setUpClass(cls): def tearDown(self) -> None: manager.cluster_routing(enabled=True) - return super().tearDown() + super().tearDown() def test_invalid_index_canonical_raises(self): with self.assertRaises(CommandError): diff --git a/corehq/apps/userreports/models.py b/corehq/apps/userreports/models.py index 5f2a4bbff4f9..604cc8540d8e 100644 --- a/corehq/apps/userreports/models.py +++ b/corehq/apps/userreports/models.py @@ -228,6 +228,8 @@ def is_this_data_source(rebuild_task): return args[0] == data_source_config_id def iter_tasks(): + if not flower_url: + return task_names = ( 'corehq.apps.userreports.tasks.rebuild_indicators', 'corehq.apps.userreports.tasks.rebuild_indicators_in_place', diff --git a/corehq/apps/userreports/tests/test_data_source_config.py b/corehq/apps/userreports/tests/test_data_source_config.py index 59491338b743..3632cd39c961 100644 --- a/corehq/apps/userreports/tests/test_data_source_config.py +++ b/corehq/apps/userreports/tests/test_data_source_config.py @@ -65,7 +65,7 @@ def setUpClass(cls) -> None: "type": "property_name" } cls.config = DataSourceConfiguration.wrap(cls.config) - return super().setUpClass() + super().setUpClass() def test_raises_when_domain_has_no_permission(self): self.config.domain = 'domain_nopermission' diff --git a/corehq/messaging/smsbackends/twilio/models.py b/corehq/messaging/smsbackends/twilio/models.py index 47732d3be4d0..6604841f1fda 100644 --- a/corehq/messaging/smsbackends/twilio/models.py +++ b/corehq/messaging/smsbackends/twilio/models.py @@ -131,7 +131,10 @@ def send(self, msg, orig_phone_number=None, *args, **kwargs): msg.backend_message_id = message.sid msg.save() - def from_or_messaging_service_sid(self, phone_number: str) -> (Optional[str], Optional[str]): + def from_or_messaging_service_sid( + self, + phone_number: str, + ) -> tuple[Optional[str], Optional[str]]: if self.phone_number_is_messaging_service_sid(phone_number): return None, phone_number else: