# TODO: efficiently evaluate commands

# File processing for getting COMMAND LIST and VALID COMMANDS

script_dir = os.path.dirname(__file__)

path = os.path.join(script_dir, fi)

try:

with open(path, "r") as fh:

CommandLine = fh.readlines()

except IOError as err:

CommandLine = file_data

commandList=[]

validList=[]

for line in CommandLine:

if '[COMMAND LIST]' in line:

commandFlag = True

validFlag = False

elif '[VALID COMMANDS]' in line:

commandFlag = False

validFlag = True

if '[COMMAND LIST]' not in line and '[VALID COMMANDS]' not in line and line.rstrip() != '':

if commandFlag :

commandList.append(line.rstrip())

elif validFlag :

validList.append(line.rstrip())

#Get all the valid commands

validCommandsFromInput = set(commandList) & set(validList)

for command in validCommandsFromInput:

try:

try:

start = time.time()

commandResult = subprocess.check_output(command, shell=True, stderr=STDOUT, timeout=60 )

CommandTimeTaken = (time.time() - start)

except subprocess.TimeoutExpired as err:

print('Handling TimeoutExpired: ', err, ' for command: ', command)

CommandTimeTaken = 0 # (if the command takes > 1 minute to complete, mark a 0 which will represent "Not finished")

commandResult = b''

CommandString = command

commandLength = len(command)

print(commandResult)

print(CommandString)

print(commandLength)

print(CommandTimeTaken)

# Check if the commad is already in Table

flag = session.query(Command).filter_by(command_string=CommandString).first()

if flag:

print('Command : "', command, '" is alredy in commands table')

else:

ed_commands = Command(CommandString, commandLength, CommandTimeTaken, commandResult)

session.add(ed_commands)

session.commit()

except subprocess.CalledProcessError as err:

print('Handling CalledProcessError: ', err, ' for command: ', command)

continue