v/wideq
1
0
Fork 0
mirror of https://github.com/no2chem/wideq.git synced 2025-05-16 23:30:10 -07:00
Code Issues Projects Releases Wiki Activity

Error out when language or country are invalid

Browse Source 
Many people seem to stumble over this (I did, specifying en_US instead
of en-US for example). I don't know of any valid inputs other than two
letter combinations, but three letters could be valid.
This commit is contained in:
Frederik Gladhorn 2020-01-13 13:28:08 +01:00
parent 6461596219
commit cc3d60f882
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
example.py
View File

@ -5,6 +5,8 @@ import json
import time import time
import argparse import argparse
import sys import sys
import re
from typing import List
STATE_FILE = 'wideq_state.json' STATE_FILE = 'wideq_state.json'
@ -169,7 +171,7 @@ def example_command(client, cmd, args):
func(client, *args) func(client, *args)
def example(country, language, cmd, args): def example(country: str, language: str, cmd: str, args: List[str]) -> None:
# Load the current state for the example. # Load the current state for the example.
try: try:
with open(STATE_FILE) as f: with open(STATE_FILE) as f:
@ -207,7 +209,7 @@ def example(country, language, cmd, args):
json.dump(state, f) json.dump(state, f)
def main(): def main() -> None:
"""The main command-line entry point. """The main command-line entry point.
""" """
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
@ -220,14 +222,28 @@ def main():
parser.add_argument( parser.add_argument(
'--country', '-c', '--country', '-c',
help=f'country code for account (default: {wideq.DEFAULT_COUNTRY})' help=f'country code for account (default: {wideq.DEFAULT_COUNTRY})',
default=wideq.DEFAULT_COUNTRY
) )
parser.add_argument( parser.add_argument(
'--language', '-l', '--language', '-l',
help=f'language code for the API (default: {wideq.DEFAULT_LANGUAGE})' help=f'language code for the API (default: {wideq.DEFAULT_LANGUAGE})',
default=wideq.DEFAULT_LANGUAGE
) )
args = parser.parse_args() args = parser.parse_args()
country_regex = re.compile(r"^[A-Z]{2,3}$")
if not country_regex.match(args.country):
print("Error: Country must be two or three letters" \
f" all upper case (e.g. US, NO, KR) got: '{args.country}'",
file=sys.stderr)
exit(1)
language_regex = re.compile(r"^[a-z]{2,3}-[A-Z]{2,3}$")
if not language_regex.match(args.language):
print("Error: Language must be a combination of language" \
" and country (e.g. en-US, no-NO, kr-KR)" \
f" got: '{args.language}'",
file=sys.stderr)
exit(1)
example(args.country, args.language, args.cmd, args.args) example(args.country, args.language, args.cmd, args.args)