notes/notes.py at main · zxlwqa/notes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
from io import BytesIO
import random
import string
import sys
import argparse
from huggingface_hub import HfApi

parser = argparse.ArgumentParser(description="创建 Hugging Face Space")
parser.add_argument("--token", type=str, required=True, help="Hugging Face Token（需要写权限）")
parser.add_argument("--image", type=str, default="", help="Docker 镜像地址")
parser.add_argument("--password", type=str, required=True, help="管理员密码")
parser.add_argument("--jwt-secret", type=str, required=True, help="JWT 签名密钥（与登录密码不同）")
parser.add_argument("--webdav-url", type=str, default="", help="WebDAV 地址（可选）")
parser.add_argument("--webdav-user", type=str, default="", help="WebDAV 用户名（可选）")
parser.add_argument("--webdav-pass", type=str, default="", help="WebDAV 密码（可选）")
parser.add_argument("--neon-url", type=str, required=True, help="Neon数据库连接字符串（必需）")
parser.add_argument("--account-id", type=str, default="", help="Cloudflare账户ID（可选）")
parser.add_argument("--access-key-id", type=str, default="", help="R2访问密钥ID（可选）")
parser.add_argument("--secret-access-key", type=str, default="", help="R2秘密访问密钥（可选）")
parser.add_argument("--git-token", type=str, default="", help="GitHub Token（可选）")
args = parser.parse_args()


def generate_random_string(length=2):
    """生成包含至少一个字母的随机字符串"""
    if length < 1:
        return ""
    chars = string.ascii_letters + string.digits
    mandatory_letter = random.choice(string.ascii_letters)
    remaining_chars = random.choices(chars, k=length - 1)
    full_chars = remaining_chars + [mandatory_letter]
    random.shuffle(full_chars)
    return "".join(full_chars)


if __name__ == "__main__":
    token = args.token
    if not token:
        print("Token 不能为空")
        sys.exit(1)

    api = HfApi(token=token)
    user_info = api.whoami()
    if not user_info.get("name"):
        print("未获取到用户名信息，程序退出。")
        sys.exit(1)

    userid = user_info.get("name")

    image = args.image or "ghcr.io/zxlwq/notes:latest"

    space_name = generate_random_string(2)
    repoid = f"{userid}/{space_name}"

    readme_content = f"""
---
title: {space_name}
emoji: 📘
colorFrom: blue
colorTo: purple
sdk: docker
app_port: 3000
pinned: false
---
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
"""
    readme_obj = BytesIO(readme_content.encode("utf-8"))

    secrets = [
        {"key": "PASSWORD", "value": args.password},
        {"key": "JWT_SECRET", "value": args.jwt_secret},
        {"key": "DATABASE_URL", "value": args.neon_url},
        {"key": "WEBDAV_URL", "value": args.webdav_url},
        {"key": "WEBDAV_USER", "value": args.webdav_user},
        {"key": "WEBDAV_PASS", "value": args.webdav_pass},
        {"key": "ACCOUNT_ID", "value": args.account_id},
        {"key": "ACCESS_KEY_ID", "value": args.access_key_id},
        {"key": "SECRET_ACCESS_KEY", "value": args.secret_access_key},
        {"key": "GIT_TOKEN", "value": args.git_token},
    ]


    api.create_repo(
        repo_id=repoid,
        repo_type="space",
        space_sdk="docker",
        space_secrets=secrets,
    )

    api.upload_file(
        repo_id=repoid,
        path_in_repo="README.md",
        path_or_fileobj=readme_obj,
        repo_type="space",
    )

    dockerfile_content = f"FROM {image}\n"
    api.upload_file(
        repo_id=repoid,
        path_in_repo="Dockerfile",
        path_or_fileobj=BytesIO(dockerfile_content.encode("utf-8")),
        repo_type="space",
    )

    print(f"✅ Space 创建成功: {repoid}")