You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
3.7 KiB
Python
124 lines
3.7 KiB
Python
from sqlalchemy.ext.compiler import compiles
|
|
from sqlalchemy.sql import sqltypes
|
|
|
|
from .base import AddColumn
|
|
from .base import alter_table
|
|
from .base import ColumnComment
|
|
from .base import ColumnDefault
|
|
from .base import ColumnName
|
|
from .base import ColumnNullable
|
|
from .base import ColumnType
|
|
from .base import format_column_name
|
|
from .base import format_server_default
|
|
from .base import format_table_name
|
|
from .base import format_type
|
|
from .base import RenameTable
|
|
from .impl import DefaultImpl
|
|
|
|
|
|
class OracleImpl(DefaultImpl):
|
|
__dialect__ = "oracle"
|
|
transactional_ddl = False
|
|
batch_separator = "/"
|
|
command_terminator = ""
|
|
type_synonyms = DefaultImpl.type_synonyms + (
|
|
{"VARCHAR", "VARCHAR2"},
|
|
{"BIGINT", "INTEGER", "SMALLINT", "DECIMAL", "NUMERIC", "NUMBER"},
|
|
)
|
|
|
|
def __init__(self, *arg, **kw):
|
|
super(OracleImpl, self).__init__(*arg, **kw)
|
|
self.batch_separator = self.context_opts.get(
|
|
"oracle_batch_separator", self.batch_separator
|
|
)
|
|
|
|
def _exec(self, construct, *args, **kw):
|
|
result = super(OracleImpl, self)._exec(construct, *args, **kw)
|
|
if self.as_sql and self.batch_separator:
|
|
self.static_output(self.batch_separator)
|
|
return result
|
|
|
|
def emit_begin(self):
|
|
self._exec("SET TRANSACTION READ WRITE")
|
|
|
|
def emit_commit(self):
|
|
self._exec("COMMIT")
|
|
|
|
|
|
@compiles(AddColumn, "oracle")
|
|
def visit_add_column(element, compiler, **kw):
|
|
return "%s %s" % (
|
|
alter_table(compiler, element.table_name, element.schema),
|
|
add_column(compiler, element.column, **kw),
|
|
)
|
|
|
|
|
|
@compiles(ColumnNullable, "oracle")
|
|
def visit_column_nullable(element, compiler, **kw):
|
|
return "%s %s %s" % (
|
|
alter_table(compiler, element.table_name, element.schema),
|
|
alter_column(compiler, element.column_name),
|
|
"NULL" if element.nullable else "NOT NULL",
|
|
)
|
|
|
|
|
|
@compiles(ColumnType, "oracle")
|
|
def visit_column_type(element, compiler, **kw):
|
|
return "%s %s %s" % (
|
|
alter_table(compiler, element.table_name, element.schema),
|
|
alter_column(compiler, element.column_name),
|
|
"%s" % format_type(compiler, element.type_),
|
|
)
|
|
|
|
|
|
@compiles(ColumnName, "oracle")
|
|
def visit_column_name(element, compiler, **kw):
|
|
return "%s RENAME COLUMN %s TO %s" % (
|
|
alter_table(compiler, element.table_name, element.schema),
|
|
format_column_name(compiler, element.column_name),
|
|
format_column_name(compiler, element.newname),
|
|
)
|
|
|
|
|
|
@compiles(ColumnDefault, "oracle")
|
|
def visit_column_default(element, compiler, **kw):
|
|
return "%s %s %s" % (
|
|
alter_table(compiler, element.table_name, element.schema),
|
|
alter_column(compiler, element.column_name),
|
|
"DEFAULT %s" % format_server_default(compiler, element.default)
|
|
if element.default is not None
|
|
else "DEFAULT NULL",
|
|
)
|
|
|
|
|
|
@compiles(ColumnComment, "oracle")
|
|
def visit_column_comment(element, compiler, **kw):
|
|
ddl = "COMMENT ON COLUMN {table_name}.{column_name} IS {comment}"
|
|
|
|
comment = compiler.sql_compiler.render_literal_value(
|
|
(element.comment if element.comment is not None else ""),
|
|
sqltypes.String(),
|
|
)
|
|
|
|
return ddl.format(
|
|
table_name=element.table_name,
|
|
column_name=element.column_name,
|
|
comment=comment,
|
|
)
|
|
|
|
|
|
@compiles(RenameTable, "oracle")
|
|
def visit_rename_table(element, compiler, **kw):
|
|
return "%s RENAME TO %s" % (
|
|
alter_table(compiler, element.table_name, element.schema),
|
|
format_table_name(compiler, element.new_table_name, None),
|
|
)
|
|
|
|
|
|
def alter_column(compiler, name):
|
|
return "MODIFY %s" % format_column_name(compiler, name)
|
|
|
|
|
|
def add_column(compiler, column, **kw):
|
|
return "ADD %s" % compiler.get_column_specification(column, **kw)
|