moviewyrm/bookwyrm/management/commands/initdb.py

139 lines
4.3 KiB
Python
Raw Normal View History

2021-04-30 17:42:27 +00:00
""" What you need in the database to make it work """
from django.core.management.base import BaseCommand
2020-11-08 18:29:33 +00:00
from django.contrib.auth.models import Group, Permission
from django.contrib.contenttypes.models import ContentType
2021-04-05 22:26:53 +00:00
from bookwyrm.models import Connector, FederatedServer, SiteSettings, User
2020-11-08 18:29:33 +00:00
2021-03-08 16:49:10 +00:00
2020-11-08 18:29:33 +00:00
def init_groups():
2021-04-30 17:42:27 +00:00
"""permission levels"""
2021-03-08 16:49:10 +00:00
groups = ["admin", "moderator", "editor"]
2020-11-08 18:29:33 +00:00
for group in groups:
Group.objects.create(name=group)
2021-03-08 16:49:10 +00:00
2020-11-08 18:29:33 +00:00
def init_permissions():
"""permission types"""
2021-03-08 16:49:10 +00:00
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"],
},
]
2020-11-08 18:29:33 +00:00
content_type = ContentType.objects.get_for_model(User)
for permission in permissions:
permission_obj = Permission.objects.create(
2021-03-08 16:49:10 +00:00
codename=permission["codename"],
name=permission["name"],
2020-11-08 18:29:33 +00:00
content_type=content_type,
)
# add the permission to the appropriate groups
2021-03-08 16:49:10 +00:00
for group_name in permission["groups"]:
2020-11-08 18:29:33 +00:00
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():
2021-04-30 17:42:27 +00:00
"""access book data sources"""
2020-11-08 18:29:33 +00:00
Connector.objects.create(
2021-03-08 16:49:10 +00:00
identifier="bookwyrm.social",
name="BookWyrm dot Social",
connector_file="bookwyrm_connector",
base_url="https://bookwyrm.social",
books_url="https://bookwyrm.social/book",
2021-03-13 21:55:20 +00:00
covers_url="https://bookwyrm.social/images/",
2021-03-08 16:49:10 +00:00
search_url="https://bookwyrm.social/search?q=",
isbn_search_url="https://bookwyrm.social/isbn/",
2020-11-08 18:29:33 +00:00
priority=2,
)
Connector.objects.create(
identifier="inventaire.io",
name="Inventaire",
connector_file="inventaire",
2021-04-29 17:30:38 +00:00
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,
)
2020-11-08 18:29:33 +00:00
Connector.objects.create(
2021-03-08 16:49:10 +00:00
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:",
2020-11-08 18:29:33 +00:00
priority=3,
)
2021-03-08 16:49:10 +00:00
2021-04-05 22:26:53 +00:00
def init_federated_servers():
2021-04-26 16:15:42 +00:00
"""big no to nazis"""
2021-04-05 22:26:53 +00:00
built_in_blocks = ["gab.ai", "gab.com"]
for server in built_in_blocks:
FederatedServer.objects.create(
server_name=server,
status="blocked",
)
2020-12-13 00:01:43 +00:00
def init_settings():
2021-04-30 18:09:06 +00:00
"""info about the instance"""
2021-04-30 17:42:27 +00:00
SiteSettings.objects.create(
support_link="https://www.patreon.com/bookwyrm",
support_title="Patreon",
)
2020-12-13 00:01:43 +00:00
2021-03-08 16:49:10 +00:00
2020-11-08 18:29:33 +00:00
class Command(BaseCommand):
2021-03-08 16:49:10 +00:00
help = "Initializes the database with starter data"
2020-11-08 18:29:33 +00:00
def handle(self, *args, **options):
init_groups()
init_permissions()
2020-12-13 00:01:43 +00:00
init_connectors()
2021-04-05 22:26:53 +00:00
init_federated_servers()
2020-12-13 00:01:43 +00:00
init_settings()