mgithub/GithubWork.py at main · Websoft9Archive/mgithub · 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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
#!/usr/bin/env python3
# coding=utf-8
import os
import signal
import sys
from operator import methodcaller

from GithubProductCmd import GithubProductCmd
from GithubException import CustomException
from GithubUtils import GithubUtils

#########################################
#          GithubWorks                  #
# mgithub loop framework class          #
# Created by: Haozhe Chen               #
#########################################

class GithubWork():

    def __init__(self, ctx):
        self.ctx = ctx
        self.current_proj = None
        self.repo_str = None
        self.skip_broken = ctx.obj['skip_broken']
        self.skip_get_repo = ctx.obj['skip_get_repositories']
        self.organization = ctx.obj['organization']
        self.url = ctx.obj['url']
        self.product_kind = ctx.obj['product_kind']

    # 添加对Ctrl-C的监听器
    def signal_handler(self, signal, frame):
        if os.path.isfile(self.repo_str):
            if self.current_proj is None:
                # 如果当前还没有开始执行项目，则任务项目列表中的第一个项目执行失败
                GithubUtils().complete_work("ABORT", open(self.repo_str).read().splitlines()[0], self.ctx)
            else:
                # 如果已经开始执行项目，则当前项目执行失败
                GithubUtils().complete_work("ABORT", self.current_proj, self.ctx)
                GithubUtils().rollback_proj(self.current_proj, self.organization)
        print("用户主动中断任务")
        sys.exit(0)

    # 对本组织下的仓库进行循环处理任务
    def auto_make(self):

        # 将监听器注册到本function
        signal.signal(signal.SIGINT, self.signal_handler)

        print("============================ Automake task starting... ============================ \n")

        self.ctx.obj['repo_str'] = "data/" + self.ctx.obj['organization'] + "_repositories.txt"
        self.repo_str = self.ctx.obj['repo_str']

        # 判断项目列表是否存在
        if os.path.isfile(self.repo_str):
            if len(open(self.repo_str).read().splitlines()) == 0:
                # 空的项目列表
                print("项目列表为空，请执行mgithub [option] repocache来获取本组织/用户的项目列表")
                return
            else:
                # 非空项目列表
                # 向用户展示当前项目列表
                print("当前项目列表内容：")
                GithubUtils.execute_CmdCommand("cat " + self.repo_str)

                # 等待用户确认列表待办事项
                if GithubUtils().continue_select() == "0":

                    # 获取项目列表内容
                    # project_list = open(self.repo_str).read().splitlines()
                    project_list = []
                    for line in open(self.repo_str).read().splitlines():
                        info = line.split(" ")
                        project_list.append(info[0])

                    # # 根据项目列表，将每个项目的远程仓库clone到本地
                    # try:
                    #     GithubHelperFunc().clone_repo_list(project_list, self.skip_get_repo,
                    #                                    self.skip_broken, self.organization, self.url)
                    # except CustomException as e:
                    #     print(e.msg)
                    #     if str(self.skip_broken) != "True":
                    #         return

                    # 根据项目列表，对每个项目进行循环command操作
                    self.loop_proj_work(project_list)
                else:
                    return
        else:
            print("未找到项目列表，请执行mgithub [option] repocache来获取本组织/用户的项目列表")
            return

    # 根据项目列表，对每个项目进行循环command操作
    def loop_proj_work(self, project_list):
        for proj in project_list:
            # 记录本次正在执行command的项目，方便与监听器记录
            self.current_proj = proj
            # 执行相应的command
            try:
                self.product_execute(proj)
            except CustomException as e:
                # 捕捉执行过程中出现异常
                print(e.msg)
                # 已 FAILED 记录本次任务
                GithubUtils().complete_work("FAILED", proj, self.ctx)
                # 对项目进行回滚
                GithubUtils().rollback_proj(proj, self.organization)
                # 如果用户使用 mgithub --skip-broken
                if str(self.skip_broken) == "True":
                    continue
                else:
                    break

    def product_execute(self, project):

        print("\n============================ [[" + project + "]]: 开始执行自动化构建")

        # 对本地仓库进行更新
        GithubUtils().update_repo(self.organization, self.url, project)

        # 创建GithubProduct对象并传入对应项目与ctx参数
        product = GithubProductCmd(self.ctx)

        try:
            # 利用operator.methodcaller从字符串反射出函数名来执行函数，简化extension步骤
            methodcaller(self.product_kind, project)(product)
            # 将本地改动PUSH到远程仓库
            print("\n正在将本地改动push到远程仓库...")
            GithubUtils().push_repo(project, self.organization, self.product_kind)
        except CustomException as e:
            raise e

        # 本次任务完成
        GithubUtils().complete_work("OK", project, self.ctx)