moviewyrm/bookwyrm/management/commands/initdb.py

from django.core.management.base import BaseCommand, CommandError
from django.contrib.auth.models import Group, Permission
from django.contrib.contenttypes.models import ContentType

from bookwyrm.models import Connector, SiteSettings, User
from bookwyrm.settings import DOMAIN


def init_groups():
    groups = ["admin", "moderator", "editor"]
    for group in groups:
        Group.objects.create(name=group)


def init_permissions():
    permissions = [
        {
            "codename": "edit_instance_settings",
            "name": "change the instance info",
            "groups": [
                "admin",
            ],
        },
        {
            "codename": "set_user_group",
            "name": "change what group a user is in",
            "groups": ["admin", "moderator"],
        },
        {
            "codename": "control_federation",
            "name": "control who to federate with",
            "groups": ["admin", "moderator"],
        },
        {
            "codename": "create_invites",
            "name": "issue invitations to join",
            "groups": ["admin", "moderator"],
        },
        {
            "codename": "moderate_user",
            "name": "deactivate or silence a user",
            "groups": ["admin", "moderator"],
        },
        {
            "codename": "moderate_post",
            "name": "delete other users' posts",
            "groups": ["admin", "moderator"],
        },
        {
            "codename": "edit_book",
            "name": "edit book info",
            "groups": ["admin", "moderator", "editor"],
        },
    ]

    content_type = ContentType.objects.get_for_model(User)
    for permission in permissions:
        permission_obj = Permission.objects.create(
            codename=permission["codename"],
            name=permission["name"],
            content_type=content_type,
        )
        # add the permission to the appropriate groups
        for group_name in permission["groups"]:
            Group.objects.get(name=group_name).permissions.add(permission_obj)

    # while the groups and permissions shouldn't be changed because the code
    # depends on them, what permissions go with what groups should be editable


def init_connectors():
    Connector.objects.create(
        identifier=DOMAIN,
        name="Local",
        local=True,
        connector_file="self_connector",
        base_url="https://%s" % DOMAIN,
        books_url="https://%s/book" % DOMAIN,
        covers_url="https://%s/images/" % DOMAIN,
        search_url="https://%s/search?q=" % DOMAIN,
        isbn_search_url="https://%s/isbn/" % DOMAIN,
        priority=1,
    )

    Connector.objects.create(
        identifier="bookwyrm.social",
        name="BookWyrm dot Social",
        connector_file="bookwyrm_connector",
        base_url="https://bookwyrm.social",
        books_url="https://bookwyrm.social/book",
        covers_url="https://bookwyrm.social/images/",
        search_url="https://bookwyrm.social/search?q=",
        isbn_search_url="https://bookwyrm.social/isbn/",
        priority=2,
    )

    Connector.objects.create(
        identifier="openlibrary.org",
        name="OpenLibrary",
        connector_file="openlibrary",
        base_url="https://openlibrary.org",
        books_url="https://openlibrary.org",
        covers_url="https://covers.openlibrary.org",
        search_url="https://openlibrary.org/search?q=",
        isbn_search_url="https://openlibrary.org/api/books?jscmd=data&format=json&bibkeys=ISBN:",
        priority=3,
    )


def init_settings():
    SiteSettings.objects.create()


class Command(BaseCommand):
    help = "Initializes the database with starter data"

    def handle(self, *args, **options):
        init_groups()
        init_permissions()
        init_connectors()
        init_settings()