auto-mini-program-tool/server/main.py

import os
import flask
import requests
import json
import time
from flask import Flask,request,redirect
from flask_cors import CORS 
app = Flask(__name__,static_folder="static")
CORS(app, resources=r'/*')


# current_path
current_path = os.getcwd()
res_json = {"code":0,"msg":"成功"}
send_status = {'a':False}

# 1.小程序打包
def build_mini_program(branch):
    res = False
    command_str = (
    "cd "+current_path+ ";"
    "cd longVideoFactory;"
    "rm -rf dist;"
    "git checkout "+branch+";"
    "git pull;"
    "n 10;"
    "cnpm install && npm install ;"
    "npm run build && npm run build:other;"
    )
    
    if os.system(command_str) == 0:
        res_json["code"] = 0
        res_json["msg"] = "打包成功"
        
    else:
        res_json["code"] = 1
        res_json["msg"] = "打包失败"

    
# # 3.微信编译图片
def create_preview(project,page):
    # 小程序 cli path for mac
    wx_cli ="/Applications/wechatwebdevtools.app/Contents/MacOS/cli "
    
    dist_path =  os.path.join(current_path,'longVideoFactory/dist/'+project)
    code_img_path =  os.path.join(current_path,'1.png')
    
    path = """ '{"pathName":"%s"}' """ % page
    os.system(wx_cli+ " open --project " + dist_path)
    os.system(wx_cli+ " reset-fileutils --project " + dist_path)
    print(wx_cli+" preview  --project "+dist_path+"  --compile-condition "+path+"   --qr-output "+code_img_path+" --qr-format image")
    f_read = os.popen(wx_cli+" preview  --project "+dist_path+"  --compile-condition "+path+"   --qr-output "+code_img_path+" --qr-format image").read()
    os.system(wx_cli+ " quit")
    if "[error]" in f_read:
        res_json["code"] = 3
        res_json["msg"] = f_read
    else:
        res_json["code"] = 0
        res_json["msg"] = "微信小程序编译成功"

# 修改环境文件
def edit_mode_by_file(project,mode):
    path = os.path.join(current_path,"longVideoFactory/dist/"+project+"/network/volatileConfig.js")
    file_lines = []
    with open(path,"r") as f:
        for item in f.readlines():
            if "export const currentMode" in item:
                item = "export const currentMode = MODE."+mode+" \n"

            file_lines.append(item)
        f.close()
    
    with open(path,"w") as f2:
        f2.writelines(file_lines) 
        f2.close()
    res_json["code"] = 0
    res_json["msg"] = "文件修改成功"
    

def upload_img_by_feishu(token):
    key = ""
    url = "https://open.feishu.cn/open-apis/im/v1/images"
    files = {
        "image_type":(None,"message"),
        "image":('1.png',open(os.path.join(current_path,"1.png"),"rb"),"image/png")
    }
    headers = {
        "Authorization":"Bearer "+token,
    }
    res = requests.post(url,headers=headers,files=files).json()
    if res["code"] == 0:
        key = res["data"]["image_key"]
        res_json["code"] = 0
    else:
        res_json["code"] = 4
        res_json["msg"] = "飞书上传图片失败"
        
    return key

def tenant_access_token_by_feishu():
    token = ""
    url = "https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal"
    params = {
        "app_id":"cli_a1ec07458838d00c",
        "app_secret":"Ngm0kfaLkZnpM2T2DH7L8cM5hdAqY0hI"
    }
    res = requests.post(url,data=params).json()
    if res["code"] == 0:
        token = res["tenant_access_token"]
        res_json["msg"] = "获取飞书签名成功"
        res_json["code"] = 0
    else:
        res_json["code"] = 4
        res_json["msg"] = "获取飞书签名失败"

    return token

def send_msg_by_feishu(token,img_key,url,project_name,mode_name):
    # webhook
    # webhook_url = "https://open.feishu.cn/open-apis/bot/v2/hook/a924b6a5-86ef-4c76-a7e4-eb42056f70cc"
    webhook_url = url
    headers = {
        "Authorization":"Bearer "+token,
        "Content-Type": "application/json;charset=UTF-8"
    }
    params = {
        "msg_type": "post",
        "content": {
            "post": {
            "zh_cn": {
                "title": mode_name+':'+project_name+"小程序预览码",
                "content": [
                    [
                        {
                            "tag": "img",
                            "image_key": img_key,
                            "width": 500,
                            "height": 500
                        }
                    ]]
                }
            }
        }
    }
    
    res = requests.post(webhook_url,headers=headers,data=json.dumps(params)).json()
    if res["StatusCode"] == 0:
        res_json["code"] = 0
        res_json["msg"] = "飞书消息发送成功"
    else:
        res_json["code"] = 5
        res_json["msg"] = "飞书消息发送失败"


@app.route('/')
def func():    
    # return 'Welcome to auto-mini-program-build-preview server. only can run on MacOS.'
    return redirect("static/dist/index.html")
     
@app.route('/send',methods=['POST'])
def send():
    global send_status
    try:
        if send_status == True:
            res_json["code"] = 9
            res_json["msg"] = "当前有人打包，请稍后尝试"
            return res_json
        send_status = True
        res_json["code"] = 0
        res_json["msg"] = "成功"
        webhook_url = request.json["webhook"]
        branch = request.json["branch"]
        page = request.json["page"]
        project = request.json["project"]
        mode = request.json["mode"]
        project_name = request.json["project_name"]
        mode_name = request.json["mode_name"]
        # ------- 微信打包 ------
        # 1.小程序打包
        build_mini_program(branch)
        if res_json["code"] != 0:
            send_status = False
            return res_json
        # 2.修改环境
        edit_mode_by_file(project,mode)
        if res_json["code"] != 0:
            send_status = False
            return res_json
        # 3.微信编译图片
        time.sleep(1)
        create_preview(project,page)
        

        # -------- 飞书 --------
        # 1.获取签名
        if res_json["code"] != 0:
            send_status = False
            return res_json
        token = tenant_access_token_by_feishu()
        # # 2.获取imgKey
        if res_json["code"] != 0:
            send_status = False
            return res_json
        img_key = upload_img_by_feishu(token)
        # # 3.飞书发送消息
        if res_json["code"] != 0:
            send_status = False
            return res_json
        send_msg_by_feishu(token,img_key,webhook_url,project_name,mode_name)

        send_status = False
    except:
        send_status = False

    return res_json


@app.route("/getBranch")
def get_branch():
    # 查看所有分支
    command_str= (
        "cd "+current_path+ ";"
        "git clone https://git.yishihui.com/weapp/longVideoFactory.git;"
        "cd longVideoFactory;"
        "git branch -a;"
    )
    f_read = os.popen(command_str).read()

    branch_list = []
    for item in f_read.split():
        if "->" not in item and "origin/master" not in item and "HEAD" not in item and "*" not in item:
            branch_list.append(item.replace("remotes/origin/",""))

    res = {"code":0,"data":branch_list}
    return res

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7777, debug=False)