Lslightly/gitlab-issue-bot

1from typing import List2import os3import sys4import gitlab5from gitlab import const, exceptions6import pandas as pd7from dotenv import load_dotenv8 9 10 11def invite_users_to_group(gl: gitlab.Gitlab, group_name: str, user_ids: List[int], access_level=const.AccessLevel.DEVELOPER):12    try:13        group = gl.groups.get(group_name)14    except exceptions.GitlabGetError as e:15        print(f"❌ Failed to get group '{group_name}': {e}")16        sys.exit(1)17 18    for uid in user_ids:19        try:20            # 检查是否已经是成员21            try:22                group.members.get(uid)23                print(f"ℹ️ User {uid} is already a member of '{group_name}'")24                continue25            except exceptions.GitlabGetError:26                pass  # 用户不是成员，可以继续邀请27 28            group.invitations.create(data={29                "user_id": str(uid),30                "access_level": access_level,31            })32            print(f"✅ Invitation sent to user {uid} for group '{group_name}'")33        except exceptions.GitlabCreateError as e:34            print(f"❌ Failed to invite user {uid}: {e}")35 36 37if __name__ == "__main__":38    load_dotenv()39    token = os.getenv("GITLAB_PERSONAL_ACCESS_TOKEN")40    if not token:41        print("❌ Please provide private token via GITLAB_PERSONAL_ACCESS_TOKEN env var")42        sys.exit(1)43 44    url = os.getenv("GITLAB_URL", "https://git.lug.ustc.edu.cn/")45    group_name = os.getenv("GITLAB_GROUP", "Compiler25")46 47    gl = gitlab.Gitlab(url=url, private_token=token)48 49    df = pd.read_csv("uid.csv")50    uids = df["uid"].tolist()51 52    invite_users_to_group(gl, group_name, uids)53