Handle drop constraint (doableware#211)

pahpa · Nov 18, 2018 · e40f1df · e40f1df
1 parent 4b485d7
commit e40f1df
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/djongo/sql2mongo/query.py b/djongo/sql2mongo/query.py
@@ -555,19 +555,25 @@ def _drop(self, tok_id):
 
         while tok_id is not None:
             if tok.match(tokens.Keyword, (
-                    'COLUMN', 'CASCADE'
+                'CASCADE'
             )):
                 pass
             elif isinstance(tok, Identifier):
                 self._iden_name = tok.get_name()
+            elif tok.match(tokens.Keyword, 'CONSTRAINT'):
+                self.execute = self._drop_constraint
+            elif tok.match(tokens.Keyword, 'COLUMN'):
+                self.execute = self._drop_column
             else:
                 raise NotImplementedError
 
             tok_id, tok = sm.token_next(tok_id)
 
-        self.execute = self._drop_column
         return tok_id
 
+    def _drop_constraint(self):
+        self.db_ref[self.left_table].drop_index(self._iden_name)
+
     def _drop_column(self):
         self.db_ref[self.left_table].update(
             {},