def test_custom_op(self):
from alembic.operations import Operations, MigrateOperation
@Operations.register_operation("create_sequence")
class CreateSequenceOp(MigrateOperation):
"""Create a SEQUENCE."""
def __init__(self, sequence_name, **kw):
self.sequence_name = sequence_name
self.kw = kw
@classmethod
def create_sequence(cls, operations, sequence_name, **kw):
"""Issue a "CREATE SEQUENCE" instruction."""
op = CreateSequenceOp(sequence_name, **kw)
return operations.invoke(op)
@Operations.implementation_for(CreateSequenceOp)
def create_sequence(operations, operation):
operations.execute("CREATE SEQUENCE %s" % operation.sequence_name)
context = op_fixture()
op.create_sequence('foob')
context.assert_("CREATE SEQUENCE foob")
def test_custom_op(self):
from alembic.operations import Operations, MigrateOperation
@Operations.register_operation("create_sequence")
class CreateSequenceOp(MigrateOperation):
"""Create a SEQUENCE."""
def __init__(self, sequence_name, **kw):
self.sequence_name = sequence_name
self.kw = kw
@classmethod
def create_sequence(cls, operations, sequence_name, **kw):
"""Issue a "CREATE SEQUENCE" instruction."""
op = CreateSequenceOp(sequence_name, **kw)
return operations.invoke(op)
@Operations.implementation_for(CreateSequenceOp)
def create_sequence(operations, operation):
operations.execute("CREATE SEQUENCE %s" % operation.sequence_name)
context = op_fixture()
op.create_sequence("foob")
context.assert_("CREATE SEQUENCE foob")
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.create_sequence("Employee_seq_autoid")
op.create_sequence("Employee_id_autoid")
op.create_table('Employee',
sa.Column('id', sa.Integer(), server_default=sa.text("nextval('employee_id_autoid')"), nullable=False),
sa.Column('employeeId', sa.VARCHAR(length=25), nullable=False),
sa.Column('firstName', sa.VARCHAR(length=200), nullable=True),
sa.Column('lastName', sa.VARCHAR(length=200), nullable=True),
sa.Column('password', sa.VARCHAR(length=500), nullable=True),
sa.Column('email', sa.VARCHAR(length=200), nullable=False),
sa.Column('phoneNumber', sa.VARCHAR(length=20), nullable=True),
sa.Column('joinedDate', sa.TIMESTAMP(), nullable=True),
sa.Column('employeeType', postgresql.ENUM('EMPLOYEE', 'ADMIN', name='EmployeeTYPE'), server_default='EMPLOYEE', nullable=False),
sa.Column('employeeDesignation', postgresql.ENUM('DEVELOPER', 'SENIOR DEVELOPER', 'MANAGER', name='EmployeeDESIGNATION'), nullable=False),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('email'),
sa.UniqueConstraint('employeeId')
)
op.create_sequence("Holiday_id_autoid")
op.create_table('Holiday',
sa.Column('id', sa.Integer(), server_default=sa.text("nextval('holiday_id_autoid')"), nullable=False),
sa.Column('date', sa.TIMESTAMP(), nullable=True),
sa.Column('festivalName', sa.VARCHAR(length=200), nullable=True),
sa.Column('description', sa.VARCHAR(length=500), nullable=True),
sa.Column('holidayBanner', sa.VARCHAR(length=8000), nullable=True),
sa.PrimaryKeyConstraint('id')
)
op.create_sequence("LeaveManagement_seq_autoid")
op.create_sequence("LeaveManagement_id_autoid")
op.create_table('LeaveManagement',
sa.Column('id', sa.Integer(), server_default=sa.text('nextval(\'"leavemanagement_id_autoid"\')'), nullable=False),
sa.Column('leaveId', sa.VARCHAR(length=25), nullable=False),
sa.Column('employeeId', sa.VARCHAR(length=25), nullable=False),
sa.Column('fromDate', sa.TIMESTAMP(), nullable=True),
sa.Column('toDate', sa.TIMESTAMP(), nullable=True),
sa.Column('reason', sa.VARCHAR(length=500), nullable=False),
sa.Column('leaveType', postgresql.ENUM('SICK', 'MEDICAL', 'CASUAL', name='LeaveTYPE'), nullable=False),
sa.Column('approver', sa.VARCHAR(length=200), nullable=True),
sa.Column('approvedStatus', postgresql.ENUM('PENDING', 'ACCEPTED', 'DENIED', name='ApprovedSTATUS'), nullable=False),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('leaveId')
)
op.create_sp(Employees_trigger)
op.create_sp(Leave_trigger)