summaryrefslogtreecommitdiffstats
path: root/sqlglot/executor/env.py
diff options
context:
space:
mode:
Diffstat (limited to 'sqlglot/executor/env.py')
-rw-r--r--sqlglot/executor/env.py32
1 files changed, 32 insertions, 0 deletions
diff --git a/sqlglot/executor/env.py b/sqlglot/executor/env.py
new file mode 100644
index 0000000..72b0558
--- /dev/null
+++ b/sqlglot/executor/env.py
@@ -0,0 +1,32 @@
+import datetime
+import re
+import statistics
+
+
+class reverse_key:
+ def __init__(self, obj):
+ self.obj = obj
+
+ def __eq__(self, other):
+ return other.obj == self.obj
+
+ def __lt__(self, other):
+ return other.obj < self.obj
+
+
+ENV = {
+ "__builtins__": {},
+ "datetime": datetime,
+ "locals": locals,
+ "re": re,
+ "float": float,
+ "int": int,
+ "str": str,
+ "desc": reverse_key,
+ "SUM": sum,
+ "AVG": statistics.fmean if hasattr(statistics, "fmean") else statistics.mean,
+ "COUNT": lambda acc: sum(1 for e in acc if e is not None),
+ "MAX": max,
+ "MIN": min,
+ "POW": pow,
+}