o ip@s|ddlmZddlmZmZddlmZmZhdZd d d Z d!d dZ d"ddZ d#ddZ eGdddZ d$ddZdS)%) annotations) dataclassfield)ListOptional>{}()[]valuestr field_namereturnNonecCs<|D]}t|dkst|dks|tvrt|dqdS)zEnsure value contains only printable ASCII without spaces or braces. This mirrors the constraints enforced by other Redis clients for values that will appear in CLIENT LIST / CLIENT INFO output. !~zG must not contain spaces, newlines, non-printable characters, or bracesN)ord_BRACES ValueError)r rchrE/home/ubuntu/.local/lib/python3.10/site-packages/redis/driver_info.py_validate_no_invalid_chars s rnamecCs*ddl}t|d|d|stddS)aoValidate an upstream driver name. The name should look like a typical Python distribution or package name, following a simplified form of PEP 503 normalisation rules: * start with a lowercase ASCII letter * contain only lowercase letters, digits, hyphens and underscores Examples of valid names: ``"django-redis"``, ``"celery"``, ``"rq"``. rNz Driver namez^[a-z][a-z0-9_-]*$zUpstream driver name must use a Python package-style name: start with a lowercase letter and contain only lowercase letters, digits, hyphens, and underscores (e.g., 'django-redis').)rermatchr)rrrrr_validate_driver_names  rversioncCst|ddS)NzDriver version)r)rrrr_validate_driver_version/r driver_namedriver_versioncCs|d|S)N_vr)r"r#rrr_format_driver_entry3r!r%c@sjeZdZUdZdZded<dZded<eedZ d ed <d d Z e dddZ dddZ e dddZdS) DriverInfoaDriver information used to build the CLIENT SETINFO LIB-NAME and LIB-VER values. This class consolidates all driver metadata (redis-py version and upstream drivers) into a single object that is propagated through connection pools and connections. The formatted name follows the pattern:: name(driver1_vVersion1;driver2_vVersion2) Parameters ---------- name : str, optional The base library name (default: "redis-py") lib_version : str, optional The redis-py library version. If None, the version will be determined automatically from the installed package. Examples -------- >>> info = DriverInfo() >>> info.formatted_name 'redis-py' >>> info = DriverInfo().add_upstream_driver("django-redis", "5.4.0") >>> info.formatted_name 'redis-py(django-redis_v5.4.0)' >>> info = DriverInfo(lib_version="5.0.0") >>> info.lib_version '5.0.0' redis-pyrrN Optional[str] lib_version)default_factory List[str] _upstreamcCs&|jdurddlm}||_dSdS)z'Initialize lib_version if not provided.Nrget_lib_version)r) redis.utilsr.)selfr.rrr __post_init__]s   zDriverInfo.__post_init__rcCs t|jS)zuReturn a copy of the upstream driver entries. Each entry is in the form ``"driver-name_vversion"``. )listr,r0rrrupstream_driversds zDriverInfo.upstream_driversr"r# 'DriverInfo'cCsL|durtd|durtdt|t|t||}|jd||S)zAdd an upstream driver to this instance and return self. The most recently added driver appears first in :pyattr:`formatted_name`. NzDriver name must not be NonezDriver version must not be Noner)rrr r%r,insert)r0r"r#entryrrradd_upstream_driverms zDriverInfo.add_upstream_drivercCs&|js|jS|jdd|jdS)zReturn the base name with upstream drivers encoded, if any. With no upstream drivers, this is just :pyattr:`name`. Otherwise:: name(driver1_vX;driver2_vY) r ;r )r,rjoinr3rrrformatted_names zDriverInfo.formatted_name)rr+)r"rr#rrr5)rr)__name__ __module__ __qualname____doc__r__annotations__r)rr2r,r1propertyr4r8r;rrrrr&7s    r& driver_infoOptional[DriverInfo]lib_namer(r)cCsF|dur|Sddlm}|dur|nd}|dur|n|}t||dS)aResolve driver_info from parameters. If driver_info is provided, use it. Otherwise, create DriverInfo from lib_name and lib_version (using defaults if not provided). Parameters ---------- driver_info : DriverInfo, optional The DriverInfo instance to use lib_name : str, optional The library name (default: "redis-py") lib_version : str, optional The library version (default: auto-detected) Returns ------- DriverInfo The resolved DriverInfo instance Nrr-r')rr))r/r.r&)rBrDr)r.rrrrrresolve_driver_infos   rEN)r rrrrr)rrrr)rrrr)r"rr#rrr)rBrCrDr(r)r(rr&) __future__r dataclassesrrtypingrrrrrr r%r&rErrrrs     X