deploy-with-fallback-language.py

import argparse
import os
import os.path
import platform
from shutil import copy2


def has_translation(filepath, language_code):
    translated_path = filepath.replace('.md', f'.{language_code}.md')
    return os.path.exists(translated_path)


def add_missing_translation_warning(filepath: str):
    # Read in the file
    with open(filepath, 'r', encoding='utf8') as file:
        content = file.read()

    frontmatter_delimiter = '---'
    frontmatter_assignment = ': '
    if content.startswith('+++'):
        # When editing with Forestry.io, TOML front matter is generated.
        # Ensure the correct front matter syntax is used when generating fallback pages for Swedish
        frontmatter_delimiter = '+++'
        frontmatter_assignment = '='
    # Replace the target string
    content = content.replace(frontmatter_delimiter, '', 1)  # Remove the top --- marker
    # Add the leading --- back, along with a property to indicate which page acted as the source
    # Remove the leading .../content/ part of the path
    content = frontmatter_delimiter + '\ngenerated_fallback_page' + frontmatter_assignment +  'true' + content

    # Write the file out again
    with open(filepath, 'w', encoding='utf8') as file:
        file.write(content)


def generate_fallback_page(source_file_path, language_code, dry_run=False):
    translation_path = source_file_path.replace('.md', f'.{language_code}.md')
    print(f"Creating fallback page for lang {language_code}, "
          + f"from {translation_path} to {source_file_path}")
    if not dry_run:
        copy2(source_file_path, translation_path)
        add_missing_translation_warning(translation_path)


def generate_fallback_pages_if_needed(dir: str, dry_run=False):
    for root, dirs, files in os.walk(dir):
        for filename in files:
            filepath = os.path.join(root, filename)
            if filepath.endswith(".md") \
                    and not filepath.endswith(".sv.md") \
                    and not filepath.endswith(".en.md"):
                # This is a content file which isn't language specific
                if not has_translation(filepath, 'sv'):
                    generate_fallback_page(filepath, 'sv', dry_run)
        for dirname in dirs:
            generate_fallback_pages_if_needed(os.path.join(root, dirname), dry_run=dry_run)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.set_defaults(dry=False)
    parser.add_argument('--dry', dest='dry', action='store_true')
    args = parser.parse_args()
    generate_fallback_pages_if_needed(os.path.join(os.getcwd(), 'content/api'), dry_run=args.dry)
    generate_fallback_pages_if_needed(os.path.join(os.getcwd(), 'content/cases'), dry_run=args.dry)
    generate_fallback_pages_if_needed(os.path.join(os.getcwd(), 'content/docs'), dry_run=args.dry)
    generate_fallback_pages_if_needed(os.path.join(os.getcwd(), 'content/news'), dry_run=args.dry)

    # Logowal logos don't need translations, they have no translateable text
    generate_fallback_pages_if_needed(os.path.join(os.getcwd(), 'content/headless/logowall'), dry_run=args.dry)

    # Continue build
    if os.path.exists("public"):
        os.removedirs("public")
    exitCode = os.system("hugo -d public --minify")
    if platform.system() != "Windows":
        exitCode = os.WEXITSTATUS(exitCode)
    print(f"Hugo build process exited with code {exitCode}")
    exit(exitCode)