def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'users',
        sa.Column('uuid',
                  postgresql.UUID(as_uuid=True),
                  server_default=sa.text('uuid_generate_v4()'),
                  nullable=False),
        sa.Column('username', sa.String(length=64), nullable=True),
        sa.Column('passhash', sa.Binary(), nullable=False),
        sa.Column('salt', sa.Binary(), nullable=False),
        sa.Column('acl', sa.Integer(), nullable=True),
        sa.PrimaryKeyConstraint('uuid'), sa.UniqueConstraint('username'),
        sa.UniqueConstraint('uuid'))
    op.create_table(
        'interfaces',
        sa.Column('uuid',
                  postgresql.UUID(as_uuid=True),
                  server_default=sa.text('uuid_generate_v4()'),
                  nullable=False),
        sa.Column('device_uuid', postgresql.UUID(as_uuid=True), nullable=True),
        sa.Column('read_acl', sa.Integer(), nullable=True),
        sa.Column('write_acl', sa.Integer(), nullable=True),
        sa.ForeignKeyConstraint(
            ['device_uuid'],
            ['devices.uuid'],
        ), sa.PrimaryKeyConstraint('uuid'), sa.UniqueConstraint('device_uuid'),
        sa.UniqueConstraint('uuid'))
    op.create_unique_constraint(None, 'device_room', ['uuid'])
    op.create_unique_constraint(None, 'rooms', ['uuid'])
Ejemplo n.º 2
0
def downgrade() -> None:
    # sqlite has no `drop column` command, so we recreate the original table
    # then load it from a temp table
    conn = op.get_bind()
    conn.execute("PRAGMA legacy_alter_table=ON")
    op.rename_table("journalists", "journalists_tmp")

    op.create_table(
        "journalists",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("username", sa.String(length=255), nullable=False),
        sa.Column("pw_salt", sa.Binary(), nullable=True),
        sa.Column("pw_hash", sa.Binary(), nullable=True),
        sa.Column("is_admin", sa.Boolean(), nullable=True),
        sa.Column("otp_secret", sa.String(length=16), nullable=True),
        sa.Column("is_totp", sa.Boolean(), nullable=True),
        sa.Column("hotp_counter", sa.Integer(), nullable=True),
        sa.Column("last_token", sa.String(length=6), nullable=True),
        sa.Column("created_on", sa.DateTime(), nullable=True),
        sa.Column("last_access", sa.DateTime(), nullable=True),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("username"),
    )

    conn.execute(
        """
        INSERT INTO journalists
        SELECT id, username, pw_salt, pw_hash, is_admin, otp_secret, is_totp,
               hotp_counter, last_token, created_on, last_access
        FROM journalists_tmp
    """
    )

    op.drop_table("journalists_tmp")
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'setpoint_params',
        sa.Column('uuid',
                  postgresql.UUID(as_uuid=True),
                  server_default=sa.text('uuid_generate_v4()'),
                  nullable=False),
        sa.Column('ep_uuid', postgresql.UUID(as_uuid=True), nullable=True),
        sa.Column('value', sa.Binary(), nullable=True),
        sa.ForeignKeyConstraint(
            ['ep_uuid'],
            ['endpoints.uuid'],
        ), sa.PrimaryKeyConstraint('uuid'), sa.UniqueConstraint('uuid'))
    op.create_table(
        'setpoint_pairs',
        sa.Column('uuid',
                  postgresql.UUID(as_uuid=True),
                  server_default=sa.text('uuid_generate_v4()'),
                  nullable=False),
        sa.Column('params_uuid', postgresql.UUID(as_uuid=True), nullable=True),
        sa.Column('name', sa.String(length=64), nullable=True),
        sa.Column('value', sa.Binary(), nullable=True),
        sa.ForeignKeyConstraint(
            ['params_uuid'],
            ['setpoint_params.uuid'],
        ), sa.PrimaryKeyConstraint('uuid'), sa.UniqueConstraint('uuid'))
    op.create_unique_constraint(None, 'conditions', ['uuid'])
    op.create_unique_constraint(None, 'instruction_params', ['uuid'])
    op.create_unique_constraint(None, 'instructions', ['uuid'])

    op.execute("COMMIT")
    op.execute("ALTER TYPE drivertypeenum ADD VALUE 'setpoint'")
def upgrade():
    HASH_SIZE = 32

    users = sa.table(
        'users',
        sa.Column('id'),
        sa.Column('full_name', sa.String),
        sa.Column('email', sa.String),
        sa.Column('_salt', sa.Binary(HASH_SIZE)),
        sa.Column('_hashed_password', sa.Binary(HASH_SIZE)),
        sa.Column('_active', sa.Boolean),
        sa.Column('_is_admin', sa.Boolean),
    )

    default_password = '******'
    salt = os.urandom(HASH_SIZE)
    pwd = hashlib.sha256(salt +
                         bytes(default_password, encoding='utf8')).digest()

    op.bulk_insert(users, [{
        'full_name': 'Administrator',
        'email': '*****@*****.**',
        '_salt': salt,
        '_hashed_password': pwd,
        '_active': True,
        '_is_admin': True
    }])
