evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update country province validation (#84463)

Browse source 
* Update country `province` validation.

* Run pre-commit.

* Add tests

* Mod config flow

---------

Co-authored-by: G Johansson <goran.johansson@shiftit.se>
This commit is contained in:
Arkadii Yakovets 2023-08-18 12:20:04 -07:00 • committed by GitHub
parent f96446cb24
commit 7827f9ccae
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 48 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

15
homeassistant/components/workday/config_flow.py
View file

@ -3,8 +3,7 @@ from __future__ import annotations
from typing import Any
import holidays
from holidays import HolidayBase, list_supported_countries
from holidays import HolidayBase, country_holidays, list_supported_countries
import voluptuous as vol
from homeassistant.config_entries import (
@ -77,12 +76,14 @@ def validate_custom_dates(user_input: dict[str, Any]) -> None:
if dt_util.parse_date(add_date) is None:
raise AddDatesError("Incorrect date")
cls: HolidayBase = getattr(holidays, user_input[CONF_COUNTRY])
cls: HolidayBase = country_holidays(user_input[CONF_COUNTRY])
year: int = dt_util.now().year
obj_holidays = cls(
subdiv=user_input.get(CONF_PROVINCE), years=year, language=cls.default_language
) # type: ignore[operator]
obj_holidays: HolidayBase = country_holidays(
user_input[CONF_COUNTRY],
subdiv=user_input.get(CONF_PROVINCE),
years=year,
language=cls.default_language,
)
for remove_date in user_input[CONF_REMOVE_HOLIDAYS]:
if dt_util.parse_date(remove_date) is None: