在使用alembic開(kāi)發(fā)管理數(shù)據(jù)庫(kù)時(shí)，會(huì)遇到一個(gè)比較麻煩的問(wèn)題，就是變更某列的枚舉類型，事實(shí)上使用sql命令變更相當(dāng)?shù)暮?jiǎn)單，一條alter的執(zhí)行即可：

創(chuàng)新互聯(lián)公司服務(wù)項(xiàng)目包括方城網(wǎng)站建設(shè)、方城網(wǎng)站制作、方城網(wǎng)頁(yè)制作以及方城網(wǎng)絡(luò)營(yíng)銷策劃等。多年來(lái)，我們專注于互聯(lián)網(wǎng)行業(yè)，利用自身積累的技術(shù)優(yōu)勢(shì)、行業(yè)經(jīng)驗(yàn)、深度合作伙伴關(guān)系等，向廣大中小型企業(yè)、政府機(jī)構(gòu)等提供互聯(lián)網(wǎng)行業(yè)的解決方案，方城網(wǎng)站推廣取得了明顯的社會(huì)效益與經(jīng)濟(jì)效益。目前，我們服務(wù)的客戶以成都為中心已經(jīng)輻射到方城省份的部分城市，未來(lái)相信會(huì)繼續(xù)擴(kuò)大服務(wù)區(qū)域并繼續(xù)獲得客戶的支持與信任！ALTER TYPE status ADD value 'output_limit_exceeded' after 'timed_out'; #刪除 DROP TYPE status

但這樣并不能滿足alembic管理的初衷，也無(wú)法實(shí)現(xiàn)downgrade。

    使用google搜索關(guān)鍵字“alembic enum type”，第一個(gè)出現(xiàn)的是stackflow的一個(gè)帖子Altering an Enum field using Alembic。

from alembic import opimport sqlalchemy as sa old_options = ('nonexistent_executable', 'signal', 'success', 'timed_out') new_options = sorted(old_options + ('output_limit_exceeded',))old_type = sa.Enum(*old_options, name='status')new_type = sa.Enum(*new_options, name='status')tmp_type = sa.Enum(*new_options, name='_status')tcr = sa.sql.table('testcaseresult',                    sa.Column('status', new_type, nullable=False))                     def upgrade():     # Create a tempoary "_status" type, convert and drop the "old" type     tmp_type.create(op.get_bind(), checkfirst=False)     op.execute('ALTER TABLE testcaseresult ALTER COLUMN status TYPE _status'                ' USING status::text::_status')     old_type.drop(op.get_bind(), checkfirst=False)     # Create and convert to the "new" status type     new_type.create(op.get_bind(), checkfirst=False)     op.execute('ALTER TABLE testcaseresult ALTER COLUMN status TYPE status'                ' USING status::text::status')     tmp_type.drop(op.get_bind(), checkfirst=False)           def downgrade():     # Convert 'output_limit_exceeded' status into 'timed_out'     op.execute(tcr.update().where(tcr.c.status==u'output_limit_exceeded')                .values(status='timed_out'))     # Create a tempoary "_status" type, convert and drop the "new" type     tmp_type.create(op.get_bind(), checkfirst=False)     op.execute('ALTER TABLE testcaseresult ALTER COLUMN status TYPE _status'                ' USING status::text::_status')     new_type.drop(op.get_bind(), checkfirst=False)     # Create and convert to the "old" status type     old_type.create(op.get_bind(), checkfirst=False)     op.execute('ALTER TABLE testcaseresult ALTER COLUMN status TYPE status'                ' USING status::text::status')     tmp_type.drop(op.get_bind(), checkfirst=False)

這個(gè)方法提供了解決了問(wèn)題，但稍顯繁瑣。我們可以做一下簡(jiǎn)單的封裝，生成一個(gè)通用函數(shù)：

def upgrade_enum(table, column_name, enum_name, old_options, new_options):     old_type = sa.Enum(*old_options, name=enum_name)     new_type = sa.Enum(*new_options, name=enum_name)     tmp_type = sa.Enum(*new_options, name="_" + enum_name)     # Create a temporary type, convert and drop the "old" type     tmp_type.create(op.get_bind(), checkfirst=False)     op.execute(         u'ALTER TABLE {0} ALTER COLUMN {1} TYPE _{2}'         u' USING {1}::text::_{2}'.format(             table,             column_name,             enum_name         )     )     old_type.drop(op.get_bind(), checkfirst=False)     # Create and convert to the "new" type     new_type.create(op.get_bind(), checkfirst=False)     op.execute(         u'ALTER TABLE {0} ALTER COLUMN {1} TYPE {2}'         u' USING {1}::text::{2}'.format(             table,             column_name,             enum_name         )     )     tmp_type.drop(op.get_bind(), checkfirst=False)

 這樣就可以直接通過(guò)傳參直接來(lái)執(zhí)行升級(jí)或降級(jí)操作了。

操作環(huán)境：

 alembic-0.8.2 -bash-4.2$ psql --version psql (PostgreSQL) 9.3.10

另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)cdcxhl.cn，海內(nèi)外云服務(wù)器15元起步，三天無(wú)理由+7*72小時(shí)售后在線，公司持有idc許可證，提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國(guó)服務(wù)器、虛擬主機(jī)、免備案服務(wù)器”等云主機(jī)租用服務(wù)以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡(jiǎn)單易用、服務(wù)可用性高、性價(jià)比高”等特點(diǎn)與優(yōu)勢(shì)，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應(yīng)用場(chǎng)景需求。

分享題目：Openstack數(shù)據(jù)庫(kù)管理工具alembic更新Enum類型-創(chuàng)新互聯(lián)
文章分享：http://www.chinadenli.net/article10/ceecgo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供網(wǎng)站改版、關(guān)鍵詞優(yōu)化、網(wǎng)站制作、網(wǎng)站排名、網(wǎng)頁(yè)設(shè)計(jì)公司、自適應(yīng)網(wǎng)站

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如需處理請(qǐng)聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來(lái)源： 創(chuàng)新互聯(lián)