fix: drop table sql

googleapis · Oct 19, 2023 · b47363d · b47363d
1 parent d2222d5
commit b47363d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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.