#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""专题九 · 步骤 1：验证控制台登录（base64 密码 + X-CSRF-Token）。"""
from __future__ import annotations

import sys

import _bootstrap  # noqa: F401
import _config as C
from dify_api import Dify, pp


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        ok, _, csrf = d.login()
        print("LOGIN ok =", ok, "csrf =", bool(csrf))
        if not ok:
            return 1
        r = d.get("/datasets")
        pp("datasets（登录后会话可用）", r)
        return 0 if r.get("code") == "200" else 1
    finally:
        d.close()


if __name__ == "__main__":
    sys.exit(main())