Question. How to collapse AND or OR expression. about sqlglot

[tobymao]

check out transform, or replace methods.

https://github.com/tobymao/sqlglot/blob/main/posts/ast_primer.md

https://github.com/tobymao/sqlglot/blob/main/sqlglot/expressions.py#L583

[davlee1972]

Got it working..

import sqlglot

def remove_in_sql(sql_exp):
    while True:
        in_sqls = [s for s in sql_exp.find_all(sqlglot.exp.In) if 'expressions' in s.args]
        if in_sqls:
            saved_parent = in_sqls[0].parent
            in_sqls[0].pop()
            if isinstance(saved_parent, sqlglot.exp.And):
                if 'this' in saved_parent.args and saved_parent.args['this']:
                    saved_parent.parent.args['this'] = saved_parent.args['this']
                elif 'expression' in saved_parent.args and saved_parent.args['expression']:
                    saved_parent.parent.args['this'] = saved_parent.args['expression']
                else:
                    saved_parent.parent.parent.args['where'] = None
            elif isinstance(saved_parent, sqlglot.exp.Where):
                saved_parent.parent.args['where'] = None
        else:
            break

>>> z = sqlglot.parse_one("select a, b from abc where a in (1,2,3) and b in (7,8,9)")
>>> remove_in_sql(z)
>>> z.sql()
'SELECT a, b FROM abc'
>>>
>>> z = sqlglot.parse_one("select a, b from abc where a in (1,2,3)")
>>> remove_in_sql(z)
>>> z.sql()
'SELECT a, b FROM abc'
>>>
>>> z = sqlglot.parse_one("select * from abc where (a in (1,2,3) and c = 'abc') and b in (7,8,9)")
>>> remove_in_sql(z)
>>> z.sql()
"SELECT * FROM abc WHERE (c = 'abc')"
>>>
>>> z = sqlglot.parse_one("select * from abc where a in (1,2,3) and c = 'abc' and b in (select z from zzz where d in (1,2,3))")
>>> remove_in_sql(z)
>>> z.sql()
"SELECT * FROM abc WHERE c = 'abc' AND b IN (SELECT z FROM zzz)"
>>>
>>> z = sqlglot.parse_one("select * from abc where a in (1,2,3) and b in (select z from zzz where d in (1,2,3))")
>>> remove_in_sql(z)
>>> z.sql()
'SELECT * FROM abc WHERE b IN (SELECT z FROM zzz)'