34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from bs4 import BeautifulSoup
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from ...models import Page
|
|
from pathlib import Path
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "add missing cms files to database"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"--force",
|
|
action="store_true",
|
|
help="Always overwrite content",
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
content_path = Path(__file__).resolve().parent.parent.parent / "default_content"
|
|
for file in content_path.iterdir():
|
|
slug = file.stem
|
|
p, created = Page.objects.get_or_create(url=slug)
|
|
if (not created) and (not options["force"]):
|
|
continue
|
|
|
|
soup = BeautifulSoup(file.read_text(), "html.parser")
|
|
if soup.title:
|
|
p.title = soup.title.string
|
|
soup.title.decompose()
|
|
else:
|
|
p.title = slug.title()
|
|
p.content = str(soup).strip()
|
|
p.save()
|
|
print(f'created new page "{p.title}" for slug {slug}')
|