support pyarrow compute funcs (apache#61)

* update compute.pyi

* impl Aggregation funcs

* impl arithmetic

* imit bit-wise functions

* imit rounding functions

* optimize annotation

* impl logarithmic functions

* update

* impl comparisons funcs

* impl logical funcs

* impl string predicates and transforms

* impl string padding

* impl string trimming

* impl string splitting and component extraction

* impl string joining and slicing

* impl Containment tests

* impl Categorizations

* impl Structural transforms

* impl Conversions

* impl Temporal component extraction

* impl random, Timezone handling

* impl Array-wise functions

* fix timestamp scalar

Author: zen-xu

pyarrow-stubs/__lib_pxi/scalar.pyi

@@ -1,4 +1,4 @@
-# mypy: disable-error-code="overload-overlap,misc"
+# mypy: disable-error-code="overload-overlap,misc,type-arg"
 import collections.abc
 import datetime as dt
 
@@ -281,6 +281,10 @@ def scalar(
     value: Decimal, *, from_pandas: bool | None = None, memory_pool: MemoryPool | None = None
 ) -> Decimal128Scalar: ...
 @overload
+def scalar(
+    value: dt.datetime, *, from_pandas: bool | None = None, memory_pool: MemoryPool | None = None
+) -> TimestampScalar: ...
+@overload
 def scalar(
     value: dt.date, *, from_pandas: bool | None = None, memory_pool: MemoryPool | None = None
 ) -> Date32Scalar: ...
@@ -349,6 +353,13 @@ def scalar(
     memory_pool: MemoryPool | None = None,
 ) -> ListScalar[types.ListType[types.Decimal128Type]]: ...
 @overload
+def scalar(
+    value: CollectionValue[dt.datetime],
+    *,
+    from_pandas: bool | None = None,
+    memory_pool: MemoryPool | None = None,
+) -> ListScalar[types.ListType[types.TimestampType]]: ...
+@overload
 def scalar(
     value: CollectionValue[dt.date],
     *,

pyarrow-stubs/__lib_pxi/table.pyi

@@ -1,4 +1,4 @@
-# mypy: disable-error-code="overload-overlap"
+# mypy: disable-error-code="overload-overlap,type-arg,misc"
 
 import datetime as dt
 
@@ -162,6 +162,11 @@ def chunked_array(
     type: None = None,
 ) -> ChunkedArray[scalar.StructScalar]: ...
 @overload
+def chunked_array(
+    values: NullableIterable[dt.datetime],
+    type: None = None,
+) -> ChunkedArray[scalar.TimestampScalar]: ...
+@overload
 def chunked_array(
     values: NullableIterable[dt.date],
     type: None = None,