import os
import sys
import subprocess
class admL1Api():
    def __init__(self):
        self.body = ""
    
    def adminLevelOne(self,body):     
        #获取obsid
        obsid=int(str(body)[0])

        #当前模块的id
        current_job_id=os.getenv('JOB_ID')
        print("current_job_id : %s" %(current_job_id))

        sink_job_name=""
        print("body : %s" %(body))
        # body = body.replace(' ', '-')
        # 将数字转换为字符串
        body_str = str(body)
        # 获取前三位
        obsid = body_str[:3]
        print("obsid : %s" %(obsid))
        # 获取后两位
        chipid = body_str[-2:]
        print("chipid : %s" %(chipid))

        if obsid in {"101", "102", "103", "104", "105", "106", "110", "111", "112"}:
            if chipid in {"06", "07", "08", "09", "11", "12", "13", "14", "15", "16", "17", "18", "19", "20", "22", "23", "24", "25"}:
                admL1Api.sum_numbers(body,"msc-l1-mbi",current_job_id)
                
            elif chipid in {"01", "02", "03", "04", "05", "10", "21", "26", "27", "28", "29", "30"}:
                admL1Api.sum_numbers(body,"msc-l1-sls",current_job_id) 
            else:
                print("无效chipid %s" %(body))
                sys.exit(100)
        elif obsid in {"107", "108"}:
            if chipid in {"06", "07", "08", "09", "11", "12", "13", "14", "15", "16", "17", "18", "19", "20", "22", "23", "24", "25"}:
                admL1Api.sum_numbers(body,"msc-l1-mbi",current_job_id)
                
            elif chipid in {"01", "02", "03", "04", "05", "10", "21", "26", "27", "28", "29", "30"}:
                admL1Api.sum_numbers(body,"msc-l1-sls",current_job_id) 
            else:
                print("无效chipid %s" %(body))
                sys.exit(100)
        elif "120" <=obsid <= "129":
            admL1Api.sum_numbers(body,"msc-l1-ooc",current_job_id) 
        else:
            print("无效obsid: %s" %(body))
            sys.exit(101)
    
        print("执行完毕")

    @classmethod 
    def sum_numbers(self,body,sink_job_name,current_job_id):
        print("0000000000000001")
        print(sink_job_name)
        print(body)
        command = f"scalebox task add -sink-job={sink_job_name} {body}"
        result=subprocess.run(command, shell=True)
        if result.returncode == 0:
            print("命令执行成功")
            sys.exit(0)
        else:
            print(f"命令执行失败，返回码为: {result.returncode}")
            return result.returncode
if __name__ == '__main__':
    parameter = sys.argv
    body=parameter[1]
    w=admL1Api()
    w.adminLevelOne(body)