Ejemplo n.º 5
0
class User(Base):
    __tablename__ = 'users'
    sid = sa.Column(sa.BigInteger, primary_key=True)
    name = sa.Column(sa.String(length=100), nullable=False)
    type_id = sa.Column(sa.Integer, sa.ForeignKey('types.id'))
    waiverSigned = sa.Column(sa.DateTime)
    photo = sa.Column(sa.String(length=100), default='')
    location_id = sa.Column(sa.INTEGER, sa.ForeignKey('locations.id'), nullable=False, primary_key=True)
    pin = sa.Column(sa.Binary(length=16))
    pin_salt = sa.Column(sa.Binary(length=16))

    def set_pin(self, pin):
        self.pin_salt = os.urandom(16)
        # 100,000 rounds of sha256 w/ a random salt
        self.pin = hashlib.pbkdf2_hmac('sha256', bytearray(pin, 'utf-8'), self.pin_salt, 100000)

    def verify_pin(self, attempt):
        digest = hashlib.pbkdf2_hmac('sha256', bytearray(attempt, 'utf-8'), self.pin_salt, 100000)
        return hmac.compare_digest(self.pin, digest)

    def __repr__(self):
        return "<Location %s>" % self.name

    type = relationship('Type')
    location = relationship('Location')
    trainings = relationship('Training', foreign_keys=[Training.trainee_id])
    access = relationship('Access', order_by='Access.timeIn')
    cards = relationship('HawkCard')

    def __repr__(self):
        return "<User A%d (%s)>" % (self.sid, self.name)
Ejemplo n.º 6
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column(
        'cluster',
        sa.Column('k8s_service_account_private_key',
                  sa.Binary(),
                  nullable=True))
    op.add_column(
        'cluster',
        sa.Column('k8s_service_account_public_key', sa.Binary(),
                  nullable=True))
    op.execute(
        'UPDATE cluster SET k8s_service_account_private_key=(SELECT key FROM credentials_data WHERE credentials_data.id = cluster.service_account_keypair_id)'
    )
    op.execute(
        'UPDATE cluster SET k8s_service_account_public_key=(SELECT cert FROM credentials_data WHERE credentials_data.id = cluster.service_account_keypair_id)'
    )
    op.alter_column('cluster',
                    'k8s_service_account_private_key',
                    nullable=False)
    op.alter_column('cluster',
                    'k8s_service_account_public_key',
                    nullable=False)
    op.drop_constraint('cluster_service_account_keypair_id_fkey',
                       'cluster',
                       type_='foreignkey')
    op.drop_column('cluster', 'service_account_keypair_id')
Ejemplo n.º 7
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('otp_session',
                    sa.Column('secret', sa.String(length=100), nullable=False),
                    sa.Column('form', sa.Binary(), nullable=True),
                    sa.Column('session', sa.Binary(), nullable=True),
                    sa.PrimaryKeyConstraint('secret'))
    op.create_table(
        'session', sa.Column('id', sa.String(length=36), nullable=False),
        sa.Column('username', sa.String(length=64), nullable=False),
        sa.Column('created', sa.DateTime(), nullable=True),
        sa.PrimaryKeyConstraint('id'))
    op.create_table('phone_codes',
                    sa.Column('code', sa.String(length=6), nullable=False),
                    sa.Column('session', sa.String(length=36), nullable=True),
                    sa.ForeignKeyConstraint(
                        ['session'],
                        ['session.id'],
                    ), sa.PrimaryKeyConstraint('code'))
    op.create_table(
        'token', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('username', sa.String(length=64), nullable=False),
        sa.Column('created', sa.DateTime(), nullable=True),
        sa.Column('token', sa.String(length=36), nullable=True),
        sa.Column('session', sa.String(length=36), nullable=True),
        sa.Column('used', sa.Boolean(), nullable=True),
        sa.ForeignKeyConstraint(
            ['session'],
            ['session.id'],
        ), sa.PrimaryKeyConstraint('id'))
