PythonのテンプレートエンジンJinja2を使ってみる

{{ 変数名 }}

{% if 条件 %}

{% else %}

{% endif %}

{% for 要素 in リスト %}

{%endfor%}

{# コメント #}

{%- if is_am -%}
おはようございます。
{%- else -%}
こんにちは。
{%- endif %}
はじめまして、私は{{ name }}と申します。
現在{{ company }}で{{ work }}をしています。

{#- 複数表示可能 #}
趣味は
{%- for hobby in hobbies -%}
{{ hobby }} {% if not loop.last %},{% endif %}
{%- endfor -%}
です。
よろしくお願いいたします。

from jinja2 import Environment, FileSystemLoader

# テンプレートファイルがあるディレクトリを取得
env = Environment(loader=FileSystemLoader('templates'))

# テンプレートファイルの読み込み
tpl = env.get_template('self_introduction.txt')

# テンプレートに埋め込むデータを用意
data = {
    'is_am': False,
    'name': '前田',
    'company': 'ロジカルビート',
    'work':'ゲーム開発',
    'hobbies':['ゲーム','映画観賞','散歩']
}

# テンプレートファイルに埋め込む
output = tpl.render(data)

# テキストファイルに保存
with open('output.txt', 'w', newline ='\n', encoding='utf-8') as f:
    f.write(output)

public class StateMachine
{
    public IState CurrentState { get; private set; }

    public void ChangeState(IState nextState)
    {
        CurrentState?.Exit();
        CurrentState = nextState;
        CurrentState.Enter();
    }

    public void Update(float deltaTime)
    {
        CurrentState.Update(deltaTime);
    }
}

public interface IState
{
    public void Enter() { }

    public void Update(float deltaTime) { }

    public void Exit() { }
}

using UnityEngine;

public class IdleState : IState
{
    public void Enter()
    {
        Debug.Log("IdleStateのEnter()実行！");
    }

    public void Update(float deltaTime)
    {
        Debug.Log("IdleStateのUpdate()実行！");
    }

    public void Exit()
    {
        Debug.Log("IdleStateのExit()実行！");
    }
}

using UnityEngine;

public class MoveState : IState
{
    public void Enter()
    {
        Debug.Log("MoveStateのEnter()実行！");
    }

    public void Update(float deltaTime)
    {
        Debug.Log("MoveStateのUpdate()実行！");
    }

    public void Exit()
    {
        Debug.Log("MoveStateのExit()実行！");
    }
}

using UnityEngine;

public class {{ class_name }} : IState
{
    public void Enter()
    {
        Debug.Log("{{ class_name }}のEnter()実行！");
    }

    public void Update(float deltaTime)
    {
        Debug.Log("{{ class_name }}のUpdate()実行！");
    }

    public void Exit()
    {
        Debug.Log("{{ class_name }}のExit()実行！");
    }
}

import os
from pathlib import Path
import argparse
from jinja2 import Environment, FileSystemLoader

# Unityからクラス名を取得
parser = argparse.ArgumentParser()
parser.add_argument("--class_name", default = "DefaultClass")
args = parser.parse_args()

# テンプレートファイルが置かれているディレクトリを指定
base_dir = os.path.dirname(os.path.abspath(__file__))
template_dir = os.path.join(base_dir, "templates")
file_loader = FileSystemLoader(template_dir)
env = Environment(loader = file_loader)

# テンプレートファイルの読み込み
template = env.get_template('class.txt')

# テンプレートに埋め込むデータを用意
data = {'class_name': args.class_name}

# テンプレートファイルに埋め込む
output = template.render(data)

parent_dir = Path(__file__).resolve().parent.parent
output_file_dir = os.path.join(parent_dir, "Scripts")
output_file = os.path.join(output_file_dir, args.class_name +".cs")

# C#スクリプト生成
with open(output_file, 'w', encoding='utf-8') as f:
    f.write(output)

using UnityEngine;
using UnityEditor;
using System.Diagnostics;

public class GeneratorWindow : EditorWindow
{
    private string arg = "";

    [MenuItem("Tools/IState Generator")]
    public static void ShowWindow()
    {
        GetWindow<GeneratorWindow>("IState Generator");
    }

    void OnGUI()
    {
        GUILayout.Label("クラス名入力", EditorStyles.boldLabel);
        arg = EditorGUILayout.TextField("クラス名", arg);

        if (GUILayout.Button("コード生成開始"))
        {
            if (string.IsNullOrWhiteSpace(arg))
            {
                UnityEngine.Debug.LogError("class_name が未指定です");
                return;
            }

            Run(arg);
        }
    }

    void Run(string arg)
    {
        var fileName = "python.exe"; // フルパスを指定してください
        var scriptPath = $"{Application.dataPath}//Tool//generator.py";
        
        var info  = new ProcessStartInfo();
        info.FileName = fileName;
        info.Arguments = $"\"{scriptPath}\" --class_name \"{arg}\"";

        var process = Process.Start(info);
        process.WaitForExit();
    }
}