FilesExpand file tree

 main

/

__init__.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

124 lines (100 loc) · 2.87 KB

 main

/

__init__.py

Copy path

Top

File metadata and controls

• Code
• Blame

124 lines (100 loc) · 2.87 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

# Licensed to the Apache Software Foundation (ASF) under one

# or more contributor license agreements. See the NOTICE file

# distributed with this work for additional information

# regarding copyright ownership. The ASF licenses this file

# to you under the Apache License, Version 2.0 (the

# "License"); you may not use this file except in compliance

# with the License. You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing,

# software distributed under the License is distributed on an

# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

# KIND, either express or implied. See the License for the

# specific language governing permissions and limitations

# under the License.

"""DataFusion python package.

This is a Python library that binds to Apache Arrow in-memory query engine DataFusion.

See https://datafusion.apache.org/python for more information.

"""

try:

import importlib.metadata as importlib_metadata

except ImportError:

import importlib_metadata

from . import functions, object_store, substrait

# The following imports are okay to remain as opaque to the user.

from ._internal import Config

from .catalog import Catalog, Database, Table

from .common import (

DFSchema,

)

from .context import (

RuntimeEnvBuilder,

SessionConfig,

SessionContext,

SQLOptions,

)

from .dataframe import DataFrame

from .expr import (

Expr,

WindowFrame,

)

from .plan import ExecutionPlan, LogicalPlan

from .record_batch import RecordBatch, RecordBatchStream

from .udf import Accumulator, AggregateUDF, ScalarUDF, WindowUDF

__version__ = importlib_metadata.version(__name__)

__all__ = [

"Accumulator",

"Config",

"DataFrame",

"SessionContext",

"SessionConfig",

"SQLOptions",

"RuntimeEnvBuilder",

"Expr",

"ScalarUDF",

"WindowFrame",

"column",

"col",

"literal",

"lit",

"DFSchema",

"Catalog",

"Database",

"Table",

"AggregateUDF",

"WindowUDF",

"LogicalPlan",

"ExecutionPlan",

"RecordBatch",

"RecordBatchStream",

"common",

"expr",

"functions",

"object_store",

"substrait",

]

def column(value: str):

"""Create a column expression."""

return Expr.column(value)

def col(value: str):

"""Create a column expression."""

return Expr.column(value)

def literal(value):

"""Create a literal expression."""

return Expr.literal(value)

def string_literal(value):

"""Create a UTF8 literal expression.

It differs from `literal` which creates a UTF8view literal.

"""

return Expr.string_literal(value)

def str_lit(value):

"""Alias for `string_literal`."""

return string_literal(value)

def lit(value):

"""Create a literal expression."""

return Expr.literal(value)

udf = ScalarUDF.udf

udaf = AggregateUDF.udaf

udwf = WindowUDF.udwf