Ejemplo n.º 8
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'identity', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('issuer_id', sa.Integer(), nullable=True),
        sa.Column('name', sa.String(length=255), nullable=False),
        sa.ForeignKeyConstraint(
            ['issuer_id'],
            ['identity.id'],
        ), sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('issuer_id', 'name', name='_issuer_name_uc'),
        sa.UniqueConstraint('name'))
    op.create_table(
        'pair', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('identity_id', sa.Integer(), nullable=False),
        sa.Column('issued_at', sa.DateTime(), nullable=True),
        sa.Column('cert', sa.Binary(), nullable=False),
        sa.Column('key', sa.Binary(), nullable=False),
        sa.ForeignKeyConstraint(
            ['identity_id'],
            ['identity.id'],
        ), sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('identity_id',
                            'issued_at',
                            name='_identity_issued_at_uc'))
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('tag', sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('name', sa.String(length=100), nullable=False),
                    sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('name'))
    op.create_table('user', sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('name', sa.String(), nullable=False),
                    sa.Column('email', sa.String(), nullable=False),
                    sa.Column('password_hash', sa.Binary(), nullable=False),
                    sa.Column('password_salt', sa.Binary(), nullable=False),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('email'))
    op.create_table(
        'post', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('user_id', sa.Integer(), nullable=False),
        sa.Column('title', sa.String(length=200), nullable=False),
        sa.Column('text', sa.Text(), nullable=False),
        sa.Column('published_date_time', sa.DateTime(), nullable=True),
        sa.Column('published', sa.Boolean(), nullable=True),
        sa.Column('image_uuid', sa.String(length=36), nullable=True),
        sa.Column('image_name', sa.String(length=100), nullable=True),
        sa.ForeignKeyConstraint(
            ['user_id'],
            ['user.id'],
        ), sa.PrimaryKeyConstraint('id'))
    op.create_table('post_tag',
                    sa.Column('post_id', sa.Integer(), nullable=False),
                    sa.Column('tag_id', sa.Integer(), nullable=False),
                    sa.ForeignKeyConstraint(
                        ['post_id'],
                        ['post.id'],
                    ), sa.ForeignKeyConstraint(
                        ['tag_id'],
                        ['tag.id'],
                    ), sa.PrimaryKeyConstraint('post_id', 'tag_id'))
Ejemplo n.º 10
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'trades', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('user_id', sa.Integer(), nullable=False),
        sa.Column('account_id', sa.Integer(), nullable=False),
        sa.Column('stock_id', sa.Integer(), nullable=False),
        sa.Column('trade_type',
                  sa.Enum('buy', 'sell', native_enum=False),
                  nullable=False),
        sa.Column('process_date', sa.DateTime(), nullable=False),
        sa.Column('price',
                  sa.Numeric(precision=19, scale=2, asdecimal=False),
                  nullable=False),
        sa.Column('shares', sa.Integer(), nullable=False),
        sa.PrimaryKeyConstraint('id'))
    op.create_table(
        'users', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('username', sa.String(length=80), nullable=False),
        sa.Column('password', sa.Binary(), nullable=False),
        sa.Column('salt', sa.Binary(), nullable=False),
        sa.Column('first_name', sa.String(length=50), nullable=True),
        sa.Column('last_name', sa.String(length=50), nullable=True),
        sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('username'))
    op.create_table(
        'accounts', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('user_id', sa.Integer(), nullable=True),
        sa.Column('cash_amount',
                  sa.Numeric(precision=19, scale=2, asdecimal=False),
                  nullable=True),
        sa.Column('equity_amount',
                  sa.Numeric(precision=19, scale=2, asdecimal=False),
                  nullable=True),
        sa.Column('initial_amount',
                  sa.Numeric(precision=19, scale=2, asdecimal=False),
                  nullable=True),
        sa.ForeignKeyConstraint(
            ['user_id'],
            ['users.id'],
        ), sa.PrimaryKeyConstraint('id'))
    op.create_table(
        'stocks', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('account_id', sa.Integer(), nullable=True),
        sa.Column('bought_at',
                  sa.Numeric(precision=19, scale=2, asdecimal=False),
                  nullable=False),
        sa.Column('bought_on', sa.DateTime(), nullable=False),
        sa.Column('sold_on', sa.DateTime(), nullable=True),
        sa.Column('initial_cost',
                  sa.Numeric(precision=19, scale=2, asdecimal=False),
                  nullable=False),
        sa.Column('shares', sa.Integer(), nullable=False),
        sa.Column('symbol', sa.String(length=5), nullable=False),
        sa.ForeignKeyConstraint(
            ['account_id'],
            ['accounts.id'],
        ), sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('symbol'))
Ejemplo n.º 11
0
def upgrade():
    op.create_table(
        "users",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("name", sa.String(50), nullable=False),
        sa.Column("username", sa.String(50), nullable=False, unique=True),
        sa.Column("password", sa.Binary(), nullable=False),
        sa.Column("salt", sa.Binary(), nullable=False),
        sa.Column("is_authenticated", sa.Boolean(), nullable=False, default=False),
    )
