mirror of
https://github.com/Findus23/RPGnotes.git
synced 2024-09-20 16:53:44 +02:00
44 lines
1.8 KiB
Python
44 lines
1.8 KiB
Python
from django.core.management.base import BaseCommand
|
|
from django.db import transaction
|
|
from django_tenants.utils import tenant_context
|
|
|
|
from campaigns.models import Campaign
|
|
from characters.models import Character
|
|
from days.models import IngameDay
|
|
from factions.models import Faction
|
|
from locations.models import Location
|
|
from loot.models import Loot
|
|
from notes.models import Note
|
|
from utils.diff import print_diff_call
|
|
from utils.markdown import md_to_html
|
|
from utils.urls import name2url
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'regenerate HTML from markdown (including autolinks)'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('--store', action='store_true')
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(Command, self).__init__(*args, **kwargs)
|
|
|
|
def handle(self, *args, **options):
|
|
store = options["store"]
|
|
for campaign in Campaign.objects.exclude(pk=1):
|
|
with tenant_context(campaign):
|
|
with transaction.atomic():
|
|
replacements = name2url()
|
|
objects = list(Character.objects.all())
|
|
objects.extend(list(Location.objects.all()))
|
|
objects.extend(list(Loot.objects.all()))
|
|
objects.extend(list(Faction.objects.all()))
|
|
objects.extend(list(Note.objects.all()))
|
|
objects.extend(list(IngameDay.objects.all()))
|
|
for object in objects:
|
|
fresh_html = md_to_html(object.description_md, replacements=replacements)
|
|
if object.description_html != fresh_html:
|
|
print_diff_call(object.description_html, fresh_html, str(object))
|
|
if store:
|
|
object.description_html = fresh_html
|
|
object.save()
|