Bladeren bron

fix archivebox setup User model import

Nick Sweeting 1 jaar geleden
bovenliggende
commit
7a734ce410
1 gewijzigde bestanden met toevoegingen van 2 en 1 verwijderingen
  1. 2 1
      archivebox/main.py

+ 2 - 1
archivebox/main.py

@@ -960,7 +960,8 @@ def setup(out_dir: Path=OUTPUT_DIR) -> None:
         run_subcommand('init', stdin=None, pwd=out_dir)
         run_subcommand('init', stdin=None, pwd=out_dir)
 
 
     setup_django(out_dir=out_dir, check_db=True)
     setup_django(out_dir=out_dir, check_db=True)
-    from core.models import User
+    from django.contrib.auth import get_user_model
+    User = get_user_model()
 
 
     if not User.objects.filter(is_superuser=True).exists():
     if not User.objects.filter(is_superuser=True).exists():
         stderr('\n[+] Creating new admin user for the Web UI...', color='green')
         stderr('\n[+] Creating new admin user for the Web UI...', color='green')