示例#1
0
    def teardown():
        app.logger.info("Dropping Database")
        try:
            db.drop_all()
        except:
            app.logger.info("No tables detected.")
        app.logger.info("Dropping Alembic table")

        migrations = os.path.join(os.path.dirname(os.path.realpath(__file__)),
                                  r'../migrations')
        app.logger.info(f"Migrations Folder: {migrations}")
        app.logger.info(
            f"Migrations folder exists: {os.path.exists(migrations)}")

        app.logger.info("Removing migrations folder.")
        try:
            shutil.rmtree(migrations)
        except FileNotFoundError:
            app.logger.info("Migrations folder does not exists.")
        app.logger.info(
            f"Migrations folder exists: {os.path.exists(migrations)}")

        try:
            db.engine.execute("DROP TABLE alembic_version")
        except:
            app.logger.info("Alemic Table does not exists.")
示例#2
0
    def forge(user, follow, photo, file, tag, collect, comment):
        """Generate fake data."""

        from server.fakes import fake_admin, fake_comment, fake_follow, fake_photo, fake_file, fake_tag, fake_user, fake_collect

        db.drop_all()
        db.create_all()

        click.echo('Initializing the roles and permissions...')
        Role.init_role()
        click.echo('Generating the administrator...')
        fake_admin()
        click.echo('Generating %d users...' % user)
        fake_user(user)
        click.echo('Generating %d follows...' % follow)
        fake_follow(follow)
        click.echo('Generating %d tags...' % tag)
        fake_tag(tag)
        click.echo('Generating %d photos...' % photo)
        fake_photo(photo)
        click.echo("Generating files ")
        fake_file(file)
        click.echo('Generating %d collects...' % photo)
        fake_collect(collect)
        click.echo('Generating %d comments...' % comment)
        fake_comment(comment)
        click.echo('Done.')
示例#3
0
def init():
    """
    Initialize the database.

    :return: db session create_all result
    """
    db.drop_all()
    return db.create_all()
示例#4
0
def create_db():
    """Drop and create the database."""
    # Create uuid-ossp extension, if it doesn't exist.
    db.session.execute('CREATE EXTENSION IF NOT EXISTS "uuid-ossp";')
    db.session.commit()

    db.reflect()
    db.drop_all()
    db.create_all()
示例#5
0
def drop(force=False):
    """
    Drop all database tables
    """
    if force or prompt_bool("Are you sure you want to drop all the database tables?"):
        db.drop_all()
        return True

    return False
示例#6
0
 def initdb(drop):
     """Initialize the database."""
     if drop:
         click.confirm(
             'This operation will delete the database, do you want to continue?',
             abort=True)
         db.drop_all()
         click.echo('Drop tables.')
     db.create_all()
     click.echo('Initialized database.')
示例#7
0
def dbmodels(app):
    """Database and models.

    If the database defined in `SQLALCHEMY_DATABASE_URI` already exists, it is
    dropped and re-created.
    """
    with app.app_context():
        # If there is an existing database, make sure to drop it and re-create
        # it in order to make sure that we're getting a clean testrun.
        db.drop_all()
        db.create_all()
示例#8
0
def db(app):
    """
    Setup our database, this only gets executed once per session.

    :param app: Pytest fixture
    :return: SQLAlchemy database session
    """
    _db.drop_all()
    _db.create_all()

    return _db
示例#9
0
def init_database():
    # Create the database and the database table
    db.create_all()

    # Insert user data
    # ...

    # Commit the changes for the users
    db.session.commit()

    yield db  # this is where the testing happens!

    db.drop_all()
示例#10
0
def init(with_testdb):
    """
    Initialize the database.

    :param with_testdb: Create a test database
    :return: None
    """
    db.reflect()
    db.drop_all()
    db.create_all()

    if with_testdb:
        db_uri = '{0}_test'.format(app.config['SQLALCHEMY_DATABASE_URI'])

        if not database_exists(db_uri):
            create_database(db_uri)

    return None
def db(app):
    """
    Setup database, executed once per session.

    :param app: Pytest fixture
    :return: SQLAlchemy database session
    """
    _db.drop_all()
    _db.create_all()

    # Create a single user because a lot of tests do not mutate this user.
    params = {
        'role': 'admin',
        'email': '*****@*****.**',
        'password': '******',
        'coins': 100
    }

    admin = User(**params)

    _db.session.add(admin)
    _db.session.commit()

    return _db
示例#12
0
 def setUp(self):
     super(CLITestCase, self).setUp()
     db.drop_all()
示例#13
0
 def tearDownClass(cls):
     db.session.remove()
     db.drop_all()
示例#14
0
from server.app import create_app
from server.elastic_models import SamVendorsIndex
from server.database_models import SamVendors


load_dotenv(find_dotenv())


if __name__ == "__main__":

    app = create_app()

    # create tables on the database
    with app.app_context(): 
        print("Dropping tables...")
        db.drop_all()

        print("Dropping functions...")
        db.engine.execute("DROP FUNCTION IF EXISTS modify_updated_at CASCADE;")

        print("Creating tables.")
        db.create_all()


    with open("sam_gov_data.csv") as f:
        data = [tuple(line) for line in csv.reader(f)]


    insert_statment = """
        INSERT INTO sam_vendors(
            duns,
示例#15
0
 def tearDown(self):
     with self.app.app_context():
         db.session.remove()
         db.drop_all()
示例#16
0
def drop_db():
    """Drops the db tables."""
    db.drop_all()
示例#17
0
 def tearDown(self):
     with self.app.app_context():
         db.session.remove()
         db.drop_all()