omlish 0.0.0.dev252__py3-none-any.whl → 0.0.0.dev253__py3-none-any.whl

omlish/sql/parsing/_antlr/MinisqlVisitor.py

@@ -0,0 +1,291 @@
+# type: ignore
+# ruff: noqa
+# flake8: noqa
+# Generated from Minisql.g4 by ANTLR 4.13.2
+from omlish.antlr._runtime._all import *
+if "." in __name__:
+    from .MinisqlParser import MinisqlParser
+else:
+    from MinisqlParser import MinisqlParser
+
+# This class defines a complete generic visitor for a parse tree produced by MinisqlParser.
+
+class MinisqlVisitor(ParseTreeVisitor):
+
+    # Visit a parse tree produced by MinisqlParser#singleStmt.
+    def visitSingleStmt(self, ctx:MinisqlParser.SingleStmtContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#select.
+    def visitSelect(self, ctx:MinisqlParser.SelectContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#cteSelect.
+    def visitCteSelect(self, ctx:MinisqlParser.CteSelectContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#cte.
+    def visitCte(self, ctx:MinisqlParser.CteContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#unionSelect.
+    def visitUnionSelect(self, ctx:MinisqlParser.UnionSelectContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#unionItem.
+    def visitUnionItem(self, ctx:MinisqlParser.UnionItemContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#primarySelect.
+    def visitPrimarySelect(self, ctx:MinisqlParser.PrimarySelectContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#allSelectItem.
+    def visitAllSelectItem(self, ctx:MinisqlParser.AllSelectItemContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#exprSelectItem.
+    def visitExprSelectItem(self, ctx:MinisqlParser.ExprSelectItemContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#expr.
+    def visitExpr(self, ctx:MinisqlParser.ExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#unaryBooleanExpr.
+    def visitUnaryBooleanExpr(self, ctx:MinisqlParser.UnaryBooleanExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#predicatedBooleanExpr.
+    def visitPredicatedBooleanExpr(self, ctx:MinisqlParser.PredicatedBooleanExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#binaryBooleanExpr.
+    def visitBinaryBooleanExpr(self, ctx:MinisqlParser.BinaryBooleanExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#castBooleanExpr.
+    def visitCastBooleanExpr(self, ctx:MinisqlParser.CastBooleanExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#cmpPredicate.
+    def visitCmpPredicate(self, ctx:MinisqlParser.CmpPredicateContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#isNullPredicate.
+    def visitIsNullPredicate(self, ctx:MinisqlParser.IsNullPredicateContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#inListPredicate.
+    def visitInListPredicate(self, ctx:MinisqlParser.InListPredicateContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#inSelectPredicate.
+    def visitInSelectPredicate(self, ctx:MinisqlParser.InSelectPredicateContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#likePredicate.
+    def visitLikePredicate(self, ctx:MinisqlParser.LikePredicateContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#arithValueExpr.
+    def visitArithValueExpr(self, ctx:MinisqlParser.ArithValueExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#unaryValueExpr.
+    def visitUnaryValueExpr(self, ctx:MinisqlParser.UnaryValueExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#primaryValueExpr.
+    def visitPrimaryValueExpr(self, ctx:MinisqlParser.PrimaryValueExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#functionCallExpr.
+    def visitFunctionCallExpr(self, ctx:MinisqlParser.FunctionCallExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#starFunctionCallExpr.
+    def visitStarFunctionCallExpr(self, ctx:MinisqlParser.StarFunctionCallExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#caseExpr.
+    def visitCaseExpr(self, ctx:MinisqlParser.CaseExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#selectExpr.
+    def visitSelectExpr(self, ctx:MinisqlParser.SelectExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#parenExpr.
+    def visitParenExpr(self, ctx:MinisqlParser.ParenExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#simplePrimaryExpr.
+    def visitSimplePrimaryExpr(self, ctx:MinisqlParser.SimplePrimaryExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#simpleExpr.
+    def visitSimpleExpr(self, ctx:MinisqlParser.SimpleExprContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#caseItem.
+    def visitCaseItem(self, ctx:MinisqlParser.CaseItemContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#over.
+    def visitOver(self, ctx:MinisqlParser.OverContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#sortItem.
+    def visitSortItem(self, ctx:MinisqlParser.SortItemContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#aliasedRelation.
+    def visitAliasedRelation(self, ctx:MinisqlParser.AliasedRelationContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#joinRelation.
+    def visitJoinRelation(self, ctx:MinisqlParser.JoinRelationContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#selectRelation.
+    def visitSelectRelation(self, ctx:MinisqlParser.SelectRelationContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#tableRelation.
+    def visitTableRelation(self, ctx:MinisqlParser.TableRelationContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#parenRelation.
+    def visitParenRelation(self, ctx:MinisqlParser.ParenRelationContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#groupBy.
+    def visitGroupBy(self, ctx:MinisqlParser.GroupByContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#qualifiedName.
+    def visitQualifiedName(self, ctx:MinisqlParser.QualifiedNameContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#identList.
+    def visitIdentList(self, ctx:MinisqlParser.IdentListContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#ident.
+    def visitIdent(self, ctx:MinisqlParser.IdentContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#quotedIdent.
+    def visitQuotedIdent(self, ctx:MinisqlParser.QuotedIdentContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#integerNumber.
+    def visitIntegerNumber(self, ctx:MinisqlParser.IntegerNumberContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#decimalNumber.
+    def visitDecimalNumber(self, ctx:MinisqlParser.DecimalNumberContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#floatNumber.
+    def visitFloatNumber(self, ctx:MinisqlParser.FloatNumberContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#string.
+    def visitString(self, ctx:MinisqlParser.StringContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#null.
+    def visitNull(self, ctx:MinisqlParser.NullContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#true.
+    def visitTrue(self, ctx:MinisqlParser.TrueContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#false.
+    def visitFalse(self, ctx:MinisqlParser.FalseContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#setQuantifier.
+    def visitSetQuantifier(self, ctx:MinisqlParser.SetQuantifierContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#joinType.
+    def visitJoinType(self, ctx:MinisqlParser.JoinTypeContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#cmpOp.
+    def visitCmpOp(self, ctx:MinisqlParser.CmpOpContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#arithOp.
+    def visitArithOp(self, ctx:MinisqlParser.ArithOpContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#unaryOp.
+    def visitUnaryOp(self, ctx:MinisqlParser.UnaryOpContext):
+        return self.visitChildren(ctx)
+
+
+    # Visit a parse tree produced by MinisqlParser#unquotedIdent.
+    def visitUnquotedIdent(self, ctx:MinisqlParser.UnquotedIdentContext):
+        return self.visitChildren(ctx)
+
+
+
+del MinisqlParser

omlish/sql/parsing/parsing.py

@@ -0,0 +1,119 @@
+# ruff: noqa: N802 N803
+import typing as ta
+
+from ... import check
+from ...antlr import runtime as antlr4
+from ...antlr.delimit import DelimitingLexer
+from ...antlr.errors import SilentRaisingErrorListener
+from .. import queries as no
+from ._antlr.MinisqlLexer import MinisqlLexer  # type: ignore
+from ._antlr.MinisqlParser import MinisqlParser  # type: ignore
+from ._antlr.MinisqlVisitor import MinisqlVisitor  # type: ignore
+
+
+##
+
+
+class _ParseVisitor(MinisqlVisitor):
+    def visit(self, ctx: antlr4.ParserRuleContext):
+        check.isinstance(ctx, antlr4.ParserRuleContext)
+        node = ctx.accept(self)
+        return node
+
+    def aggregateResult(self, aggregate, nextResult):
+        if aggregate is not None:
+            check.none(nextResult)
+            return aggregate
+        else:
+            check.none(aggregate)
+            return nextResult
+
+    #
+
+    def visitExprSelectItem(self, ctx: MinisqlParser.ExprSelectItemContext):
+        value = self.visit(ctx.expr())
+        label = self.visit(ctx.ident()) if ctx.ident() is not None else None
+        return no.SelectItem(value, label)
+
+    def visitIntegerNumber(self, ctx: MinisqlParser.IntegerNumberContext):
+        return no.Literal(int(ctx.INTEGER_VALUE().getText()))
+
+    def visitPrimarySelect(self, ctx: MinisqlParser.PrimarySelectContext):
+        items = [self.visit(i) for i in ctx.selectItem()]
+        relations = [self.visit(r) for r in ctx.relation()]
+        where = self.visit(ctx.where) if ctx.where is not None else None
+        return no.Select(
+            items=items,
+            from_=check.single(relations) if relations else None,
+            where=where,
+        )
+
+    def visitQuotedIdent(self, ctx: MinisqlParser.QuotedIdentContext):
+        name = unquote(ctx.QUOTED_IDENT().getText(), '"')
+        return no.Ident(name)
+
+    def visitUnquotedIdent(self, ctx: MinisqlParser.UnquotedIdentContext):
+        return no.Ident(ctx.getText())
+
+
+##
+
+
+def create_parser(buf: str) -> MinisqlParser:
+    lexer = MinisqlLexer(antlr4.InputStream(buf))
+    lexer.removeErrorListeners()
+    lexer.addErrorListener(SilentRaisingErrorListener())
+
+    stream = antlr4.CommonTokenStream(lexer)
+    stream.fill()
+
+    parser = MinisqlParser(stream)
+    parser.removeErrorListeners()
+    parser.addErrorListener(SilentRaisingErrorListener())
+
+    return parser
+
+
+##
+
+
+def parse_stmt(buf: str, **kwargs) -> no.Stmt:
+    parser = create_parser(buf, **kwargs)
+    node = _ParseVisitor().visit(parser.singleStmt())
+    return check.isinstance(node, no.Stmt)
+
+
+class _DelimitingLexer(DelimitingLexer, MinisqlLexer):
+    pass
+
+
+def split_stmts(buf: str) -> ta.Sequence[str]:
+    lexer = _DelimitingLexer(
+        antlr4.InputStream(buf),
+        delimiter_token=MinisqlParser.DELIMITER,
+        delimiters=[';'],
+    )
+    lexer.removeErrorListeners()
+    lexer.addErrorListener(SilentRaisingErrorListener())
+
+    lst, part = lexer.split()
+    if part.strip():
+        raise ValueError(part)
+
+    return [s for s, _ in lst]
+
+
+def parse_stmts(buf: str, **kwargs) -> ta.Sequence[no.Stmt]:
+    return [parse_stmt(sb, **kwargs) for sb in split_stmts(buf)]
+
+
+##
+
+
+def quote(val: str, q: str) -> str:
+    return q + val.replace(q, q * 2) + q
+
+
+def unquote(val: str, q: str) -> str:
+    check.arg(val.startswith(q) and val.endswith(q))
+    return val[1:-1].replace(q * 2, q)

{omlish-0.0.0.dev252.dist-info → omlish-0.0.0.dev253.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: omlish
-Version: 0.0.0.dev252
+Version: 0.0.0.dev253
 Summary: omlish
 Author: wrmsr
 License: BSD-3-Clause

{omlish-0.0.0.dev252.dist-info → omlish-0.0.0.dev253.dist-info}/RECORD

@@ -1,5 +1,5 @@
 omlish/.manifests.json,sha256=x26AIwDzScUvnX-p4xlq6Zc5QYrAo0Vmgf1qHc1KL_M,8253
-omlish/__about__.py,sha256=irK5EMsRK45Vveo-bLYvfj4i3jGa_hT0w6FNpRHdcso,3380
+omlish/__about__.py,sha256=SUzyh4gNJezXoDjwGUqKWpJF2LoBdZqSNlCL-GRYdsk,3380
 omlish/__init__.py,sha256=SsyiITTuK0v74XpKV8dqNaCmjOlan1JZKrHQv5rWKPA,253
 omlish/c3.py,sha256=ubu7lHwss5V4UznbejAI0qXhXahrU01MysuHOZI9C4U,8116
 omlish/cached.py,sha256=MLap_p0rdGoDIMVhXVHm1tsbcWobJF0OanoodV03Ju8,542
@@ -13,6 +13,7 @@ omlish/outcome.py,sha256=ABIE0zjjTyTNtn-ZqQ_9_mUzLiBQ3sDAyqc9JVD8N2k,7852
 omlish/runmodule.py,sha256=PWvuAaJ9wQQn6bx9ftEL3_d04DyotNn8dR_twm2pgw0,700
 omlish/shlex.py,sha256=bsW2XUD8GiMTUTDefJejZ5AyqT1pTgWMPD0BMoF02jE,248
 omlish/sync.py,sha256=QJ79kxmIqDP9SeHDoZAf--DpFIhDQe1jACy8H4N0yZI,2928
+omlish/_antlr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omlish/algorithm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omlish/algorithm/all.py,sha256=FudUHwoaRLNNmqYM3jhP2Yd2BpmYhNBRPaVZzARMoSc,194
 omlish/algorithm/distribute.py,sha256=kO60F7HtMF3j-IxuhJdtsfgIR1Vpf1okQfacb4eNukQ,1533
@@ -669,6 +670,14 @@ omlish/sql/api/errors.py,sha256=YtC2gz5DqRTT3uCJniUOufVH1GEnFIc5ElkYLK3BHwM,230
 omlish/sql/api/funcs.py,sha256=-H6V-o9JPSHFXsxdHtutB4mP2LwJfCzlLbRrPHunmB4,990
 omlish/sql/api/queries.py,sha256=IgB8_sDe40-mKE-ByTmZ4GVOCdLLJDzJGJCevMd8R5s,1207
 omlish/sql/api/rows.py,sha256=MEK9LNYEe8vLEJXQJD63MpnSOiE22cawJL-dUWQD6sU,1246
+omlish/sql/parsing/Minisql.g4,sha256=DlriPFcnGT4QYG7O7VH2EmDVJQZq5zewoQ-tpXsSbIY,4748
+omlish/sql/parsing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+omlish/sql/parsing/parsing.py,sha256=hYIZdEKik4q22aEHHOX6lCpMa1jvgUk2aC1Llx10NMI,3343
+omlish/sql/parsing/_antlr/MinisqlLexer.py,sha256=f4WzRrwqM7tW6sNn01035hRWPZL7xtuvba9EMR06j2E,17611
+omlish/sql/parsing/_antlr/MinisqlListener.py,sha256=2py_bfDQeEtJaJh9rtpKghbSrQlEGAEKf5ZBLErKqsI,16633
+omlish/sql/parsing/_antlr/MinisqlParser.py,sha256=y9SFjXdQlWYJa2PbPm30d5SfcYM_8M8ts46IHhENbNc,132457
+omlish/sql/parsing/_antlr/MinisqlVisitor.py,sha256=NCPorucLLOZ-Q99BtNbDOAfHytQl8wyroR8pI1uVovg,10030
+omlish/sql/parsing/_antlr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omlish/sql/queries/__init__.py,sha256=N8oQFKY99g_MQhrPmvlBAkMeGIRURE9UxMO244mytzY,1332
 omlish/sql/queries/base.py,sha256=_8O3MbH_OEjBnhp2oIJUZ3ClaQ8l4Sj9BdPdsP0Ie-g,224
 omlish/sql/queries/binary.py,sha256=dcEzeEn104AMPuQ7QrJU2O-YCN3SUdxB5S4jaWKOUqY,2253
@@ -744,9 +753,9 @@ omlish/text/mangle.py,sha256=kfzFLfvepH-chl1P89_mdc5vC4FSqyPA2aVtgzuB8IY,1133
 omlish/text/minja.py,sha256=jZC-fp3Xuhx48ppqsf2Sf1pHbC0t8XBB7UpUUoOk2Qw,5751
 omlish/text/parts.py,sha256=JkNZpyR2tv2CNcTaWJJhpQ9E4F0yPR8P_YfDbZfMtwQ,6182
 omlish/text/random.py,sha256=jNWpqiaKjKyTdMXC-pWAsSC10AAP-cmRRPVhm59ZWLk,194
-omlish-0.0.0.dev252.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omlish-0.0.0.dev252.dist-info/METADATA,sha256=d3I-8A12dfY6BUTJyIZ70ySFdyaJLUuDZstZPmrYvD0,4176
-omlish-0.0.0.dev252.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-omlish-0.0.0.dev252.dist-info/entry_points.txt,sha256=Lt84WvRZJskWCAS7xnQGZIeVWksprtUHj0llrvVmod8,35
-omlish-0.0.0.dev252.dist-info/top_level.txt,sha256=pePsKdLu7DvtUiecdYXJ78iO80uDNmBlqe-8hOzOmfs,7
-omlish-0.0.0.dev252.dist-info/RECORD,,
+omlish-0.0.0.dev253.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omlish-0.0.0.dev253.dist-info/METADATA,sha256=Mo7bxTA92y4u34s-B-sWSsWvkHFRu4wOhRnIk5VTauA,4176
+omlish-0.0.0.dev253.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+omlish-0.0.0.dev253.dist-info/entry_points.txt,sha256=Lt84WvRZJskWCAS7xnQGZIeVWksprtUHj0llrvVmod8,35
+omlish-0.0.0.dev253.dist-info/top_level.txt,sha256=pePsKdLu7DvtUiecdYXJ78iO80uDNmBlqe-8hOzOmfs,7
+omlish-0.0.0.dev253.dist-info/RECORD,,