mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-11-22 17:41:08 +00:00
Move initdb into a management command
This commit is contained in:
parent
e279ae0d88
commit
8c6c28c7d8
4 changed files with 111 additions and 95 deletions
100
bookwyrm/management/commands/initdb.py
Normal file
100
bookwyrm/management/commands/initdb.py
Normal file
|
@ -0,0 +1,100 @@
|
||||||
|
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, 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/covers' % DOMAIN,
|
||||||
|
search_url='https://%s/search?q=' % 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/covers',
|
||||||
|
search_url='https://bookwyrm.social/search?q=',
|
||||||
|
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=',
|
||||||
|
priority=3,
|
||||||
|
)
|
||||||
|
|
||||||
|
class Command(BaseCommand):
|
||||||
|
help = 'Initializes the database with starter data'
|
||||||
|
|
||||||
|
def handle(self, *args, **options):
|
||||||
|
init_groups()
|
||||||
|
init_permissions()
|
||||||
|
init_connectors()
|
13
fr-dev
13
fr-dev
|
@ -12,7 +12,7 @@ case "$1" in
|
||||||
;;
|
;;
|
||||||
initdb)
|
initdb)
|
||||||
docker-compose exec web python manage.py migrate
|
docker-compose exec web python manage.py migrate
|
||||||
docker-compose exec web python manage.py shell -c 'import init_db'
|
docker-compose exec web python manage.py initdb
|
||||||
;;
|
;;
|
||||||
resetdb)
|
resetdb)
|
||||||
docker-compose stop web
|
docker-compose stop web
|
||||||
|
@ -20,7 +20,7 @@ case "$1" in
|
||||||
docker-compose exec db createdb -U fedireads fedireads
|
docker-compose exec db createdb -U fedireads fedireads
|
||||||
docker-compose start web
|
docker-compose start web
|
||||||
docker-compose exec web python manage.py migrate
|
docker-compose exec web python manage.py migrate
|
||||||
docker-compose exec web python manage.py shell -c 'import init_db'
|
docker-compose exec web python manage.py initdb
|
||||||
;;
|
;;
|
||||||
makemigrations)
|
makemigrations)
|
||||||
docker-compose exec web python manage.py makemigrations
|
docker-compose exec web python manage.py makemigrations
|
||||||
|
@ -47,7 +47,14 @@ case "$1" in
|
||||||
collectstatic)
|
collectstatic)
|
||||||
docker-compose exec web python manage.py collectstatic --no-input
|
docker-compose exec web python manage.py collectstatic --no-input
|
||||||
;;
|
;;
|
||||||
|
build)
|
||||||
|
docker-compose build
|
||||||
|
;;
|
||||||
|
clean)
|
||||||
|
docker-compose stop
|
||||||
|
docker-compose rm -f
|
||||||
|
;;
|
||||||
*)
|
*)
|
||||||
echo "Unrecognised command. Try: up, initdb, resetdb, makemigrations, migrate, shell, dbshell, restart_celery, test, test_report"
|
echo "Unrecognised command. Try: build, clean, up, initdb, resetdb, makemigrations, migrate, shell, dbshell, restart_celery, test, test_report"
|
||||||
;;
|
;;
|
||||||
esac
|
esac
|
||||||
|
|
91
init_db.py
91
init_db.py
|
@ -1,91 +0,0 @@
|
||||||
''' starter data '''
|
|
||||||
from django.contrib.auth.models import Group, Permission
|
|
||||||
from django.contrib.contenttypes.models import ContentType
|
|
||||||
|
|
||||||
from bookwyrm.models import Connector, User
|
|
||||||
from bookwyrm.settings import DOMAIN
|
|
||||||
|
|
||||||
|
|
||||||
groups = ['admin', 'moderator', 'editor']
|
|
||||||
for group in groups:
|
|
||||||
Group.objects.create(name=group)
|
|
||||||
|
|
||||||
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
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
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/covers' % DOMAIN,
|
|
||||||
search_url='https://%s/search?q=' % 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/covers',
|
|
||||||
search_url='https://bookwyrm.social/search?q=',
|
|
||||||
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=',
|
|
||||||
priority=3,
|
|
||||||
)
|
|
|
@ -21,5 +21,5 @@ fi
|
||||||
python manage.py makemigrations fedireads
|
python manage.py makemigrations fedireads
|
||||||
python manage.py migrate
|
python manage.py migrate
|
||||||
|
|
||||||
python manage.py shell < init_db.py
|
python manage.py initdb
|
||||||
python manage.py runserver
|
python manage.py runserver
|
||||||
|
|
Loading…
Reference in a new issue