Ejemplo n.º 12
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('blog_posts',
                    sa.Column('created_at', sa.DateTime(), nullable=True),
                    sa.Column('modified_at', sa.DateTime(), nullable=True),
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('author_id', sa.Integer(), nullable=True),
                    sa.Column('title', sa.String(length=100), nullable=False),
                    sa.Column('body', sa.String(length=200), nullable=False),
                    sa.Column('image', sa.Binary(), nullable=False),
                    sa.ForeignKeyConstraint(
                        ['author_id'],
                        ['users.id'],
                    ), sa.PrimaryKeyConstraint('id'))
    op.create_table(
        'bookmarks_user', sa.Column('created_at', sa.DateTime(),
                                    nullable=True),
        sa.Column('modified_at', sa.DateTime(), nullable=True),
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('user_id', sa.Integer(), nullable=True),
        sa.Column('bookmark_user_id', sa.Integer(), nullable=True),
        sa.ForeignKeyConstraint(
            ['bookmark_user_id'],
            ['users.id'],
        ), sa.ForeignKeyConstraint(
            ['user_id'],
            ['users.id'],
        ), sa.PrimaryKeyConstraint('id'))
    op.create_table('profiles',
                    sa.Column('created_at', sa.DateTime(), nullable=True),
                    sa.Column('modified_at', sa.DateTime(), nullable=True),
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('user_id', sa.Integer(), nullable=True),
                    sa.Column('nickname', sa.String(length=80), nullable=True),
                    sa.Column('comment', sa.String(length=200), nullable=True),
                    sa.Column('icon', sa.Binary(), nullable=True),
                    sa.ForeignKeyConstraint(
                        ['user_id'],
                        ['users.id'],
                    ), sa.PrimaryKeyConstraint('id'))
    op.create_table(
        'bookmarks_post', sa.Column('created_at', sa.DateTime(),
                                    nullable=True),
        sa.Column('modified_at', sa.DateTime(), nullable=True),
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('user_id', sa.Integer(), nullable=True),
        sa.Column('bookmark_post_id', sa.Integer(), nullable=True),
        sa.ForeignKeyConstraint(
            ['bookmark_post_id'],
            ['blog_posts.id'],
        ), sa.ForeignKeyConstraint(
            ['user_id'],
            ['users.id'],
        ), sa.PrimaryKeyConstraint('id'))
def upgrade() -> None:
    conn = op.get_bind()
    conn.execute("PRAGMA legacy_alter_table=ON")
    # Save existing journalist table.
    op.rename_table("journalists", "journalists_tmp")

    # Add nonce column.
    op.add_column("journalists_tmp", sa.Column("session_nonce", sa.Integer()))

    # Populate nonce column.
    journalists = conn.execute(
        sa.text("SELECT * FROM journalists_tmp")).fetchall()

    for journalist in journalists:
        conn.execute(
            sa.text("""UPDATE journalists_tmp SET session_nonce=0 WHERE
                       id=:id""").bindparams(id=journalist.id))

    # Now create new table with null constraint applied.
    op.create_table(
        "journalists",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("uuid", sa.String(length=36), nullable=False),
        sa.Column("username", sa.String(length=255), nullable=False),
        sa.Column("first_name", sa.String(length=255), nullable=True),
        sa.Column("last_name", sa.String(length=255), nullable=True),
        sa.Column("pw_salt", sa.Binary(), nullable=True),
        sa.Column("pw_hash", sa.Binary(), nullable=True),
        sa.Column("passphrase_hash", sa.String(length=256), nullable=True),
        sa.Column("is_admin", sa.Boolean(), nullable=True),
        sa.Column("session_nonce", sa.Integer(), nullable=False),
        sa.Column("otp_secret", sa.String(length=16), nullable=True),
        sa.Column("is_totp", sa.Boolean(), nullable=True),
        sa.Column("hotp_counter", sa.Integer(), nullable=True),
        sa.Column("last_token", sa.String(length=6), nullable=True),
        sa.Column("created_on", sa.DateTime(), nullable=True),
        sa.Column("last_access", sa.DateTime(), nullable=True),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("username"),
        sa.UniqueConstraint("uuid"),
    )

    conn.execute("""
        INSERT INTO journalists
        SELECT id, uuid, username, first_name, last_name, pw_salt, pw_hash,
               passphrase_hash, is_admin, session_nonce, otp_secret, is_totp,
               hotp_counter, last_token, created_on, last_access
        FROM journalists_tmp
    """)

    # Now delete the old table.
    op.drop_table("journalists_tmp")
Ejemplo n.º 14
0
def create_api_data_table(engine):
    metadata = db.MetaData()
    api_data = db.Table('api_data', metadata,
        db.Column('id',db.Integer(), primary_key=True,nullable=False),
        db.Column('year',db.Integer()),
        db.Column('country',db.String(255)),
        db.Column('genre',db.Binary()),
        db.Column('style', db.Binary()),
        db.Column('label', db.Binary()),
        db.Column('community_have',db.Integer()),
        db.Column('community_want',db.Integer()),
        db.Column('formats',db.Binary())
    )
