def run_migrate(*args):
'''
running some migration.
:return:
'''
print('Begin migrate ...')
torcms_migrator = migrate.PostgresqlMigrator(config.DB_CON)
float_field = migrate.FloatField(null=False, default=5)
try:
migrate.migrate(torcms_migrator.add_column('g_post', 'rating', float_field))
except:
pass
pid = migrate.CharField(null=False, max_length=4, default='xxxx', help_text='parent id')
tmpl = migrate.IntegerField(null=False, default=9, help_text='tmplate type')
try:
migrate.migrate(torcms_migrator.add_column('g_tag', 'pid', pid))
except:
pass
try:
migrate.migrate(torcms_migrator.add_column('g_tag', 'tmpl', tmpl))
except:
pass
try:
migrate.migrate(torcms_migrator.drop_column('g_tag', 'role_mask'))
except:
pass
print('QED')
def run_migrate(*args):
'''
running some migration.
:return:
'''
print('Begin migrate ...')
torcms_migrator = migrate.PostgresqlMigrator(config.DB_CON)
# try:
# migrate.migrate(
# torcms_migrator.rename_table('e_layout', 'mablayout')
# )
# except:
# pass
memo_field = migrate.TextField(null=False, default='', help_text='Memo', )
try:
migrate.migrate(torcms_migrator.add_column('tabpost', 'memo', memo_field))
except:
pass
# try:
# migrate.migrate(torcms_migrator.drop_column('tabtag', 'role_mask'))
# except:
# pass
print('QED')
def run_migrate(*args):
'''
running some migration.
:return:
'''
print('Begin migrate ...')
torcms_migrator = migrate.PostgresqlMigrator(config.DB_CON)
memo_field = migrate.TextField(null=False, default='', help_text='Memo', )
try:
migrate.migrate(torcms_migrator.add_column('tabpost', 'memo', memo_field))
except:
pass
desc_field = migrate.CharField(null=False, default='', max_length=255, help_text='')
try:
migrate.migrate(torcms_migrator.add_column('tabentity', 'desc', desc_field))
except:
pass
extinfo_field = BinaryJSONField(null=False, default={}, help_text='Extra data in JSON.')
try:
migrate.migrate(torcms_migrator.add_column('tabmember', 'extinfo', extinfo_field))
except:
pass
print('Migration finished.')
def down():
migrator = migrate.PostgresqlMigrator(database.get_connector())
migrate.migrate(migrator.drop_column('reservoir', 'latitude'),
migrator.drop_column('reservoir', 'longitude'),
migrator.drop_column('tweet', 'tweet_id'))
def run_migrate(*args):
'''
running some migration.
'''
print('Begin migrate ...')
torcms_migrator = migrate.PostgresqlMigrator(config.DB_CON)
memo_field = migrate.TextField(null=False, default='', help_text='Memo', )
try:
migrate.migrate(torcms_migrator.add_column('tabpost', 'memo', memo_field))
except:
pass
desc_field = migrate.CharField(null=False, default='', max_length=255, help_text='')
try:
migrate.migrate(torcms_migrator.add_column('tabentity', 'desc', desc_field))
except:
pass
extinfo_field = BinaryJSONField(null=False, default={}, help_text='Extra data in JSON.')
try:
migrate.migrate(torcms_migrator.add_column('tabmember', 'extinfo', extinfo_field))
except:
pass
par_id_field = migrate.CharField(null=False, default='', max_length=4,
help_text='父类id，对于label，top_id为""')
try:
migrate.migrate(torcms_migrator.add_column('tabpost2tag', 'par_id', par_id_field))
except:
pass
category_field = migrate.CharField(null=False, default='0', max_length=1,
help_text='0为评论，1为回复')
try:
migrate.migrate(torcms_migrator.add_column('tabreply', 'category', category_field))
except:
pass
print('Migration finished.')
def run_migrate(*args):
'''
running some migration.
:return:
'''
print('Begin migrate ...')
torcms_migrator = migrate.PostgresqlMigrator(config.DB_CON)
version_field = migrate.IntegerField(null=False, default=1)
try:
migrate.migrate(
torcms_migrator.add_column('mabgson', 'version', version_field))
except:
pass
print('Migration finished.')
def up():
migrator = migrate.PostgresqlMigrator(database.get_connector())
migrate.migrate(
migrator.add_column('reservoir', 'nearest_city',
peewee.CharField(null=True)),
migrator.add_column('reservoir', 'twitter_place_id',
peewee.CharField(null=True)))
res_list = json.load(open('reservoirs.json'))
for reservoir in models.Reservoir.select():
res_entries = [
r for r in res_list if r['station_id'] == reservoir.station_id
]
if len(res_entries) > 0:
res = res_entries[0]
reservoir.nearest_city = res['nearest_city']
reservoir.twitter_place_id = res['twitter_place_id']
reservoir.save()
def up():
migrator = migrate.PostgresqlMigrator(database.get_connector())
migrate.migrate(
migrator.add_column('reservoir', 'latitude',
peewee.FloatField(null=True)),
migrator.add_column('reservoir', 'longitude',
peewee.FloatField(null=True)),
migrator.add_column('tweet', 'tweet_id',
peewee.BigIntegerField(null=True)))
reservoirs = json.load(open('reservoirs.json'))
for reservoir in models.Reservoir.select():
res_entries = [
r for r in reservoirs if r['station_id'] == reservoir.station_id
]
if len(res_entries) > 0:
res = res_entries[0]
reservoir.latitude = res['latitude']
reservoir.longitude = res['longitude']
reservoir.save()
def migrate(migrations=MIGRATIONS):
db = get_db()
migrator = m.PostgresqlMigrator(db)
for migration in migrations:
migration(db, migrator)
def down():
migrator = migrate.PostgresqlMigrator(database.get_connector())
migrate.migrate(migrator.drop_column('reservoir', 'nearest_city'),
migrator.drop_column('reservoir', 'twitter_place_id'))
def run_migrate(*args):
'''
for database schema migration.
Memo for Usage:
migrate.migrate(torcms_migrator.rename_table('e_layout', 'mablayout'))
migrate.migrate(torcms_migrator.drop_column('tabtag', 'role_mask'))
'''
'''
在psql状态下查询表结构
\d
tablename
'''
try:
conn = config.DB_CON
cur = conn.cursor()
cur.execute(
'''alter table tabmember alter column user_name type character varying(255)'''
)
print(" Table TabMember altered successfully")
conn.commit()
except:
pass
print('Begin migrate ...')
torcms_migrator = migrate.PostgresqlMigrator(config.DB_CON)
memo_field = migrate.TextField(
null=False,
default='',
help_text='Memo',
)
try:
migrate.migrate(
torcms_migrator.add_column('tabpost', 'memo', memo_field))
except:
pass
desc_field = migrate.CharField(null=False,
default='',
max_length=255,
help_text='')
try:
migrate.migrate(
torcms_migrator.add_column('tabentity', 'desc', desc_field))
except:
pass
extinfo_field = BinaryJSONField(null=False,
default={},
help_text='Extra data in JSON.')
try:
migrate.migrate(
torcms_migrator.add_column('tabmember', 'extinfo', extinfo_field))
except:
pass
par_id_field = migrate.CharField(null=False,
default='',
max_length=4,
help_text='父类id，对于label，top_id为""')
try:
migrate.migrate(
torcms_migrator.add_column('tabpost2tag', 'par_id', par_id_field))
except:
pass
category_field = migrate.CharField(null=False,
default='0',
max_length=1,
help_text='0为评论，1为回复')
try:
migrate.migrate(
torcms_migrator.add_column('tabreply', 'category', category_field))
except:
pass
try:
migrate.migrate(torcms_migrator.drop_column('tabentity2user', 'count'))
# print('删除字段成功：count.')
except:
pass
user_ip_field = migrate.CharField(
null=False,
default='0',
help_text='User IP Address',
)
try:
migrate.migrate(
torcms_migrator.add_column('tabentity2user', 'user_ip',
user_ip_field))
# print('添加字段成功：user_ip.')
except:
pass
view_count_1d = migrate.IntegerField(default=0, help_text='24小时内阅读量')
try:
migrate.migrate(
torcms_migrator.add_column('tabpost', 'access_1d', view_count_1d))
except:
pass
view_count_7d = migrate.IntegerField(default=0, help_text='7*24小时内阅读量')
try:
migrate.migrate(
torcms_migrator.add_column('tabpost', 'access_7d', view_count_7d))
except:
pass
view_count_30d = migrate.IntegerField(default=0, help_text='30*24小时内阅读量')
try:
migrate.migrate(
torcms_migrator.add_column('tabpost', 'access_30d',
view_count_30d))
except:
pass
print('Migration finished.')