7 changed files with 399 additions and 11 deletions
			
			
		@ -0,0 +1,45 @@ | 
				
			|||||||
 | 
					# A generic, single database configuration. | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					[alembic] | 
				
			||||||
 | 
					# template used to generate migration files | 
				
			||||||
 | 
					# file_template = %%(rev)s_%%(slug)s | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# set to 'true' to run the environment during | 
				
			||||||
 | 
					# the 'revision' command, regardless of autogenerate | 
				
			||||||
 | 
					# revision_environment = false | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Logging configuration | 
				
			||||||
 | 
					[loggers] | 
				
			||||||
 | 
					keys = root,sqlalchemy,alembic | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					[handlers] | 
				
			||||||
 | 
					keys = console | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					[formatters] | 
				
			||||||
 | 
					keys = generic | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					[logger_root] | 
				
			||||||
 | 
					level = WARN | 
				
			||||||
 | 
					handlers = console | 
				
			||||||
 | 
					qualname = | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					[logger_sqlalchemy] | 
				
			||||||
 | 
					level = WARN | 
				
			||||||
 | 
					handlers = | 
				
			||||||
 | 
					qualname = sqlalchemy.engine | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					[logger_alembic] | 
				
			||||||
 | 
					level = INFO | 
				
			||||||
 | 
					handlers = | 
				
			||||||
 | 
					qualname = alembic | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					[handler_console] | 
				
			||||||
 | 
					class = StreamHandler | 
				
			||||||
 | 
					args = (sys.stderr,) | 
				
			||||||
 | 
					level = NOTSET | 
				
			||||||
 | 
					formatter = generic | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					[formatter_generic] | 
				
			||||||
 | 
					format = %(levelname)-5.5s [%(name)s] %(message)s | 
				
			||||||
 | 
					datefmt = %H:%M:%S | 
				
			||||||
