Update

Signed-off-by: Lev Rusanov <30170278+JDM170@users.noreply.github.com>

main.py

@@ -0,0 +1,65 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+
+from sys import exit, argv
+from os import system
+from subprocess import Popen
+from time import sleep
+from filesio import FilesIO
+from match_name import MatchIO
+
+
+default_values = {
+    "expr_list": [
+        [r"^[a-zA-Z]+\d+$", "R54-630300"],  # THE01
+        [r"^\d+[a-zA-Z]+\d+$", "R54-"],  # 630300THE01
+        [r"^[rR]\d*[-]\d+[a-zA-Z]+\d+$", ""]  # R54-630300THE01
+    ],
+    "pc": {
+        "2": [  # Пятидневка
+            "IT01",
+            "IT02"
+        ],
+        "7": [  # Цех
+            "PIS09",
+            "PIS10"
+        ]
+    }
+}
+loaded_file = None
+
+
+def main(comp_key, is_prod):
+    global loaded_file
+    list_pc_names = loaded_file.get("pc").get(comp_key)
+    if list_pc_names is not None:
+        for pc_name in list_pc_names:
+            formatted_pc_name = matchio.check_arm_name(pc_name)
+            if formatted_pc_name is not False:
+                # print("pc_name:", formatted_pc_name)
+                if is_prod is True:
+                    # system("shutdown /m \\\{} /r /f /t 60 /c \"Плановая перезагрузка компьютера через 1 минуту!\"".format(formatted_pc_name))
+                    Popen("shutdown /m \\\{} /r /f /t 60 /c \"Плановая перезагрузка компьютера через 1 минуту!\"".format(formatted_pc_name)).wait()
+                else:
+                    print("shutdown /m \\\{} /r /f /t 60 /c \"Плановая перезагрузка компьютера через 1 минуту!\"".format(formatted_pc_name))
+    else:
+        print("Попробуйте еще раз выбрать область запуска скрипта!")
+
+
+if __name__ == "__main__":
+    if len(argv) < 3:
+        print("Укажите номер запускаемой области!\n2 - Кабинеты; 7 - Цех")
+        exit()
+    try:
+        filesio = FilesIO(default_values=default_values)
+        loaded_file = filesio.get_data()
+        if loaded_file is False:
+            raise OSError
+        matchio = MatchIO(loaded_file.get("expr_list"))
+        main(argv[1], argv[2].strip() == "prod")
+    except OSError:
+        print("Файл с настройками не найден. Создан файл со стандартными настройками.")
+        sleep(20)
+        exit()
+    except KeyboardInterrupt:
+        exit()