reposync/pr_sync/pr_gitlink_to_gitee.py

import urllib
import requests
import mysql.connector

####################################先将gitee的pr状态更改为合并#####################################
def prework():
    # 连接到MySQL数据库
    conn = mysql.connector.connect(
        host="localhost",
        user="rtsw",
        password="123456",
        database="reposyncer"
    )
    cursor = conn.cursor()
    cursor.execute('SELECT number, state FROM gitee_pr_list')

    gitee_pr_info = cursor.fetchall()
    for row in gitee_pr_info:
        number = row[0]
        state = row[1]
        owner = 'xumingyang21'
        repo = 'reposyncer2'
        if state == 'open':
            # Gitee获取issue的API URL
            url = f'https://gitee.com/api/v5/repos/{owner}/{repo}/pulls/{number}'
            # 构造请求头
            token = 'f2be2313581c1fde50b16bf35bb655c5'
            headers = {'Authorization': f'token {token}'}
            # 发送POST请求 获取状态为open的issue
            data = {
                "access_token": token,
                "owner": owner,
                "repo": repo,
                "title": '尝试更改pr',
                "body": '好冲',
                "state": 'closed'
            }
            # 将字典转换为查询字符串
            query_string = urllib.parse.urlencode(data)
            # 完整的请求 URL，包括查询字符串
            full_url = f"{url}?{query_string}"
            response = requests.patch(full_url, headers=headers, json=data)

            if response.status_code == 200:
                # 注意：这里假设响应体中包含一个'number'字段作为issue的ID，但实际上Gitee可能返回不同的结构
                # 你需要根据实际的响应结构来调整以下代码
                pr_info = response.json()
                print(pr_info)
            else:
                print(f"获取pr失败，状态码：{response.status_code}，错误信息：{response.text}")

    #########################获取gitlink中的信息#########################################

    cursor.execute('DROP TABLE IF EXISTS gitlink_pr_list')
    cursor.execute('''CREATE TABLE IF NOT EXISTS gitlink_pr_list
                      (id INT AUTO_INCREMENT PRIMARY KEY, number VARCHAR(10), state VARCHAR(15), title VARCHAR(100), body VARCHAR(1000), head VARCHAR(15), base VARCHAR(15))''')

    url = "https://gitlink.org.cn/api/v1/xumingyang21/reposyncer2/pulls.json?keyword&status&priority_id&issue_tag_id&version_id&reviewer_id&assign_user_id&sort_by&sort_direction"

    payload = {}

    access_token = 'I9uyRzjEIObdgsD8fegdQoN2d3p3cU_7_uaNGV03S_Q'
    headers = {
        'Authorization': f'Bearer {access_token}',  # 使用 Bearer 标记
        'User-Agent': 'Apifox/1.0.0 (https://apifox.com)'
    }

    response = requests.request("GET", url, headers=headers, data=payload)
    data = response.json()
    for pr_info in data['pulls']:
        print('id:', pr_info['id'])
        print('state:', pr_info['status'])
        print('title:', pr_info['title'])
        print('body:', pr_info['body'])
        print('head:', pr_info['head'])
        print('base:', pr_info['base'])
        cursor.execute(
            'INSERT INTO gitlink_pr_list (number, state, title, body, head, base) VALUES (%s, %s, %s, %s, %s, %s)',
            (pr_info['id'], pr_info['status'], pr_info['title'], pr_info['body'], pr_info['head'], pr_info['base']))
    print(response.text)

    # 提交事务
    conn.commit()
    # 关闭游标和连接
    cursor.close()
    conn.close()


#################################这个地方执行仓库的同步########################################


################################在gitee里面添加新的pr#######################################
def push():
    # 连接到MySQL数据库
    conn = mysql.connector.connect(
        host="localhost",
        user="rtsw",
        password="123456",
        database="reposyncer"
    )
    cursor = conn.cursor()
    owner = 'xumingyang21'
    repo = 'reposyncer2'
    cursor.execute('SELECT state, title, body, head, base FROM gitlink_pr_list')
    gitlink_pr_info = cursor.fetchall()
    for row in gitlink_pr_info:
        gitlink_state = row[0]
        gitlink_title = row[1]
        gitlink_body = row[2]
        gitlink_head = row[3]
        gitlink_base = row[4]
        if gitlink_state == 'open':
            # Gitee创建pr的API URL
            url = f'https://gitee.com/api/v5/repos/{owner}/{repo}/pulls'
            # 构造请求头
            token = 'f2be2313581c1fde50b16bf35bb655c5'
            headers = {'Authorization': f'token {token}'}

            # 发送POST请求 获取状态为open的issue
            data = {
                "access_token": token,
                "owner": owner,
                "repo": repo,
                "title": gitlink_title,
                "body": gitlink_body,
                "head": gitlink_head,
                "base": gitlink_base
            }

            # 将字典转换为查询字符串
            query_string = urllib.parse.urlencode(data)
            # 完整的请求 URL，包括查询字符串
            full_url = f"{url}?{query_string}"
            response = requests.post(full_url, headers=headers, json=data)
            # 打印响应
            # print(response.text)
            # 检查响应状态码
            if response.status_code == 201:
                # 注意：这里假设响应体中包含一个'number'字段作为issue的ID，但实际上Gitee可能返回不同的结构
                # 你需要根据实际的响应结构来调整以下代码
                pr_info = response.json()
                print(pr_info)

            else:
                print(f"获取pr失败，状态码：{response.status_code}，错误信息：{response.text}")
    # 提交事务
    conn.commit()
    # 关闭游标和连接
    cursor.close()
    conn.close()