U dG@sdZddlZddlZddlmZmZmZddlmZmZddl m Z ddl Z ddl Z eddde j dZedddd e j dZed ejZed ejZed Zd dZddde_ddZddde_ddZddde_GdddeZeddddddZGdddeZGdddeZddZd d!Zd"d#Z d?d%d&Z!dd'de!_d(d)Z"dd)de"_d*d+Z#d@d,d-Z$d.dd/e$_dAd0d1Z%d.ddd2e%_Gd3d4d4eZ&d5d6Z'd7d6ie'_d8d9Z(d:d;Z)ddZ*dd)de*_dS)Ba This module provide some helpers for advanced types parsing. You can define you own parser using the same pattern: .. code-block:: python def my_type(value): if not condition: raise ValueError('This is not my type') return parse(value) # Swagger documentation my_type.__schema__ = {'type': 'string', 'format': 'my-custom-format'} The last line allows you to document properly the type in the Swagger documentation. N)datetimetime timedelta) parsedate_tz mktime_tz)urlparsetzinfo;i?Ba(?:(?P[^:@]+?(?::[^:@]*?)?)@)?(?:(?Plocalhost)|(?P\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})|(?:\[?(?P[A-F0-9]*:[A-F0-9:]+)\]?)|(?P(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)))(?::(?P\d+))?$z9^(?P[^@]*[^@.])@(?P[^@\.]+(?:\.[^@\.]+)*)$z \d{2}:\d{2}cCsLz"t||ddkr |WSWntjk r8YnXtd|dS)zValidate an IPv4 address.{0} is not a valid ipv4 addressN)socket inet_atoncounterror ValueErrorformatvaluer6/tmp/pip-unpacked-wheel-dt_sn2ih/flask_restx/inputs.pyipv48s  rstringtypercCs>zttj||WStjk r8td|YnXdS)zValidate an IPv6 addressrN)r inet_ptonAF_INET6rrrrrrripv6Fs rcCsRz t|WStk rYnXz t|WStk rLtd|YnXdS)z+Validate an IP address (both IPv4 and IPv6)z{0} is not a valid ipN)rrrrrrrripRs  r c@s8eZdZdZd ddZd ddZdd Zed d ZdS)URLaP Validate an URL. Example:: parser = reqparse.RequestParser() parser.add_argument('url', type=inputs.URL(schemes=['http', 'https'])) Input to the ``URL`` argument will be rejected if it does not match an URL with specified constraints. If ``check`` is True it will also be rejected if the domain does not exists. :param bool check: Check the domain exists (perform a DNS resolution) :param bool ip: Allow IP (both ipv4/ipv6) as domain :param bool local: Allow localhost (both string or ip) as domain :param bool port: Allow a port to be present :param bool auth: Allow authentication to be present :param list|tuple schemes: Restrict valid schemes to this list :param list|tuple domains: Restrict valid domains to this list :param list|tuple exclude: Exclude some domains FNc Cs4||_||_||_||_||_||_||_||_dSN)checkr localportauthschemesdomainsexclude) selfr#r r$r%r&r'r(r)rrr__init__xs z URL.__init__cCs(d}|rd||f}t||dS)Nz{0} is not a valid URLz. )joinrr)r*rdetailsmsgrrrrsz URL.errorc Cst|}t|j}t|j|jfsdt|jpJ|jddddddrZ||d|||jr|j r|j|j kr||d|s||| }|ds|drd|j s||d nNzt |dp|dWn4t k r}z||t |W5d}~XYnX|js\|drB|dd rB||d n|dd kr\||d |jrd|d r|js||d|dr|js||d |dr|js||dn2t|d}d|krdksn||d|dr|jr"|d|jkr"||dn$|jrF|d|jkrF||d|jrzt|ddWn$tjk r||dYnX|S)N/r?zDid you mean: http://{0}zProtocol is not allowedrrzIP is not allowed127.zLocalhost is not allowed::1r&zAuthentication is not allowed localhostr%zCustom port is not allowedizPort is out of rangedomainzDomain is not allowedzDomain does not exists)r netloc_regexmatchnetlocallschemepathsplitrr' groupdictr rstrr$ startswithr#r&r%intr(r)r getaddrinfo)r*rparsedZ netloc_matchdataer%rrr__call__s^ "    "        z URL.__call__cCs dddS)Nrurlrrr*rrr __schema__szURL.__schema__)FFFFFNNN)N) __name__ __module__ __qualname____doc__r+rrEpropertyrHrrrrr!as  4r!T)httphttpsftpZftps)r r&r%r$r'c@s@eZdZdZdddZdddZdd Zd d Zed d Z dS)emailaN Validate an email. Example:: parser = reqparse.RequestParser() parser.add_argument('email', type=inputs.email(dns=True)) Input to the ``email`` argument will be rejected if it does not match an email and if domain does not exists. :param bool check: Check the domain exists (perform a DNS resolution) :param bool ip: Allow IP (both ipv4/ipv6) as domain :param bool local: Allow localhost (both string or ip) as domain :param list|tuple domains: Restrict valid domains to this list :param list|tuple exclude: Exclude some domains FNcCs"||_||_||_||_||_dSr")r#r r$r(r))r*r#r r$r(r)rrrr+s zemail.__init__cCs|pd}t||dS)Nz{0} is not a valid email)rrr*rr.rrrrsz email.errorcCs*zt|WdStk r$YdSXdS)NTF)r r)r*rrrris_ips z email.is_ipcCst|}|rd|kr |||d}|jrbzt|dWn tjk r`||YnX|jr~||jkr~||d|jr||jkr||d|j s|dks| dr||| |r|j s|||S)Nz..serverz-{0} does not belong to the authorized domainsz!{0} belongs to a forbidden domain)r4r3r2) email_regexr7rgroupr#rrAr(r)r$r?rSr )r*rr7rTrrrrEs,        zemail.__call__cCs dddS)NrrQrrrGrrrrHszemail.__schema__)FFFNN)N) rIrJrKrLr+rrSrErMrHrrrrrQs  rQc@s4eZdZdZddZddZddZedd Zd S) regexa` Validate a string based on a regular expression. Example:: parser = reqparse.RequestParser() parser.add_argument('example', type=inputs.regex('^[0-9]+$')) Input to the ``example`` argument will be rejected if it contains anything but numbers. :param str pattern: The regular expression the input must match cCs||_t||_dSr")patternrecompile)r*rXrrrr+)szregex.__init__cCs$|j|s d|j}t||S)Nz#Value does not match pattern: "{0}")rYsearchrrXr)r*rmessagerrrrE-s  zregex.__call__cCs t|jSr")rWrX)r*memorrr __deepcopy__3szregex.__deepcopy__cCs d|jdS)Nr)rrX)rXrGrrrrH6szregex.__schema__N) rIrJrKrLr+rEr^rMrHrrrrrWs rWcCsft|ts"t|t}t|t}|jdkrFtj|}tj|}n|tj}|tj}||fS)a Normalize datetime intervals. Given a pair of datetime.date or datetime.datetime objects, returns a 2-tuple of tz-aware UTC datetimes spanning the same interval. For datetime.date objects, the returned interval starts at 00:00:00.0 on the first date and ends at 00:00:00.0 on the second. Naive datetimes are upgraded to UTC. Timezone-aware datetimes are normalized to the UTC tzdata. Params: - start: A date or datetime - end: A date or datetime N) isinstancercombine START_OF_DAYr pytzUTCZlocalize astimezone)startendrrrr_normalize_interval>s       rgcCst|ts|tdd}nd|dd}tdd|}|d}|dkrX|tdd}n&|dkrp|tdd }n|tdd }|S) Nr0)daysTz[+-].+:r)hours)minutes)seconds)r_rrr<rYsubr)rerrfrZtime_without_offsetZnum_separatorsrrr_expand_datetime^s  rpc Csfztt|WStk r`zt|dfWYStk rZt|dfYYSXYnXdS)zg Do some nasty try/except voodoo to get some sort of datetime object(s) out of the string. N)sorted aniso8601Zparse_intervalrparse_datetime parse_daterrrr_parse_intervalvsruargumentcCsr|s tdz2t|\}}|dkr,t||}t|||\}}Wn*tk rhd}t|j||dYnX||fS)aG Parses ISO 8601-formatted datetime intervals into tuples of datetimes. Accepts both a single date(time) or a full interval using either start/end or start/duration notation, with the following behavior: - Intervals are defined as inclusive start, exclusive end - Single datetimes are translated into the interval spanning the largest resolution not specified in the input value, up to the day. - The smallest accepted resolution is 1 second. - All timezones are accepted as values; returned datetimes are localized to UTC. Naive inputs and date inputs will are assumed UTC. Examples:: "2013-01-01" -> datetime(2013, 1, 1), datetime(2013, 1, 2) "2013-01-01T12" -> datetime(2013, 1, 1, 12), datetime(2013, 1, 1, 13) "2013-01-01/2013-02-28" -> datetime(2013, 1, 1), datetime(2013, 2, 28) "2013-01-01/P3D" -> datetime(2013, 1, 1), datetime(2013, 1, 4) "2013-01-01T12:00/PT30M" -> datetime(2013, 1, 1, 12), datetime(2013, 1, 1, 12, 30) "2013-01-01T06:00/2013-01-01T12:00" -> datetime(2013, 1, 1, 6), datetime(2013, 1, 1, 12) :param str value: The ISO8601 date time as a string :return: Two UTC datetimes, the start and the end of the specified interval :rtype: A tuple (datetime, datetime) :raises ValueError: if the interval is invalid. z,Expected a valid ISO8601 date/time interval.NzIInvalid {arg}: {value}. {arg} must be a valid ISO8601 date/time interval.argr)rrurprgr)rrvrerfr.rrriso8601intervals  ryziso8601-intervalcCst|d}|S)z3Parse a valid looking date in the format YYYY-mm-ddz%Y-%m-%d)rstrptimerdaterrrr|s r|c Cs6z t|WSttfk r0td|YnXdS)Nz{0} is not a valid integer)r@ TypeErrorrrrrrr _get_integers r~cCs*t|}|dkr&d}t|j||d|S)z:Restrict input type to the natural numbers (0, 1, 2, 3...)rzlowerrrrrrbooleans   rrcCs|}t|sd|df}zFt|}t|}|ddkrPt|jtj dWSt|tj WSWn"t k rt d |YnXdS)aH Turns an RFC822 formatted date into a datetime object. Example:: inputs.datetime_from_rfc822('Wed, 02 Oct 2002 08:00:00 EST') :param str value: The RFC822-complying string to transform :return: The parsed datetime :rtype: datetime :raises ValueError: if value is an invalid date literal  z00:00:00NrInvalid date literal "{0}") time_regexr[r,rrr fromtimestampreplacerbutc Exceptionrr)rraw timetuple timestamprrrdatetime_from_rfc822s  rcCsnzFzt|WWStk rBt|}t|j|j|jYWSXWn"tk rhtd |YnXdS)a> Turns an ISO8601 formatted date into a datetime object. Example:: inputs.datetime_from_iso8601("2012-01-01T23:30:00+02:00") :param str value: The ISO8601-complying string to transform :return: A datetime :rtype: datetime :raises ValueError: if value is an invalid date literal rN) rrrsrrtryearmonthdayrrr{rrrdatetime_from_iso86019s rz date-timecCs t|S)a! Turns an ISO8601 formatted date into a date object. Example:: inputs.date_from_iso8601("2012-01-01") :param str value: The ISO8601-complying string to transform :return: A date :rtype: date :raises ValueError: if value is an invalid date literal )rr|rrrrdate_from_iso8601Tsr)rv)rv)rv)+rLrYrrrr email.utilsrr urllib.parserrrrbrcraZ END_OF_DAYrZ IGNORECASEr6rUrrrHrr objectr!rFrQrWrgrpruryr|r~rrrrrrrrrrrsl      qC$  0    !