def upgrade():
    # Save existing journalist table.
    op.rename_table("journalists", "journalists_tmp")

    # Add UUID column.
    op.add_column("journalists_tmp", sa.Column("uuid", sa.String(length=36)))

    # Add UUIDs to journalists_tmp table.
    conn = op.get_bind()
    journalists = conn.execute(
        sa.text("SELECT * FROM journalists_tmp")).fetchall()

    for journalist in journalists:
        conn.execute(
            sa.text("""UPDATE journalists_tmp SET uuid=:journalist_uuid WHERE
                       id=:id""").bindparams(journalist_uuid=str(uuid.uuid4()),
                                             id=journalist.id))

    # Now create new table with unique constraint applied.
    op.create_table(
        "journalists",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("uuid", sa.String(length=36), nullable=False),
        sa.Column("username", sa.String(length=255), nullable=False),
        sa.Column("pw_salt", sa.Binary(), nullable=True),
        sa.Column("pw_hash", sa.Binary(), nullable=True),
        sa.Column("passphrase_hash", sa.String(length=256), nullable=True),
        sa.Column("is_admin", sa.Boolean(), nullable=True),
        sa.Column("otp_secret", sa.String(length=16), nullable=True),
        sa.Column("is_totp", sa.Boolean(), nullable=True),
        sa.Column("hotp_counter", sa.Integer(), nullable=True),
        sa.Column("last_token", sa.String(length=6), nullable=True),
        sa.Column("created_on", sa.DateTime(), nullable=True),
        sa.Column("last_access", sa.DateTime(), nullable=True),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("username"),
        sa.UniqueConstraint("uuid"),
    )

    conn = op.get_bind()
    conn.execute("""
        INSERT INTO journalists
        SELECT id, uuid, username, pw_salt, pw_hash, passphrase_hash,
               is_admin, otp_secret, is_totp, hotp_counter, last_token,
               created_on, last_access
        FROM journalists_tmp
    """)

    # Now delete the old table.
    op.drop_table("journalists_tmp")
def upgrade():
    # Save existing journalist table.
    op.rename_table('journalists', 'journalists_tmp')

    # Add nonce column.
    op.add_column('journalists_tmp', sa.Column('session_nonce', sa.Integer()))

    # Populate nonce column.
    conn = op.get_bind()
    journalists = conn.execute(
        sa.text("SELECT * FROM journalists_tmp")).fetchall()

    for journalist in journalists:
        conn.execute(
            sa.text("""UPDATE journalists_tmp SET session_nonce=0 WHERE
                       id=:id""").bindparams(id=journalist.id))

    # Now create new table with null constraint applied.
    op.create_table(
        'journalists', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('uuid', sa.String(length=36), nullable=False),
        sa.Column('username', sa.String(length=255), nullable=False),
        sa.Column('first_name', sa.String(length=255), nullable=True),
        sa.Column('last_name', sa.String(length=255), nullable=True),
        sa.Column('pw_salt', sa.Binary(), nullable=True),
        sa.Column('pw_hash', sa.Binary(), nullable=True),
        sa.Column('passphrase_hash', sa.String(length=256), nullable=True),
        sa.Column('is_admin', sa.Boolean(), nullable=True),
        sa.Column('session_nonce', sa.Integer(), nullable=False),
        sa.Column('otp_secret', sa.String(length=16), nullable=True),
        sa.Column('is_totp', sa.Boolean(), nullable=True),
        sa.Column('hotp_counter', sa.Integer(), nullable=True),
        sa.Column('last_token', sa.String(length=6), nullable=True),
        sa.Column('created_on', sa.DateTime(), nullable=True),
        sa.Column('last_access', sa.DateTime(), nullable=True),
        sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('username'),
        sa.UniqueConstraint('uuid'))

    conn = op.get_bind()
    conn.execute('''
        INSERT INTO journalists
        SELECT id, uuid, username, first_name, last_name, pw_salt, pw_hash,
               passphrase_hash, is_admin, session_nonce, otp_secret, is_totp,
               hotp_counter, last_token, created_on, last_access
        FROM journalists_tmp
    ''')

    # Now delete the old table.
    op.drop_table('journalists_tmp')
