fix: remove unnecessary runtime dependency on psycopg2

radusuciu · Feb 28, 2024 · de86b16 · de86b16
1 parent 21cbaf1
commit de86b16
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sqlfun/management/commands/makemigrations.py b/sqlfun/management/commands/makemigrations.py
@@ -1,6 +1,5 @@
 
 import django
-import psycopg2
 from django.core.management.commands.makemigrations import Command as BaseCommand
 
 from sqlfun.utils import make_sqlfun_migrations
@@ -15,7 +14,7 @@ def handle(self, *args, **options):
                 stdout=self.stdout,
                 is_dry_run=options.get('dry_run')
             )
-        except (psycopg2.errors.UndefinedTable, django.db.utils.ProgrammingError):
+        except django.db.utils.ProgrammingError:
             self.stderr.write(
                 '[sqlfun] It seems like the SqlFunDefinition model does not exist yet. '
                 'A migration will be generated which must be applied before you can '