2008-02-06 18:39:07 +00:00
|
|
|
from migrate.changeset import ansisql
|
|
|
|
from sqlalchemy.databases import postgres as sa_base
|
|
|
|
#import sqlalchemy as sa
|
|
|
|
|
|
|
|
PGSchemaGenerator = sa_base.PGSchemaGenerator
|
|
|
|
|
2008-12-02 15:25:12 +00:00
|
|
|
class PGSchemaGeneratorMixin(object):
|
2008-11-26 22:05:19 +00:00
|
|
|
def _do_quote_table_identifier(self, identifier):
|
|
|
|
return identifier
|
|
|
|
def _do_quote_column_identifier(self, identifier):
|
|
|
|
return '"%s"'%identifier
|
|
|
|
|
2008-12-02 15:25:12 +00:00
|
|
|
class PGColumnGenerator(PGSchemaGenerator,ansisql.ANSIColumnGenerator, PGSchemaGeneratorMixin):
|
|
|
|
pass
|
2008-11-26 22:05:19 +00:00
|
|
|
|
2008-12-02 15:25:12 +00:00
|
|
|
class PGColumnDropper(ansisql.ANSIColumnDropper, PGSchemaGeneratorMixin):
|
|
|
|
pass
|
2008-11-26 22:05:19 +00:00
|
|
|
|
2008-12-02 15:25:12 +00:00
|
|
|
class PGSchemaChanger(ansisql.ANSISchemaChanger, PGSchemaGeneratorMixin):
|
|
|
|
pass
|
|
|
|
|
|
|
|
class PGConstraintGenerator(ansisql.ANSIConstraintGenerator, PGSchemaGeneratorMixin):
|
|
|
|
pass
|
|
|
|
class PGConstraintDropper(ansisql.ANSIConstraintDropper, PGSchemaGeneratorMixin):
|
2008-02-06 18:39:07 +00:00
|
|
|
pass
|
|
|
|
|
|
|
|
class PGDialect(ansisql.ANSIDialect):
|
|
|
|
columngenerator = PGColumnGenerator
|
|
|
|
columndropper = PGColumnDropper
|
|
|
|
schemachanger = PGSchemaChanger
|
|
|
|
constraintgenerator = PGConstraintGenerator
|
|
|
|
constraintdropper = PGConstraintDropper
|