Ejemplo n.º 17
0
def downgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('email_notification', 'next_event')
    op.drop_column('email_notification', 'new_paper')
    op.drop_column('email_notification', 'session_accept_reject')
    op.drop_column('email_notification', 'session_schedule')
    op.add_column('email_notification',
                  sa.Column('next_event', sa.Binary(), nullable=True))
    op.add_column('email_notification',
                  sa.Column('new_paper', sa.Binary(), nullable=True))
    op.add_column(
        'email_notification',
        sa.Column('session_accept_reject', sa.Binary(), nullable=True))
    op.add_column('email_notification',
                  sa.Column('session_schedule', sa.Binary(), nullable=True))
Ejemplo n.º 18
0
class PageDocument(Base):
    """
	A data class representing the information crawled from a web page.
	"""
    __tablename__ = "Document"
    doc_id = sa.Column("id",
                       sa.BigInteger().with_variant(sa.Integer, "sqlite"),
                       primary_key=True)
    title = sa.Column("title", sa.String(500), nullable=False)
    url = sa.Column("url", sa.String(500), nullable=False)
    content = sa.Column("content", sa.Binary(2000000), nullable=False)
    checksum = sa.Column("checksum", sa.Binary(128), nullable=False)
    anchors = relationship("Anchor")
    headers = relationship("Header")
    texts = relationship("TextSection")

    def __init__(self,
                 doc_id=0,
                 title="",
                 checksum=b"",
                 url="",
                 content=b"",
                 anchors=(),
                 texts=(),
                 headers=()):
        """
		creates a new PageDocument
		:param doc_id: the id of the document in the main repository.
		:param title: the title of the page.
		:param checksum: the checksum of the page.
		:param url: the url of the page.
		:param html: the html of the page.
		:param anchors: the anchors on the page.
		:param texts: the text sectins on the page.
		:param headers: the headings on the page.
		"""

        self.doc_id = doc_id
        self.title = title
        self.checksum = checksum
        self.url = url
        self.content = content
        self.anchors.extend(anchors)
        self.texts.extend(texts)
        self.headers.extend(headers)

    def __repr__(self) -> str:
        return str(self.__dict__)
Ejemplo n.º 19
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'PodCast', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('artistName', sa.String(), nullable=True),
        sa.Column('podCastId', sa.Integer(), nullable=True),
        sa.Column('releaseDate', sa.Date(), nullable=True),
        sa.Column('name', sa.String(), nullable=True),
        sa.Column('kind', sa.String(), nullable=True),
        sa.Column('copyright', sa.String(), nullable=True),
        sa.Column('artistId', sa.Integer(), nullable=True),
        sa.Column('contentAdvisoryRating', sa.String(), nullable=True),
        sa.Column('artistUrl', sa.String(), nullable=True),
        sa.Column('artworkUrl100', sa.String(), nullable=True),
        sa.PrimaryKeyConstraint('id'))
    op.create_table('User', sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('username', sa.String(), nullable=True),
                    sa.Column('email', sa.String(), nullable=True),
                    sa.Column('password', sa.Binary(), nullable=True),
                    sa.Column('access', sa.String(), nullable=True),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('email'),
                    sa.UniqueConstraint('username'))
    op.create_table('Genre', sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('genreId', sa.Integer(), nullable=True),
                    sa.Column('name', sa.String(), nullable=True),
                    sa.Column('url', sa.String(), nullable=True),
                    sa.Column('podCast', sa.Integer(), nullable=True),
                    sa.ForeignKeyConstraint(
                        ['podCast'],
                        ['PodCast.id'],
                    ), sa.PrimaryKeyConstraint('id'))
Ejemplo n.º 20
0
class SignoffStep(db.Model):
    """
    Recording signoff steps as distinct entities
    """

    __tablename__ = 'shipit_signoff_steps'

    uid = sa.Column(sa.String(80), primary_key=True)
    # uid = sa.Column(sa.String(80), nullable=False)
    state = sa.Column(sa.Enum(SigningStatus), default=SigningStatus.starting)
    status_message = sa.Column(sa.String(200), nullable=True)
    created = sa.Column(sa.DateTime, default=datetime.datetime.utcnow)
    completed = sa.Column(sa.DateTime, nullable=True)
    policy = sa.Column(sa.Binary())
    signatures = db.relationship('Signature')

    @property
    def policy_data(self):
        if not self.policy:
            return None
        return pickle.loads(self.policy)

    def delete(self):
        """
        Delete step, and associated signatures
        """
        signatures = Signature.query.filter_by(step_uid=self.uid).all()

        for signature in signatures:
            signature.delete()

        db.session.delete(self)
        db.session.commit()
Ejemplo n.º 21
0
def upgrade():
    op.alter_column('payment_transaction',
                    'checksum',
                    existing_type=sa.String(40),
                    type_=sa.Binary(32),
                    existing_nullable=True,
                    postgresql_using='checksum::bytea')
