from playhouse.migrate import *
import db_engine

# Automatically select correct migrator based on database type
if hasattr(db_engine.db, '__class__') and 'Postgres' in db_engine.db.__class__.__name__:
    migrator = PostgresqlMigrator(db_engine.db)
else:
    migrator = SqliteMigrator(db_engine.db)

# db_engine.db.create_tables([db_engine.PackTheme])

# pubdate_field = DateTimeField(null=True)
# comment_field = TextField(default='')


# pitcher_injury = IntegerField(null=True)
# offense_col = IntegerField(null=True)
# pos_2 = CharField(null=True)
# last_game = CharField(null=True)
# game_type = CharField(null=True)
mlb_player = ForeignKeyField(db_engine.MlbPlayer, field=db_engine.MlbPlayer.id, null=True)
result = ForeignKeyField(db_engine.Result, field=db_engine.Result.id, null=True)
# active_theme = ForeignKeyField(PackTheme, to_field='id', field_type=int, null=True)
# active_theme = ForeignKeyField(db_engine.PackTheme, field=db_engine.PackTheme.id, null=True)  # for careers
# game_type = CharField(null=True)
# pack_team = ForeignKeyField(db_engine.Team, field=db_engine.Team.id, null=True)
# pack_cardset = ForeignKeyField(db_engine.Cardset, field=db_engine.Cardset.id, null=True)
pull_rate = FloatField(default=0.333)

migrate(
    # migrator.add_column('current', 'active_theme_id', active_theme),
    # migrator.add_column('pack', 'pack_team_id', pack_team),
    # migrator.add_column('player', 'mlbplayer_id', mlb_player),
    migrator.add_column('battingstat', 'result_id', result),
    migrator.add_column('pitchingstat', 'result_id', result),
    # migrator.add_column('battingcardratings', 'pull_rate', pull_rate),
    # migrator.rename_column('cardset', 'available', 'for_purchase')
    # migrator.add_column('player', 'offense_col', offense_col),
    # migrator.add_column('comment_tbl', 'comment', comment_field),
    # migrator.rename_column('story', 'pub_date', 'publish_date'),
    # migrator.drop_column('story', 'some_old_field'),
    # migrator.drop_not_null('team', 'abbrev'),
    # migrator.add_not_null('story', 'modified_date'),
    # migrator.rename_table('story', 'stories_tbl'),
    # migrator.drop_index('team', 'team_abbrev'),
    # migrator.drop_index('player', 'player_name')
    # migrator.add_column('pack', 'pack_type', pack_type),
)

db_engine.db.close()