forked from mirrors/bookwyrm
144 lines
4.5 KiB
Python
144 lines
4.5 KiB
Python
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="inventaire.io",
|
|
name="Inventaire",
|
|
connector_file="inventaire",
|
|
base_url="https://inventaire.io",
|
|
books_url="https://inventaire.io/api/entities",
|
|
covers_url="https://inventaire.io",
|
|
search_url="https://inventaire.io/api/search?types=works&types=works&search=",
|
|
isbn_search_url="https://inventaire.io/api/entities?action=by-uris&uris=isbn%3A",
|
|
priority=3,
|
|
)
|
|
|
|
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()
|