@ -0,0 +1,96 @@ | 
				
			|||||||
 | 
					from __future__ import with_statement | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import logging | 
				
			||||||
 | 
					from logging.config import fileConfig | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					from sqlalchemy import engine_from_config | 
				
			||||||
 | 
					from sqlalchemy import pool | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					from alembic import context | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# this is the Alembic Config object, which provides | 
				
			||||||
 | 
					# access to the values within the .ini file in use. | 
				
			||||||
 | 
					config = context.config | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Interpret the config file for Python logging. | 
				
			||||||
 | 
					# This line sets up loggers basically. | 
				
			||||||
 | 
					fileConfig(config.config_file_name) | 
				
			||||||
 | 
					logger = logging.getLogger('alembic.env') | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# add your model's MetaData object here | 
				
			||||||
 | 
					# for 'autogenerate' support | 
				
			||||||
 | 
					# from myapp import mymodel | 
				
			||||||
 | 
					# target_metadata = mymodel.Base.metadata | 
				
			||||||
 | 
					from flask import current_app | 
				
			||||||
 | 
					config.set_main_option( | 
				
			||||||
 | 
					    'sqlalchemy.url', | 
				
			||||||
 | 
					    str(current_app.extensions['migrate'].db.engine.url).replace('%', '%%')) | 
				
			||||||
 | 
					target_metadata = current_app.extensions['migrate'].db.metadata | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# other values from the config, defined by the needs of env.py, | 
				
			||||||
 | 
					# can be acquired: | 
				
			||||||
 | 
					# my_important_option = config.get_main_option("my_important_option") | 
				
			||||||
 | 
					# ... etc. | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def run_migrations_offline(): | 
				
			||||||
 | 
					    """Run migrations in 'offline' mode. | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    This configures the context with just a URL | 
				
			||||||
 | 
					    and not an Engine, though an Engine is acceptable | 
				
			||||||
 | 
					    here as well.  By skipping the Engine creation | 
				
			||||||
 | 
					    we don't even need a DBAPI to be available. | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    Calls to context.execute() here emit the given string to the | 
				
			||||||
 | 
					    script output. | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    """ | 
				
			||||||
 | 
					    url = config.get_main_option("sqlalchemy.url") | 
				
			||||||
 | 
					    context.configure( | 
				
			||||||
 | 
					        url=url, target_metadata=target_metadata, literal_binds=True | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    with context.begin_transaction(): | 
				
			||||||
 | 
					        context.run_migrations() | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def run_migrations_online(): | 
				
			||||||
 | 
					    """Run migrations in 'online' mode. | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    In this scenario we need to create an Engine | 
				
			||||||
 | 
					    and associate a connection with the context. | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    """ | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    # this callback is used to prevent an auto-migration from being generated | 
				
			||||||
 | 
					    # when there are no changes to the schema | 
				
			||||||
 | 
					    # reference: http://alembic.zzzcomputing.com/en/latest/cookbook.html | 
				
			||||||
 | 
					    def process_revision_directives(context, revision, directives): | 
				
			||||||
 | 
					        if getattr(config.cmd_opts, 'autogenerate', False): | 
				
			||||||
 | 
					            script = directives[0] | 
				
			||||||
 | 
					            if script.upgrade_ops.is_empty(): | 
				
			||||||
 | 
					                directives[:] = [] | 
				
			||||||
 | 
					                logger.info('No changes in schema detected.') | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    connectable = engine_from_config( | 
				
			||||||
 | 
					        config.get_section(config.config_ini_section), | 
				
			||||||
 | 
					        prefix='sqlalchemy.', | 
				
			||||||
 | 
					        poolclass=pool.NullPool, | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    with connectable.connect() as connection: | 
				
			||||||
 | 
					        context.configure( | 
				
			||||||
 | 
					            connection=connection, | 
				
			||||||
 | 
					            target_metadata=target_metadata, | 
				
			||||||
 | 
					            process_revision_directives=process_revision_directives, | 
				
			||||||
 | 
					            **current_app.extensions['migrate'].configure_args | 
				
			||||||
 | 
					        ) | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        with context.begin_transaction(): | 
				
			||||||
 | 
					            context.run_migrations() | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					if context.is_offline_mode(): | 
				
			||||||
 | 
					    run_migrations_offline() | 
				
			||||||
 | 
					else: | 
				
			||||||
 | 
					    run_migrations_online() | 
				
			||||||
@ -0,0 +1,24 @@ | 
				
			|||||||
 | 
					"""${message} | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Revision ID: ${up_revision} | 
				
			||||||
 | 
					Revises: ${down_revision | comma,n} | 
				
			||||||
 | 
					Create Date: ${create_date} | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					""" | 
				
			||||||
 | 
					from alembic import op | 
				
			||||||
 | 
					import sqlalchemy as sa | 
				
			||||||
 | 
					${imports if imports else ""} | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# revision identifiers, used by Alembic. | 
				
			||||||
 | 
					revision = ${repr(up_revision)} | 
				
			||||||
 | 
					down_revision = ${repr(down_revision)} | 
				
			||||||
 | 
					branch_labels = ${repr(branch_labels)} | 
				
			||||||
 | 
					depends_on = ${repr(depends_on)} | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def upgrade(): | 
				
			||||||
 | 
					    ${upgrades if upgrades else "pass"} | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def downgrade(): | 
				
			||||||
 | 
					    ${downgrades if downgrades else "pass"} | 
				
			||||||
@ -0,0 +1,86 @@ | 
				
			|||||||
 | 
					"""update models to database documentation 1.2 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Revision ID: 08f2f998a4f6 | 
				
			||||||
 | 
					Revises: db3681fb9273 | 
				
			||||||
 | 
					Create Date: 2020-07-02 18:45:23.912159 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					""" | 
				
			||||||
 | 
					from alembic import op | 
				
			||||||
 | 
					import sqlalchemy as sa | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# revision identifiers, used by Alembic. | 
				
			||||||
 | 
					revision = '08f2f998a4f6' | 
				
			||||||
 | 
					down_revision = 'db3681fb9273' | 
				
			||||||
 | 
					branch_labels = None | 
				
			||||||
 | 
					depends_on = None | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def upgrade(): | 
				
			||||||
 | 
					    # ### commands auto generated by Alembic - please adjust! ### | 
				
			||||||
 | 
					    op.create_table('location', | 
				
			||||||
 | 
					    sa.Column('id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('player_id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('longitude', sa.Numeric(precision=15, scale=10, asdecimal=False), nullable=False), | 
				
			||||||
 | 
					    sa.Column('latitude', sa.Numeric(precision=15, scale=10, asdecimal=False), nullable=False), | 
				
			||||||
 | 
					    sa.Column('timestamp', sa.DateTime(), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=False), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['player_id'], ['player.id'], ), | 
				
			||||||
 | 
					    sa.PrimaryKeyConstraint('id') | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    op.create_table('notification', | 
				
			||||||
 | 
					    sa.Column('id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('game_id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('message', sa.Text(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('type', sa.String(length=64), nullable=False), | 
				
			||||||
 | 
					    sa.Column('timestamp', sa.DateTime(), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=False), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['game_id'], ['game.id'], ), | 
				
			||||||
 | 
					    sa.PrimaryKeyConstraint('id') | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    op.create_table('objective', | 
				
			||||||
 | 
					    sa.Column('id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('name', sa.String(length=64), nullable=False), | 
				
			||||||
 | 
					    sa.Column('game_id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('hash', sa.String(length=128), nullable=False), | 
				
			||||||
 | 
					    sa.Column('longitude', sa.Numeric(precision=15, scale=10, asdecimal=False), nullable=True), | 
				
			||||||
 | 
					    sa.Column('latitude', sa.Numeric(precision=15, scale=10, asdecimal=False), nullable=True), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['game_id'], ['game.id'], ), | 
				
			||||||
 | 
					    sa.PrimaryKeyConstraint('id'), | 
				
			||||||
 | 
					    sa.UniqueConstraint('hash') | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    op.create_table('player_caught_player', | 
				
			||||||
 | 
					    sa.Column('catching_player_id', sa.Integer(), nullable=True), | 
				
			||||||
 | 
					    sa.Column('caught_player_id', sa.Integer(), nullable=True), | 
				
			||||||
 | 
					    sa.Column('photo_reference', sa.String(length=128), nullable=False), | 
				
			||||||
 | 
					    sa.Column('timestamp', sa.DateTime(), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=False), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['catching_player_id'], ['player.id'], ), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['caught_player_id'], ['player.id'], ), | 
				
			||||||
 | 
					    sa.UniqueConstraint('photo_reference') | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    op.create_table('player_found_objective', | 
				
			||||||
 | 
					    sa.Column('game_id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('player_id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('timestamp', sa.DateTime(), server_default=sa.text('(CURRENT_TIMESTAMP)'), nullable=False), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['game_id'], ['game.id'], ), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['player_id'], ['player.id'], ), | 
				
			||||||
 | 
					    sa.PrimaryKeyConstraint('game_id', 'player_id') | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    op.create_table('notification_player', | 
				
			||||||
 | 
					    sa.Column('notification_id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('player_id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('been_shown', sa.Boolean(), server_default='True', nullable=False), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['notification_id'], ['notification.id'], ), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['player_id'], ['player.id'], ), | 
				
			||||||
 | 
					    sa.PrimaryKeyConstraint('notification_id', 'player_id') | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    # ### end Alembic commands ### | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def downgrade(): | 
				
			||||||
 | 
					    # ### commands auto generated by Alembic - please adjust! ### | 
				
			||||||
 | 
					    op.drop_table('notification_player') | 
				
			||||||
 | 
					    op.drop_table('player_found_objective') | 
				
			||||||
 | 
					    op.drop_table('player_caught_player') | 
				
			||||||
 | 
					    op.drop_table('objective') | 
				
			||||||
 | 
					    op.drop_table('notification') | 
				
			||||||
 | 
					    op.drop_table('location') | 
				
			||||||
 | 
					    # ### end Alembic commands ### | 
				
			||||||
@ -0,0 +1,56 @@ | 
				
			|||||||
 | 
					"""add game and player model | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Revision ID: db3681fb9273 | 
				
			||||||
 | 
					Revises:  | 
				
			||||||
 | 
					Create Date: 2020-07-02 13:26:00.493728 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					""" | 
				
			||||||
 | 
					from alembic import op | 
				
			||||||
 | 
					import sqlalchemy as sa | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# revision identifiers, used by Alembic. | 
				
			||||||
 | 
					revision = 'db3681fb9273' | 
				
			||||||
 | 
					down_revision = None | 
				
			||||||
 | 
					branch_labels = None | 
				
			||||||
 | 
					depends_on = None | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def upgrade(): | 
				
			||||||
 | 
					    # ### commands auto generated by Alembic - please adjust! ### | 
				
			||||||
 | 
					    op.create_table('game', | 
				
			||||||
 | 
					    sa.Column('id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('name', sa.String(length=64), nullable=False), | 
				
			||||||
 | 
					    sa.Column('password_hash', sa.String(length=128), nullable=False), | 
				
			||||||
 | 
					    sa.Column('state', sa.Integer(), nullable=True), | 
				
			||||||
 | 
					    sa.Column('start_time', sa.DateTime(), nullable=True), | 
				
			||||||
 | 
					    sa.Column('end_time', sa.DateTime(), nullable=True), | 
				
			||||||
 | 
					    sa.PrimaryKeyConstraint('id') | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    op.create_index(op.f('ix_game_name'), 'game', ['name'], unique=True) | 
				
			||||||
 | 
					    op.create_table('player', | 
				
			||||||
 | 
					    sa.Column('id', sa.Integer(), nullable=False), | 
				
			||||||
 | 
					    sa.Column('name', sa.String(length=64), nullable=False), | 
				
			||||||
 | 
					    sa.Column('auth_hash', sa.String(length=128), nullable=False), | 
				
			||||||
 | 
					    sa.Column('password_hash', sa.String(length=128), nullable=True), | 
				
			||||||
 | 
					    sa.PrimaryKeyConstraint('id'), | 
				
			||||||
 | 
					    sa.UniqueConstraint('auth_hash'), | 
				
			||||||
 | 
					    sa.UniqueConstraint('name') | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    op.create_table('game_player', | 
				
			||||||
 | 
					    sa.Column('game_id', sa.Integer(), nullable=True), | 
				
			||||||
 | 
					    sa.Column('player_id', sa.Integer(), nullable=True), | 
				
			||||||
 | 
					    sa.Column('role', sa.String(length=16), nullable=True), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['game_id'], ['game.id'], ), | 
				
			||||||
 | 
					    sa.ForeignKeyConstraint(['player_id'], ['player.id'], ) | 
				
			||||||
 | 
					    ) | 
				
			||||||
 | 
					    # ### end Alembic commands ### | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def downgrade(): | 
				
			||||||
 | 
					    # ### commands auto generated by Alembic - please adjust! ### | 
				
			||||||
 | 
					    op.drop_table('game_player') | 
				
			||||||
 | 
					    op.drop_table('player') | 
				
			||||||
 | 
					    op.drop_index(op.f('ix_game_name'), table_name='game') | 
				
			||||||
 | 
					    op.drop_table('game') | 
				
			||||||
 | 
					    # ### end Alembic commands ### | 
				
			||||||
					Loading…
					
					
				
		Reference in new issue