Ejemplo n.º 22
0
def upgrade(pyramid_env):
    with context.begin_transaction():
        op.create_table(
            'old_password',
            sa.Column(
                'id',
                sa.Integer,
                primary_key=True
            ),
            sa.Column(
                'user_id',
                sa.Integer,
                sa.ForeignKey(
                    'user.id',
                    ondelete="CASCADE",
                    onupdate="CASCADE"
                ),
                nullable=False,
                index=True
            ),
            sa.Column(
                'password',
                sa.Binary(115),
            ),
        )
Ejemplo n.º 23
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'action', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('event_name', sa.String(length=50), nullable=False),
        sa.Column('timestamp', sa.DateTime(), nullable=True),
        sa.Column('_data', sa.Text(), nullable=True),
        sa.Column('message', sa.String(length=500), nullable=True),
        sa.Column('status', sa.Boolean(), nullable=False),
        sa.PrimaryKeyConstraint('id'))
    op.create_table(
        'users', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('username', sa.String(length=80), nullable=False),
        sa.Column('email', sa.String(length=80), nullable=False),
        sa.Column('password', sa.Binary(), nullable=True),
        sa.Column('created_at', sa.DateTime(), nullable=False),
        sa.Column('first_name', sa.String(length=30), nullable=True),
        sa.Column('last_name', sa.String(length=30), nullable=True),
        sa.Column('active', sa.Boolean(), nullable=True),
        sa.Column('is_admin', sa.Boolean(), nullable=True),
        sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('email'),
        sa.UniqueConstraint('username'))
    op.create_table('roles', sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('name', sa.String(length=80), nullable=False),
                    sa.Column('user_id', sa.Integer(), nullable=True),
                    sa.ForeignKeyConstraint(
                        ['user_id'],
                        ['users.id'],
                    ), sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('name'))
Ejemplo n.º 24
0
def upgrade():
    op.create_table(
        'users',
        sa.Column('username', sa.String(length=24), nullable=False),
        sa.Column('email', sa.String(length=255), nullable=False),
        sa.Column('password', sa.Binary(), nullable=False),
    )
Ejemplo n.º 25
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'users', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('username', sa.String(length=40), nullable=False),
        sa.Column('email', sa.String(length=255), nullable=False),
        sa.Column('admin', sa.Boolean(), nullable=True),
        sa.Column('hashed_password', sa.Binary(), nullable=False),
        sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('email'),
        sa.UniqueConstraint('username'))
    op.create_table(
        'missings', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('fullName', sa.String(length=100), nullable=False),
        sa.Column('age', sa.Integer(), nullable=False),
        sa.Column('tribalAffiliation', sa.String(length=20), nullable=False),
        sa.Column('location', sa.String(length=50), nullable=False),
        sa.Column('lat', sa.Float(), nullable=False),
        sa.Column('lng', sa.Float(), nullable=False),
        sa.Column('dateLastSeen', sa.DateTime(), nullable=False),
        sa.Column('details', sa.String(length=1000), nullable=True),
        sa.Column('status', sa.String(length=10), nullable=False),
        sa.Column('userId', sa.Integer(), nullable=False),
        sa.ForeignKeyConstraint(
            ['userId'],
            ['users.id'],
        ), sa.PrimaryKeyConstraint('id'))
Ejemplo n.º 26
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'users', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('email', sa.String(), nullable=False),
        sa.Column('_password', sa.Binary(), nullable=False),
        sa.Column('authenticated', sa.Boolean(), nullable=True),
        sa.Column('email_confirmation_sent_on', sa.DateTime(), nullable=True),
        sa.Column('email_confirmed', sa.Boolean(), nullable=True),
        sa.Column('email_confirmed_on', sa.DateTime(), nullable=True),
        sa.Column('registered_on', sa.DateTime(), nullable=True),
        sa.Column('last_logged_in', sa.DateTime(), nullable=True),
        sa.Column('current_logged_in', sa.DateTime(), nullable=True),
        sa.Column('role', sa.String(), nullable=True),
        sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('email'))
    op.create_table('books', sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('title', sa.String(), nullable=False),
                    sa.Column('author', sa.String(), nullable=True),
                    sa.Column('read_date', sa.String(), nullable=True),
                    sa.Column('notes', sa.String(), nullable=True),
                    sa.Column('user_id', sa.Integer(), nullable=True),
                    sa.ForeignKeyConstraint(
                        ['user_id'],
                        ['users.id'],
                    ), sa.PrimaryKeyConstraint('id'))
Ejemplo n.º 27
0
def upgrade():
    op.create_table(
        'source_components', sa.Column('name', sa.String(), nullable=False),
        sa.Column('value', sa.Binary(), nullable=False),
        sa.Column('description', sa.Text(), nullable=True),
        sa.Column('username', sa.String(), nullable=True),
        sa.PrimaryKeyConstraint('name', name=op.f('pk_source_components')))
