jD^ddlZddlmZddlmZmZddlmZm Z ddl m Z GddeZ ej e dZGd d eZej ed Z dd Z dd Z ddZGdde eZej edZddZdS)N)compiler) DDLElementPrimaryKeyConstraint) ClauseElement Executable) get_columnsceZdZddZdS) CreateViewFc0||_||_||_dSN)name selectable materialized)selfr rrs _/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/sqlalchemy_utils/view.py__init__zCreateView.__init__ s $(NF__name__ __module__ __qualname__rrrr r s())))))rr c d|jrdnd|jj|j|j|jdS)NzCREATE {}VIEW {} AS {} MATERIALIZED T) literal_binds) formatrdialectidentifier_preparerquoter sql_compilerprocessrelementrkws r compile_create_materialized_viewr's^ # * *"/7R,227<@@%%g&8%MM  rceZdZddZdS)DropViewFTc0||_||_||_dSr )r rcascade)rr rr+s rrzDropView.__init__s ( rN)FTrrrrr)r)s(rr)c d|jrdnd|jj|j|jrdndS)NzDROP {}VIEW IF EXISTS {} {}rrCASCADE)rrrr r!r r+r$s rcompile_drop_materialized_viewr. sP ( / /"/7R,227<@@_, "  rc \|g}|tj}ifdt|D|z}tj||g|Ri|}t dt|Ds3|t dt|D|S)Nc g|]G}tj|j|j|j|j|jHS))key primary_key)saColumnr typegetr2).0caliasess r z0create_table_from_selectable..7s`     F F AFAF++       rcg|] }|j Sr)r2r7r8s rr:z0create_table_from_selectable..Bs???! ???rcg|] }|j Sr)r r<s rr:z0create_table_from_selectable..Ds"K"K"Ka16"K"K"Kr)r3MetaDatarTableanyappend_constraintr)r rindexesmetadatar9kwargsargstables ` rcreate_table_from_selectablerG)s;==    Z((     D HT8 5d 5 5 5f 5 5E ??{:'>'>??? @ @   "K"K;z3J3J"K"K"K L    Lrc Ht|||d|tj|dt ||dtj|dfd}tj|dt |dS)ai Create a view on a given metadata :param name: The name of the view to create. :param selectable: An SQLAlchemy selectable e.g. a select() statement. :param metadata: An SQLAlchemy Metadata instance that stores the features of the database being described. :param indexes: An optional list of SQLAlchemy Index instances. :param aliases: An optional dictionary containing with keys as column names and values as column aliases. Same as for ``create_view`` except that a ``CREATE MATERIALIZED VIEW`` statement is emitted instead of a ``CREATE VIEW``. N)r rrBrCr9 after_createT)rcFjD]}||dSr rBcreatetarget connectionr&idxrFs rcreate_indexesz0create_materialized_view..create_indexesn3= # #C JJz " " " " # #r before_droprGr3eventlistenr listens_forr))r rrCrBr9rQrFs @rcreate_materialized_viewrXIs. )     EHOO4$777 X(N33####43#HOOD))) LrTc@t||dtj|dt ||tj|dfd}tj|dt ||S)a5 Create a view on a given metadata :param name: The name of the view to create. :param selectable: An SQLAlchemy selectable e.g. a select() statement. :param metadata: An SQLAlchemy Metadata instance that stores the features of the database being described. The process for creating a view is similar to the standard way that a table is constructed, except that a selectable is provided instead of a set of columns. The view is created once a ``CREATE`` statement is executed against the supplied metadata (e.g. ``metadata.create_all(..)``), and dropped when a ``DROP`` is executed against the metadata. To create a view that performs basic filtering on a table. :: metadata = MetaData() users = Table('users', metadata, Column('id', Integer, primary_key=True), Column('name', String), Column('fullname', String), Column('premium_user', Boolean, default=False), ) premium_members = select(users).where(users.c.premium_user == True) # sqlalchemy 1.3: # premium_members = select([users]).where(users.c.premium_user == True) create_view('premium_users', premium_members, metadata) metadata.create_all(engine) # View is created at this point N)r rrCrIcFjD]}||dSr rKrMs rrQz#create_view..create_indexesrRrrS)r+rT)r rrCcascade_on_droprQrFs @r create_viewr\{sL )    E HOOHnjz.J.JKKKX(N33####43#HOO/// LrceZdZdZdZdS)RefreshMaterializedViewTc"||_||_dSr )r concurrently)rr r`s rrz RefreshMaterializedView.__init__s (rN)rrr inherit_cacherrrrr^r^s(M)))))rr^cd|jrdnd|jj|jS)Nz.REFRESH MATERIALIZED VIEW {concurrently}{name}z CONCURRENTLY r)r`r )rr`rr r!r )r%rs r!compile_refresh_materialized_viewrcsH ; B B(/(<D__"   1 7 7 E E C  rFct||t||dS)aD Refreshes an already existing materialized view :param session: An SQLAlchemy Session instance. :param name: The name of the materialized view to refresh. :param concurrently: Optional flag that causes the ``CONCURRENTLY`` parameter to be specified when the materialized view is refreshed. N)flushexecuter^)sessionr r`s rrefresh_materialized_viewrhs4 MMOOO OO+D,??@@@@@r)NNN)NN)Tr) sqlalchemyr3sqlalchemy.extrsqlalchemy.schemarrsqlalchemy.sql.expressionrrsqlalchemy_utils.functionsrr compilesr'r)r.rGrXr\r^rcrhrrrros######>>>>>>>>????????222222)))))))):z8    H   ////l 8888v)))))j-)))*++,+ A A A A A Ar