#!/usr/bin/env python3

"""

QMD File Formatter - Ensures consistent formatting across all QMD files

"""

import os

import re

from pathlib import Path

def format_qmd_content(content):

"""Format QMD content for consistency"""

lines = content.split('\n')

formatted_lines = []

in_yaml_front_matter = False

in_code_block = False

yaml_delimiter_count = 0

for i, line in enumerate(lines):

# Track YAML front matter

if line.strip() == '---':

yaml_delimiter_count += 1

if yaml_delimiter_count <= 2:

in_yaml_front_matter = (yaml_delimiter_count == 1)

# Track code blocks

if line.strip().startswith('```'):

in_code_block = not in_code_block

# Skip formatting inside code blocks

if in_code_block:

formatted_lines.append(line)

continue

# Skip formatting inside YAML front matter

if in_yaml_front_matter:

formatted_lines.append(line)

continue

# Remove trailing whitespace

line = line.rstrip()

# Format headers - ensure space after #

if line.startswith('#') and not line.startswith('####'):

# Count the number of # characters

hash_count = 0

for char in line:

if char == '#':

hash_count += 1

else:

break

# Extract title after #

title = line[hash_count:].strip()

if title:

line = '#' * hash_count + ' ' + title

# Add blank line before headers (except first line and if previous line is already blank)

if (line.startswith('#') and

i > 0 and

formatted_lines and

formatted_lines[-1].strip() != '' and

not formatted_lines[-1].startswith('#')):

formatted_lines.append('')

# Add blank line after headers (if next line isn't blank and isn't another header)

should_add_blank_after_header = False

if line.startswith('#') and i < len(lines) - 1:

next_line = lines[i + 1].strip()

if next_line != '' and not next_line.startswith('#'):

should_add_blank_after_header = True

formatted_lines.append(line)

if should_add_blank_after_header:

# Check if next line will be blank

if i < len(lines) - 1 and lines[i + 1].strip() != '':

formatted_lines.append('')

# Format bullet points - ensure consistent spacing

if line.strip().startswith('- ') or line.strip().startswith('* '):

# Ensure single space after bullet

stripped = line.strip()

if stripped.startswith('- '):

content_after_bullet = stripped[2:]

else: # starts with '* '

content_after_bullet = stripped[2:]

# Count leading spaces for indentation

leading_spaces = len(line) - len(line.lstrip())

formatted_lines[-1] = ' ' * leading_spaces + '- ' + content_after_bullet

# Remove excessive blank lines (more than 2 consecutive)

final_lines = []

blank_count = 0

for line in formatted_lines:

if line.strip() == '':

blank_count += 1

if blank_count <= 2:

final_lines.append(line)

else:

blank_count = 0

final_lines.append(line)

# Ensure file ends with single newline

while final_lines and final_lines[-1] == '':

final_lines.pop()

return '\n'.join(final_lines) + '\n'

def format_qmd_file(file_path):

"""Format a single QMD file"""

try:

with open(file_path, 'r', encoding='utf-8') as f:

original_content = f.read()

formatted_content = format_qmd_content(original_content)

# Only write if content changed

if formatted_content != original_content:

with open(file_path, 'w', encoding='utf-8') as f:

f.write(formatted_content)

return True

return False

except Exception as e:

print(f"Error formatting {file_path}: {e}")

return False

def main():

"""Format all QMD files in the repository"""

base_path = Path('/home/ryuichi/dev/python-tutorial')

qmd_files = list(base_path.glob('**/*.qmd'))

print(f"Found {len(qmd_files)} QMD files to format")

formatted_count = 0

for qmd_file in qmd_files:

if format_qmd_file(qmd_file):

print(f"Formatted: {qmd_file}")

formatted_count += 1

else:

print(f"No changes: {qmd_file}")

print(f"\nFormatting complete! {formatted_count} files were updated.")

if __name__ == '__main__':

main()