Skip to content

patch: domain & url modules #339

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Mar 22, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions CHANGES.md
Original file line number Diff line number Diff line change
Expand Up @@ -147,19 +147,21 @@ _**Breaking**_ ⚠️
- `extremes.py` renamed to `_extremes.py` and is no longer exposed.
- `truthy` was discarded in favour of simple `bool()` function.
- `ipv4_cidr()` and `ipv6_cidr()` has been dropped in favour of `cidr: bool = True` and `cidr: bool = True` keyword-only parameters.
- `email` API now accepts the following keyword-only arguments:
- `email()` API now accepts the following keyword-only arguments:
- `simple_host: bool = False`,
- `ipv6_address: bool = False`,
- `ipv4_address: bool = False`,
- `rfc_1034: bool = False` and
- `rfc_2782: bool = False`.
- `url` has been refactored, it accepts the following keyword-only arguments:
- `whitelist=None` has been removed from `email()`.
- `url()` has been refactored, it accepts the following keyword-only arguments:
- `skip_ipv6_addr: bool = False`,
- `skip_ipv4_addr: bool = False`,
- `may_have_port: bool = True`,
- `simple_host: bool = False`,
- `rfc_1034: bool = False` and
- `rfc_2782: bool = False`.
- `public=False` keyword argument has been removed from `url()`.
- Exposes `i18n` functions directly via `__init__.py`.
- `@validator` decorator catches `Exception`.

Expand Down
2 changes: 1 addition & 1 deletion src/validators/domain.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def domain(value: str, /, *, rfc_1034: bool = False, rfc_2782: bool = False):
# First character of the domain
rf"^(?:[a-zA-Z0-9{'_'if rfc_2782 else ''}]"
# Sub domain + hostname
+ r"(?:[a-zA-Z0-9-_]{0,61}[A-Za-z0-9])?\.)"
+ rf"(?:[a-zA-Z0-9-_]{{0,61}}[A-Za-z0-9{'_'if rfc_2782 else ''}])?\.)"
# First 61 characters of the gTLD
+ r"+[A-Za-z0-9][A-Za-z0-9-_]{0,61}"
# Last character of the gTLD
Expand Down
23 changes: 20 additions & 3 deletions src/validators/url.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,14 @@ def _validate_scheme(value: str):
"""Validate scheme."""
# More schemes will be considered later.
return (
value in {"ftp", "ftps", "git", "http", "https", "rtsp", "sftp", "ssh", "telnet"}
value
# fmt: off
in {
"ftp", "ftps", "git", "http", "https",
"rtmp", "rtmps", "rtsp", "sftp",
"ssh", "telnet",
}
# fmt: on
if value
else False
)
Expand Down Expand Up @@ -112,8 +119,18 @@ def _validate_optionals(path: str, query: str, fragment: str, strict_query: bool
optional_segments = True
if path:
optional_segments &= bool(_path_regex().match(path))
if query and parse_qs(query, strict_parsing=strict_query):
optional_segments &= True
try:
if (
query
# ref: https://github.com/python/cpython/issues/117109
and parse_qs(query, strict_parsing=strict_query, separator="&")
and parse_qs(query, strict_parsing=strict_query, separator=";")
):
optional_segments &= True
except TypeError:
# for Python < v3.9.2 (official v3.10)
if query and parse_qs(query, strict_parsing=strict_query):
optional_segments &= True
if fragment:
# See RFC3986 Section 3.5 Fragment for allowed characters
optional_segments &= bool(re.fullmatch(r"[0-9a-zA-Z?/:@\-._~%!$&'()*+,;=]*", fragment))
Expand Down
1 change: 1 addition & 0 deletions tests/test_domain.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
("11.com", False, False),
("3.cn.", True, False),
("_example.com", False, True),
("example_.com", False, True),
("a.cn", False, False),
("sub1.sub2.sample.co.uk", False, False),
("somerandomexample.xn--fiqs8s", False, False),
Expand Down
1 change: 1 addition & 0 deletions tests/test_url.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,6 +152,7 @@ def test_returns_true_on_valid_url(value: str):
"http://[2010:836B:4179::836B:4179",
"http://2010:836B:4179::836B:4179",
"http://2010:836B:4179::836B:4179:80/index.html",
"https://example.org?q=search');alert(document.domain);",
"https://www.example.com/foo/?bar=baz&inga=42&quux",
"https://foo.com/img/bar/baz.jpg?-62169987208",
"https://foo.bar.net/baz.php?-/inga/test-lenient-query/",
Expand Down