You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ORPA-pyOpenRPA/Orchestrator/orchestratorProcessor.py

135 lines
6.2 KiB

import datetime
import http.client
import json
import pdb
import os
import sys
import subprocess
import win32ts
#Глобальная переменная - все глобальные значения программы
global mGlobalDict
#Включить WTS (WIN32TS)
mWTSServer = lWTSServer = win32ts.WTSOpenServer("localhost")
#{
# actionList:
# [
# {
# type: <RemoteMachineProcessingRun>,
# host: <localhost>,
# port: <port>,
# bodyObject: <object dict, int, str, list>
# },
# {
# type: <ActivityLogScheduleListGet>
# },
# {
# type: <ActivityCMDRun>,
# code: <str>
# },
# {
# type: <ActivityRestartOrchestrator>
# },
# {
# type: <ActivitySessionCheckSetActive>
# },
# {
# type: <AdministrationGlobalDictSetKeyListValue>
# key_list: ["key1","key2",...]
# value: <List, Dict, String, int>
# },
# {
# type: <AdministrationGlobalDictGetKeyListValue>
# key_list: ["key1","key2",...]
# },
# ]
#
#}
def ProcessingRun(inConfigurationDict):
#print(mGlobalDict)
lDateTimeString=datetime.datetime.strftime(datetime.datetime.now(),"%Y.%m.%d %H:%M:%S::%f")
lResult={"dateTime":lDateTimeString, "state":"connected", "actionListResult":[]}
for lItem in inConfigurationDict["actionList"]:
#Добавить входные значения
lResult["actionListResult"].append({"inArgs":lItem})
#Обработка запроса на отправку команды на удаленную машину
if lItem["type"]=="RemoteMachineProcessingRun":
lHTTPConnection = http.client.HTTPConnection(lItem["host"], lItem["port"], timeout=5)
try:
lHTTPConnection.request("POST","/ProcessingRun",json.dumps(lItem["bodyObject"]))
except Exception as e:
#Объединение словарей
lResult["actionListResult"][-1] = {**lResult["actionListResult"][-1], **{"state":"disconnected","errorMessage":str(e)}}
#lResult["actionListResult"][-1].join({"state":"disconnected","errorMessage":str(e)})
else:
lHTTPResponse=lHTTPConnection.getresponse()
lHTTPResponseByteArray=lHTTPResponse.read()
lResult["actionListResult"][-1] = {**lResult["actionListResult"][-1], **json.loads(lHTTPResponseByteArray.decode('utf8'))}
#Обработка команды ActivityLogScheduleListGet
if lItem["type"]=="ActivityLogScheduleListGet":
#pdb.set_trace()
lResult["actionListResult"][-1] = {**lResult["actionListResult"][-1], **{"result":mGlobalDict["ActivityLogScheduleList"]}}
#Обработка команды PlanLogListGet
if lItem["type"]=="PlanLogListGet":
#pdb.set_trace()
lResult["actionListResult"][-1] = {**lResult["actionListResult"][-1], **{"result":mGlobalDict["JSONConfigurationDict"]["activityList"]}}
#Обработка команды ActivityCMDRun
if lItem["type"]=="ActivityCMDRun":
lCMDCode="cmd /c "+lItem["code"]
subprocess.Popen(lCMDCode)
lResultCMDRun=1#os.system(lCMDCode)
lResult["actionListResult"][-1] = {**lResult["actionListResult"][-1], **{"result":str(lResultCMDRun)}}
#Обработка команды ActivityRestartOrchestrator
if lItem["type"]=="ActivityRestartOrchestrator":
os.execl(sys.executable,os.path.abspath(__file__),*sys.argv)
sys.exit(0)
#Обработка команды ActivitySessionCheckSetActive
if lItem["type"]=="ActivitySessionCheckSetActive":
lResult["actionListResult"][-1] = {**lResult["actionListResult"][-1], **{"result":{"IsActivated":False}}}
lPID = os.getpid()
#pdb.set_trace()
lSessionId = win32ts.ProcessIdToSessionId(lPID)
lSessionList = win32ts.WTSEnumerateSessions(lWTSServer)
#При попытке закрыть сервер возникает ошибка
#win32ts.WTSCloseServer(lWTSServer)
#Наложить фильтр
lSessionFiltered = [d for d in lSessionList if d["SessionId"]==lSessionId]
#Выполнить переход в активное состояние , если State != 0
if lSessionFiltered[0]["State"] != 0:
lCMDCodeTSCON = "tscon " + str(lSeessionId) + " /dest:console"
lCMDCode="cmd /c "+lCMDCodeTSCON
subprocess.Popen(lCMDCode)
lResult["actionListResult"][-1] = {**lResult["actionListResult"][-1], **{"result":{"IsActivated":True}}}
#Выключить соединение
#Обработка команды AdministrationGlobalDictSetKeyListValue
if lItem["type"]=="AdministrationGlobalDictSetKeyListValue":
lGlobalDict=mGlobalDict["JSONConfigurationDict"]
for lItem2 in lItem["key_list"][:-1]:
#Check if key - value exists
if lItem2 in lGlobalDict:
pass
else:
lGlobalDict[lItem2]={}
lGlobalDict=lGlobalDict[lItem2]
#Set value
#pdb.set_trace()
lGlobalDict[lItem["key_list"][-1]]=lItem["value"]
#Обработка команды AdministrationGlobalDictGetKeyListValue
if lItem["type"]=="AdministrationGlobalDictGetKeyListValue":
lGlobalDict=mGlobalDict["JSONConfigurationDict"]
for lItem2 in lItem["key_list"][:-1]:
#Check if key - value exists
if lItem2 in lGlobalDict:
pass
else:
lGlobalDict[lItem2]={}
lGlobalDict=lGlobalDict[lItem2]
#Set value
#pdb.set_trace()
lResult["actionListResult"][-1]["key_list"]=lItem["key_list"]
lResult["actionListResult"][-1]["value"]=lGlobalDict.get(lItem["key_list"][-1],None)
#Вернуть результат
return lResult