Merge pull request #1527 from bookwyrm-social/book-file-links

Book file links

bookwyrm/management/commands/initdb.py

@@ -3,7 +3,7 @@ from django.core.management.base import BaseCommand
 from django.contrib.auth.models import Group, Permission
 from django.contrib.contenttypes.models import ContentType
 
-from bookwyrm.models import Connector, FederatedServer, SiteSettings, User
+from bookwyrm import models
 
 
 def init_groups():
@@ -55,7 +55,7 @@ def init_permissions():
         },
     ]
 
-    content_type = ContentType.objects.get_for_model(User)
+    content_type = models.ContentType.objects.get_for_model(User)
     for permission in permissions:
         permission_obj = Permission.objects.create(
             codename=permission["codename"],
@@ -72,7 +72,7 @@ def init_permissions():
 
 def init_connectors():
     """access book data sources"""
-    Connector.objects.create(
+    models.Connector.objects.create(
         identifier="bookwyrm.social",
         name="BookWyrm dot Social",
         connector_file="bookwyrm_connector",
@@ -84,7 +84,7 @@ def init_connectors():
         priority=2,
     )
 
-    Connector.objects.create(
+    models.Connector.objects.create(
         identifier="inventaire.io",
         name="Inventaire",
         connector_file="inventaire",
@@ -96,7 +96,7 @@ def init_connectors():
         priority=3,
     )
 
-    Connector.objects.create(
+    models.Connector.objects.create(
         identifier="openlibrary.org",
         name="OpenLibrary",
         connector_file="openlibrary",
@@ -113,7 +113,7 @@ def init_federated_servers():
     """big no to nazis"""
     built_in_blocks = ["gab.ai", "gab.com"]
     for server in built_in_blocks:
-        FederatedServer.objects.create(
+        models.FederatedServer.objects.create(
             server_name=server,
             status="blocked",
         )
@@ -121,18 +121,61 @@ def init_federated_servers():
 
 def init_settings():
     """info about the instance"""
-    SiteSettings.objects.create(
+    models.SiteSettings.objects.create(
         support_link="https://www.patreon.com/bookwyrm",
         support_title="Patreon",
     )
 
 
+def init_link_domains(*_):
+    """safe book links"""
+    domains = [
+        ("standardebooks.org", "Standard EBooks"),
+        ("www.gutenberg.org", "Project Gutenberg"),
+        ("archive.org", "Internet Archive"),
+        ("openlibrary.org", "Open Library"),
+        ("theanarchistlibrary.org", "The Anarchist Library"),
+    ]
+    for domain, name in domains:
+        models.LinkDomain.objects.create(
+            domain=domain,
+            name=name,
+            status="approved",
+        )
+
+
 class Command(BaseCommand):
     help = "Initializes the database with starter data"
 
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--limit",
+            default=None,
+            help="Limit init to specific table",
+        )
+
     def handle(self, *args, **options):
-        init_groups()
-        init_permissions()
-        init_connectors()
-        init_federated_servers()
-        init_settings()
+        limit = options.get("limit")
+        tables = [
+            "group",
+            "permission",
+            "connector",
+            "federatedserver",
+            "settings",
+            "linkdomain",
+        ]
+        if limit not in tables:
+            raise Exception("Invalid table limit:", limit)
+
+        if not limit or limit == "group":
+            init_groups()
+        if not limit or limit == "permission":
+            init_permissions()
+        if not limit or limit == "connector":
+            init_connectors()
+        if not limit or limit == "federatedserver":
+            init_federated_servers()
+        if not limit or limit == "settings":
+            init_settings()
+        if not limit or limit == "linkdomain":
+            init_link_domains()

bookwyrm/management/commands/populate_lists_streams.py

@@ -22,13 +22,6 @@ class Command(BaseCommand):
 
     help = "Populate list streams for all users"
 
-    def add_arguments(self, parser):
-        parser.add_argument(
-            "--stream",
-            default=None,
-            help="Specifies which time of stream to populate",
-        )
-
     # pylint: disable=no-self-use,unused-argument
     def handle(self, *args, **options):
         """run feed builder"""