o KDi(@s ddlmZmZddlmZmZddlmZddlm Z m Z ddl Z e dZ Gdd d ZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZe eeZdS))Model BufferModel) DateField StringField) MergeTree)escapeget_subclass_namesN migrationsc@eZdZdZddZdS) Operationz. Base class for migration operations. cCstN)NotImplementedErrorselfdatabaserW/var/www/Datamplify/venv/lib/python3.10/site-packages/infi/clickhouse_orm/migrations.pyapplyszOperation.applyN__name__ __module__ __qualname____doc__rrrrrr  r c@ eZdZdZddZddZdS)ModelOperationzL Base class for migration operations that work on a specific model. cCs||_||_dS)z Initializer. N) model_class table_name)rrrrr__init__szModelOperation.__init__cCs&d|j|f}t|||dS)z; Utility for running ALTER TABLE commands. zALTER TABLE $db.`%s` %sN)rloggerdebugraw)rrcmdrrr _alter_tables zModelOperation._alter_tableN)rrrrrr$rrrrrs rc@r ) CreateTablezM A migration operation that creates a table for a given model class. cCs:td|jt|jtr||jjj||jdS)Nz Create table %s) r infor issubclassrr create_tableengine main_modelrrrrr-s zCreateTable.applyNrrrrrr%(rr%c@r) AlterTablea- A migration operation that compares the table of a given model class to the model's fields, and alters the table to match the model. The operation can: - add new columns - drop obsolete columns - modify column types Default values are not altered by this operation. cCs$d|j|jf}dd||DS)NzDESC `%s`.`%s`cSsg|]}|j|jfqSr)nametype).0rowrrr @sz0AlterTable._get_table_fields..)db_namerselect)rrqueryrrr_get_table_fields>szAlterTable._get_table_fieldsc sbtd|jt|}t|t|j}|D]}td|| d|||=qd}|j D]8\}}|j pC|j }||vrntd|d||j df}|rh|rd|d|7}n|d7}| ||rr|}q:fd d |j D} |D](\} } | | vsJd | | | krtd | | | | | d | | | fqdS)Nz Alter table %sz Drop column %szDROP COLUMN %sz Add column %szADD COLUMN %s %s)dbz AFTER %sz FIRSTcs i|] \}}||jddqS)F)with_default_expressionr5)get_sql)r.r,fieldrrr gsz$AlterTable.apply..z.Model fields and table columns in disagreementz. Change type of column %s from %s to %szMODIFY COLUMN %s %s)r r&rdictr4setkeysrfieldsr$items materializedaliasr7) rr table_fieldsdeleted_fieldsr, prev_namer8is_regular_fieldr# model_fields field_name field_sqlrr9rrBsB       zAlterTable.applyN)rrrrr4rrrrrr+4s r+c@r )AlterTableWithBufferz A migration operation for altering a buffer table and its underlying on-disk table. The buffer table is dropped, the on-disk table is altered, and then the buffer table is re-created. cCsXt|jtr"t|j|t|jjj|t|j|dSt|j|dSr ) r'rr DropTablerr+r)r*r%rrrrrzs zAlterTableWithBuffer.applyNrrrrrrIss rIc@r )rJzL A migration operation that drops the table of a given model class. cCstd|j||jdS)Nz Drop table %s)r r&r drop_tablerrrrrrszDropTable.applyNrrrrrrJrrJc@r)AlterConstraintsa_ A migration operation that adds new constraints from the model to the database table, and drops obsolete ones. Constraints are identified by their names, so a change in an existing constraint will not be detected unless its name was changed too. ClickHouse does not check that the constraints hold for existing data in the table. cCstd|j||}|jjD]}|j|vr+td|j||d| q| |jq|D]}td|||d|q4dS)Nz Alter constraints for %sz Add constraint %sADD %sz Drop constraint %szDROP CONSTRAINT `%s`) r r&r_get_constraint_namesr _constraintsvaluesr,r$create_table_sqlremove)rrexisting constraintr,rrrrs   zAlterConstraints.applycC,ddl}|d|j}|d|}t|S)zZ Returns a set containing the names of existing constraints in the table. r NSHOW CREATE TABLE $db.`%s`z"\sCONSTRAINT\s+`?(.+?)`?\s+CHECK\srer"rfindallr<rrrX table_defmatchesrrrrN z&AlterConstraints._get_constraint_namesN)rrrrrrNrrrrrLs rLcs2eZdZdZd fdd ZddZddZZS) AlterIndexesz A migration operation that adds new indexes from the model to the database table, and drops obsolete ones. Indexes are identified by their names, so a change in an existing index will not be detected unless its name was changed too. Fcst|||_dS)z Initializer. By default ClickHouse does not build indexes over existing data, only for new data. Passing `reindex=True` will run `OPTIMIZE TABLE` in order to build the indexes over the existing data. N)superrreindex)rrr` __class__rrrs  zAlterIndexes.__init__cCstd|j||}t||jjD]}|j|vr0td|j||d| q| |jq|D]}td|||d|q9|j r\td| d|jdSdS)Nz Alter indexes for %sz Add index %srMz Drop index %szDROP INDEX `%s`z Build indexes on tablezOPTIMIZE TABLE $db.`%s` FINAL) r r&r_get_index_namesr_indexesrPr,r$rQrRr`r")rrrSindexr,rrrrs     zAlterIndexes.applycCrU)zV Returns a set containing the names of existing indexes in the table. r NrVz\sINDEX\s+`?(.+?)`?\s+rWrZrrrrcr]zAlterIndexes._get_index_names)F)rrrrrrrc __classcell__rrrarr^s  r^c@r) RunPythonz@ A migration operation that executes a Python function. cCst|sJd||_dS)z Initializer. The given Python function will be called with a single argument - the Database instance to apply the migration to. z 'func' argument must be functionN)callable_func)rfuncrrrrs zRunPython.__init__cCstd|jj||dS)Nz! Executing python operation %s)r r&rirrrrrrszRunPython.applyNrrrrrrrrrrrgs rgc@r)RunSQLzG A migration operation that executes arbitrary SQL statements. cCs,t|tr|g}t|tsJd||_dS)zr Initializer. The given sql argument must be a valid SQL statement or list of statements. z0'sql' argument must be string or list of stringsN) isinstancestrlist_sql)rsqlrrrrs  zRunSQL.__init__cCs$td|jD]}||qdS)Nz Executing raw SQL operations)r r&rpr")rritemrrrrs   z RunSQL.applyNrkrrrrrls rlc@s8eZdZdZeZeZeZe ddZ e ddZ dS)MigrationHistoryz_ A model for storing which migrations were already applied to the containing database. applied) package_name module_namecCsdS)Ninfi_clickhouse_orm_migrationsr)clsrrrrszMigrationHistory.table_nameN) rrrrrrurvrrtrr) classmethodrrrrrrss rs)modelsrrr>rrenginesrutilsrrlogging getLoggerr r rr%r+rIrJrLr^rgrlrslocals__all__rrrrs$    ? "0