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, FederatedServer, 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_federated_servers(): """ big no to nazis """ built_in_blocks = ["gab.ai", "gab.com"] for server in built_in_blocks: FederatedServer.objects.create( server_name=server, status="blocked", ) 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_federated_servers() init_settings()