o i@s*ddlmZddlZddlZddlmZmZmZmZmZddl m Z m Z ddl m Z ddlmZddlmZmZmZmZmZddlmZddlmZdd lmZmZmZmZm Z dd l!m"Z"e#e$ddl%m&Z'Wdn1stwYe rdd l(m)Z)dd l*m+Z+m,Z, dd ddddZ-dS)) annotationsN)datedatetimetime timedeltatimezone) TYPE_CHECKINGAny)ZoneInfo)_check_for_numpy_check_for_pytz_check_for_torchpytztorch)numpy) wrap_expr) BaseExtensionDateDatetimeDurationObject)DataTypeMappings)Expr)PolarsDataTypeTimeUnitF) allow_objectvaluer dtypePolarsDataType | NonerboolreturnrcCs4t|trt||j|St|tr$t|tr$d|d}t||t krj?@|j:jA|j:jBf}|d urt||dSn|}t t j||ddS)a} Return an expression representing a literal value. Parameters ---------- value Value that should be used as a `literal`. dtype The data type of the resulting expression. If set to `None` (default), the data type is inferred from the `value` input. allow_object If type is unknown use an 'object' type. By default, we will raise a `ValueException` if the type is unknown. Notes ----- Expected datatypes: - `pl.lit([])` -> empty List - `pl.lit([1, 2, 3])` -> List - `pl.lit(pl.Series([]))`-> empty Series Null - `pl.lit(pl.Series([1, 2, 3]))` -> Series Int64 - `pl.lit(None)` -> Null Examples -------- Literal scalar values: >>> pl.lit(1) # doctest: +IGNORE_RESULT >>> pl.lit(5.5) # doctest: +IGNORE_RESULT >>> pl.lit(None) # doctest: +IGNORE_RESULT >>> pl.lit("foo_bar") # doctest: +IGNORE_RESULT >>> pl.lit(date(2021, 1, 20)) # doctest: +IGNORE_RESULT >>> pl.lit(datetime(2023, 3, 31, 10, 30, 45)) # doctest: +IGNORE_RESULT Literal list/Series data (1D): >>> pl.lit([1, 2, 3]) # doctest: +SKIP >>> pl.lit(pl.Series("x", [1, 2, 3])) # doctest: +IGNORE_RESULT Literal list/Series data (2D): >>> pl.lit([[1, 2], [3, 4]]) # doctest: +SKIP >>> pl.lit(pl.Series("y", [[1, 2], [3, 4]])) # doctest: +IGNORE_RESULT zdtype 'z4' is a BaseExtension class, it should be an instanceliteral)rT) is_scalarF)rr"N time_unitus time_zoneUTC utcoffsetsecondsrztime zone of dtype (z#) differs from time zone of value ())tzinfoZearliestZlatest)Z ambiguous)forcerz datetime64[z timedelta64[)C isinstancerlitZ ext_storageexttotype issubclass TypeErrorrplZSeriesrplrZ_srrrgetattrr+r r rZ BaseTzInfozonestr astimezonerutchasattrr'replacecastrdtZreplace_time_zonefoldrrryearmonthdayr npZndarrayr rZTensorrlisttupleenumEnumrZgenericitemintrname startswithlenrZ NUMPY_KIND_AND_ITEMSIZE_TO_DTYPEgetkinditemsize)rrrmsgZvalue_sZtur#Zdtype_tzZvalue_tztzZdt_utcZdt_utc_sexprr%rIZ dtype_namerTK/home/app/Keep/.python/lib/python3.10/site-packages/polars/functions/lit.pyr/s 3                    r/)N)rr rrrrr r). __future__r contextlibrGrrrrrtypingrr zoneinfor Zpolars._reexportZ _reexportr5Zpolars._dependenciesr r r rrrrDZpolars._utils.wraprZpolars.datatypesrrrrrZpolars.datatypes.convertrsuppress ImportErrorZ polars._plrZ_plrr6ZpolarsrZpolars._typingrrr/rTrTrTrUs,