FdD ddlZddlZddlmZmZddlmZmZddlmZdedefdefdZ d e de fd Z d e de fd Z d e de fd ZdS) N)AnyCallable)urljoinurlparse)rfunc.returncdtjdtdtffd }|S)zqThe method decorator validate_url is used to validate urls for any command that requires a url as an argumenturlrc:tjd|stdt|stdt |rtdt |dkrtdt |g|Ri|S)a'Check if the URL is valid using a basic check, urllib check, and local file check Args: url (str): The URL to check Returns: the result of the wrapped function Raises: ValueError if the url fails any of the validation tests z ^https?://zInvalid URL formatz"Missing Scheme or Network locationz#Access to local files is restrictedizURL is too long)rematch ValueError is_valid_urlcheck_local_file_accesslen sanitize_url)r argskwargsrs PC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\url_utils\validators.pywrapperzvalidate_url..wrapper sx s++ 3122 2C   CABB B "3 ' ' DBCC C s88d??.// /tL%%7777777) functoolswrapsstrr)rrs` r validate_urlr sM_T8S8c8888884 Nrr cz t|}t|j|jgS#t$rYdSwxYw)zCheck if the URL is valid Args: url (str): The URL to check Returns: bool: True if the URL is valid, False otherwise F)rallschemenetlocr)r results rrr+sK#FM6=1222 uus ), ::cpt|}|j|jd|j}t ||S)ztSanitize the URL Args: url (str): The URL to sanitize Returns: str: The sanitized URL ?)rpathparamsqueryr)r parsed_urlreconstructed_urls rrr;sA#J%?RJ,=RR @PRR 3) * **rcBgd}tfd|DS)zCheck if the URL is a local file Args: url (str): The URL to check Returns: bool: True if the URL is a local file, False otherwise )zfile:///zfile://localhost/zfile://localhostzhttp://localhostzhttp://localhost/zhttps://localhostzhttps://localhost/zhttp://2130706433zhttp://2130706433/zhttps://2130706433zhttps://2130706433/zhttp://127.0.0.1/zhttp://127.0.0.1zhttps://127.0.0.1/zhttps://127.0.0.1zhttps://0.0.0.0/zhttps://0.0.0.0zhttp://0.0.0.0/zhttp://0.0.0.0z http://0000z http://0000/z https://0000z https://0000/c3BK|]}|VdS)N) startswith).0prefixr s r z*check_local_file_access..ks/CC&s~~f%%CCCCCCr)any)r local_prefixess` rrrIs9N2 CCCCNCCC C CCr)rr typingrr urllib.parserrrequests.compatrrboolrrrrrr5s ********######xS)cD c d     +c +c + + + +"D"D"D"D"D"D"D"Dr