diff --git a/TaskC03/solution.py b/TaskC03/solution.py new file mode 100644 index 0000000..4858bb0 --- /dev/null +++ b/TaskC03/solution.py @@ -0,0 +1,13 @@ +import re +import sys + +def is_valid_nip(line: str) -> bool: + # matching either: + # xxx-xxx-xx-xx OR xxx-xx-xx-xxx + p = re.compile(r'^(?:\d{3}-\d{3}-\d{2}-\d{2}|\d{3}-\d{2}-\d{2}-\d{3})$') + return bool(p.match(line)) + +if __name__ == "__main__": + for line in sys.stdin: + stripped_line = line.strip() + print("yes" if is_valid_nip(stripped_line) else "no") diff --git a/TaskC03/test.out b/TaskC03/test.out new file mode 100644 index 0000000..29fe7aa --- /dev/null +++ b/TaskC03/test.out @@ -0,0 +1,5 @@ +yes +yes +yes +no +no