Files
comp_auto_restart/prog.py
2022-03-10 13:19:39 +07:00

49 lines
1.7 KiB
Python

#!/usr/bin/python3
# -*- coding: utf-8 -*-
from sys import exit, argv
# from os import system
from subprocess import Popen
from filesio import FilesIO
from match_name import check_arm_name
file_name = ".\\computer_names.json"
default_values = {
"1": [ # Пятидневка
"IT01",
"IT02"
],
"2": [ # Цех
"PIS09",
"PIS10"
]
}
loaded_file = None
def main(prog_key):
global loaded_file
list_pc_names = loaded_file.get(prog_key)
if list_pc_names is not None:
for pc_name in list_pc_names:
formatted_pc_name = check_arm_name(pc_name)
if formatted_pc_name is not False:
# print("pc_name:", formatted_pc_name)
# system("shutdown /m \\{} /r /t 60 /c \"Плановая перезагрузка компьютера через 1 минуту!\"".format(formatted_pc_name))
# Popen("shutdown /m \\{} /r /t 60 /c \"Плановая перезагрузка компьютера через 1 минуту!\"".format(formatted_pc_name)).wait()
print("shutdown /m \\{} /r /t 60 /c \"Плановая перезагрузка компьютера через 1 минуту!\"".format(formatted_pc_name))
else:
print("Попробуйте еще раз выбрать область запуска скрипта!")
if __name__ == "__main__":
if len(argv) < 2:
print("Укажите номер запускаемой области!\n1 - Кабинеты; 2 - Цех")
exit()
try:
filesio = FilesIO(file_name, default_values)
loaded_file = filesio.get_data()
main(argv[1])
except KeyboardInterrupt:
exit()