From 197812302ea4bbda97bbf1184aa4b8053257ff67 Mon Sep 17 00:00:00 2001 From: Ryoma Kawajiri Date: Thu, 19 Oct 2023 14:28:01 +0900 Subject: [PATCH] fix: drop table sql --- google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py b/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py index 9fe09140..ea3e00a6 100644 --- a/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py +++ b/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py @@ -406,7 +406,12 @@ def visit_drop_table(self, drop_table, **kw): for index in drop_table.element.indexes: indexes += "DROP INDEX {};".format(self.preparer.quote(index.name)) - return indexes + constrs + str(drop_table) + text = "\nDROP TABLE " + if drop_table.if_exists: + text += "IF EXISTS " + text += self.preparer.format_table(drop_table.element) + + return indexes + constrs + text def visit_primary_key_constraint(self, constraint, **kw): """Build primary key definition.