iframe-proxy | Sunbelt Computer Software

61 lines (43 loc) · 1.44 KB

from __future__ import annotations
import ipaddress
from typing import Optional
import tldextract
import validators
from email_validator import EmailNotValidError, validate_email
from .types import TargetType
def classify_target(target: str) -> TargetType:
    """Classify the user supplied target into one of the supported types."""
    target = target.strip()
    ip_obj = parse_ip(target)
    if ip_obj and not ip_obj.is_loopback:
        return TargetType.IP
    target_type = _classify_email(target)
    if target_type:
        return target_type
    target_type = _classify_domain(target)
    if target_type:
        return target_type
    return TargetType.USERNAME
def parse_ip(target: str) -> Optional[ipaddress._BaseAddress]:  # type: ignore[name-defined]
        return ipaddress.ip_address(target)
    except ValueError:
        return None
def is_private_ip(target: str) -> bool:
    ip = parse_ip(target)
    if not ip:
        return False
    return ip.is_loopback or ip.is_private
def _classify_email(target: str) -> Optional[TargetType]:
        validate_email(target)
        return TargetType.EMAIL
    except EmailNotValidError:
        return None
def _classify_domain(target: str) -> Optional[TargetType]:
    if validators.domain(target):
        return TargetType.DOMAIN
    extracted = tldextract.extract(target)
    if extracted.domain and extracted.suffix:
        return TargetType.DOMAIN
    return None

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

input_classifier.py

Latest commit

History

input_classifier.py

File metadata and controls