maildir_create_from_csv.py（Maildir 一括作成）

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Created by Inajin (InaPC Linux Lab)
# AI（Microsoft Copilot）との共同制作教材
#
# Copyright (c) 2026
# 本スクリプトは教育目的での利用・改変を自由に行えます。
# ただし、著作権表記は削除せず残してください。
#

import csv
import os
import subprocess
import sys
from tkinter import Tk, filedialog
from datetime import datetime

DOCUMENT_ROOT = "/var/www/html"
LOG_FILE = "/var/log/maildir_create.log"


def log(message):
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    with open(LOG_FILE, "a") as f:
        f.write(f"[{timestamp}] {message}\n")
    print(message)


def select_csv_file():
    root = Tk()
    root.withdraw()
    file_path = filedialog.askopenfilename(
        title="Maildir を作成する CSV を選択してください",
        filetypes=[("CSV Files", "*.csv")]
    )
    return file_path


def fix_permissions(user, maildir):
    subprocess.run(["chown", "-R", f"{user}:ftpuser", maildir])
    subprocess.run(["chmod", "-R", "775", maildir])
    log(f"[FIX] {user}: 既存 Maildir の所有者と権限を修正しました")


def create_maildir(user, folder):
    home = f"{DOCUMENT_ROOT}/{folder}"
    maildir = f"{home}/Maildir"

    # Maildir が既に存在する場合 → 所有者・権限を修正して終了
    if os.path.exists(maildir):
        fix_permissions(user, maildir)
        return

    # Maildir とサブフォルダを作成
    commands = [
        f"sudo -u {user} maildirmake.dovecot {maildir}",
        f"sudo -u {user} maildirmake.dovecot {maildir}/.Drafts",
        f"sudo -u {user} maildirmake.dovecot {maildir}/.Sent",
        f"sudo -u {user} maildirmake.dovecot {maildir}/.Trash",
    ]

    for cmd in commands:
        subprocess.run(cmd, shell=True)

    # 所有者と権限を設定
    fix_permissions(user, maildir)

    log(f"[OK] {user}: Maildir を新規作成しました")


def main():
    if os.geteuid() != 0:
        print("エラー：このスクリプトは sudo で実行する必要があります。")
        sys.exit(1)

    csv_path = select_csv_file()
    if not csv_path:
        print("キャンセルされました。処理を終了します。")
        sys.exit(0)

    log("=== Maildir 自動作成開始 ===")
    log(f"選択された CSV: {csv_path}")

    with open(csv_path, newline="", encoding="utf-8") as csvfile:
        reader = csv.reader(csvfile)
        next(reader)

        for row in reader:
            if len(row) < 3:
                continue

            folder, user, password = row
            folder = folder.strip()
            user = user.strip()

            log(f"--- {user} の Maildir 処理開始 ---")
            create_maildir(user, folder)

    log("=== 全 Maildir 作成処理完了 ===")


if __name__ == "__main__":
    main()