Ejemplo n.º 28
0
class Job(Base):
    __tablename__ = "jobs"

    id = sa.Column(sa.Integer, primary_key=True)
    created = sa.Column(sa.DateTime, nullable=False, default=datetime.utcnow)
    updated = sa.Column(sa.DateTime, nullable=False, default=datetime.utcnow)
    manifest = sa.Column(sa.Binary(), nullable=False)
    owner_id = sa.Column(UUID,
                         sa.ForeignKey("users.id", ondelete="cascade"),
                         nullable=True)
    owner = sa.orm.relationship("User",
                                backref=sa.orm.backref("jobs",
                                                       cascade="all,delete"))
    items_spec = sa.Column(sa.String(), nullable=True)
    items = sa.Column(
        sa.Enum(ItemsSelection),
        nullable=False,
        default=ItemsSelection.all,
    )
    # job_group_id = sa.Column(sa.Integer, sa.ForeignKey('job_group.id'))
    # job_group = sa.orm.relationship('JobGroup', backref=sa.orm.backref('jobs'))
    # secrets = sa.Column(sa.Boolean, nullable=False, server_default="t")
    # note = sa.Column(sa.Unicode(4096))
    # tags = sa.Column(sa.String())
    # runner = sa.Column(sa.String)
    status = sa.Column(
        sa.Enum(JobStatus),
        nullable=False,
        default=JobStatus.pending,
    )

    tasks = sa.orm.relationship('Task', cascade="all,delete-orphan")
Ejemplo n.º 29
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'Patient', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('username', sa.String(), nullable=True),
        sa.Column('age', sa.Integer(), nullable=True),
        sa.Column('visits', sa.Integer(), nullable=True),
        sa.PrimaryKeyConstraint('id', name=op.f('pk_Patient')),
        sa.UniqueConstraint('username', name=op.f('uq_Patient_username')))
    op.create_table(
        'PatientVisitDetails',
        sa.Column('visit_id', sa.Integer(), nullable=False),
        sa.Column('patient_id', sa.Integer(), nullable=True),
        sa.Column('date_time', sa.DateTime(), nullable=True),
        sa.Column('doctor_id', sa.Integer(), nullable=True),
        sa.Column('question_1', sa.String(), nullable=False),
        sa.Column('question_2', sa.String(), nullable=False),
        sa.Column('question_3', sa.String(), nullable=False),
        sa.Column('question_4', sa.String(), nullable=False),
        sa.Column('question_5', sa.String(), nullable=False),
        sa.Column('question_6', sa.String(), nullable=False),
        sa.PrimaryKeyConstraint('visit_id',
                                name=op.f('pk_PatientVisitDetails')))
    op.create_table(
        'User', sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('username', sa.String(), nullable=True),
        sa.Column('email', sa.String(), nullable=True),
        sa.Column('password', sa.Binary(), nullable=True),
        sa.PrimaryKeyConstraint('id', name=op.f('pk_User')),
        sa.UniqueConstraint('email', name=op.f('uq_User_email')),
        sa.UniqueConstraint('username', name=op.f('uq_User_username')))
Ejemplo n.º 30
0
def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('comic',
                    sa.Column('id', sa.String(length=30), nullable=False),
                    sa.Column('month', sa.String(length=2), nullable=False),
                    sa.Column('num', sa.Integer(), nullable=False),
                    sa.Column('link', sa.String(length=100), nullable=False),
                    sa.Column('year', sa.String(length=4), nullable=False),
                    sa.Column('news', sa.Text(), nullable=False),
                    sa.Column('safe_title', sa.Text(), nullable=False),
                    sa.Column('transcript', sa.Text(), nullable=False),
                    sa.Column('alt', sa.Text(), nullable=False),
                    sa.Column('img', sa.String(length=200), nullable=False),
                    sa.Column('title', sa.String(length=100), nullable=False),
                    sa.Column('day', sa.String(length=2), nullable=False),
                    sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('num'))
    op.create_table(
        'users', sa.Column('id', sa.String(length=30), nullable=False),
        sa.Column('first_name', sa.String(length=80), nullable=False),
        sa.Column('last_name', sa.String(length=80), nullable=False),
        sa.Column('email', sa.String(length=100), nullable=False),
        sa.Column('password', sa.Binary(), nullable=False),
        sa.PrimaryKeyConstraint('id'), sa.UniqueConstraint('email'),
        sa.UniqueConstraint('id'))
    op.create_table('userprofile', sa.Column('id',
                                             sa.Integer(),
                                             nullable=False),
                    sa.Column('user_id', sa.String(length=30), nullable=False),
                    sa.ForeignKeyConstraint(
                        ['user_id'],
                        ['users.id'],
                    ), sa.PrimaryKeyConstraint('id'))