From 07b8ee96f2544c8aca15c2662495975d8c309261 Mon Sep 17 00:00:00 2001
From: RTAkland <markushammered@gmail.com>
Date: Wed, 15 Dec 2021 21:58:07 +0800
Subject: [PATCH] feat: settings.py

---
 core/settings.py | 68 +++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 67 insertions(+), 1 deletion(-)

diff --git a/core/settings.py b/core/settings.py
index d28a25e..59bc0ef 100644
--- a/core/settings.py
+++ b/core/settings.py
@@ -3,4 +3,70 @@
 # @Author: markushammered@gmail.com
 # @Development Tool: PyCharm
 # @Create Time: 2021/12/15
-# @File Name: settings.py
\ No newline at end of file
+# @File Name: settings.py
+
+import sys
+import time
+import getpass
+from core.logger import Logger
+from core.language import Language
+from core.read_excel import read_excel
+from ruamel.yaml import YAML
+
+
+def change_settings(mode: bool = False, _flag: bool = False):
+    """
+
+    :param _flag:
+    :param mode:
+    :return:
+    """
+    language = Language()
+
+    if not _flag or not mode:  # 如果配置中location未填写或status未False则触发条件
+        Logger.info(f'[Modify]{language["fill_the_config"]}')
+        Logger.info(f'[Modify]{language["input_a_city_name"]}')
+        while True:
+            time.sleep(0.3)
+            city_name = input('-->')
+            match city_name:
+                case 'q':
+                    Logger.info(f'[Modify]User quit.')
+                    sys.exit(0)
+                case '':
+                    Logger.critical(f'[Modify]{language["null_value"]}')
+                    continue
+                case _:
+                    break
+        searched_city = read_excel(city_name)
+        Logger.info(f'[Modify]{language["user_input"]}:[{city_name}]')
+        Logger.info(f'[Modify]{language["select_a_index"]}')
+        if not searched_city:
+            Logger.error(f'[Modify]{language["no_result"]}')
+            sys.exit(1)
+        time.sleep(0.3)
+        while True:
+            try:
+                time.sleep(0.3)
+                user_input = input('-->')
+                if user_input == 'q':
+                    Logger.info('[Quit]User quit')
+                    sys.exit(1)
+                index = searched_city[int(user_input)]
+                with open('./config.yml', 'r', encoding='utf-8') as of:
+                    data = YAML().load(of)
+                    data['request-settings']['location'] = index[1]
+                    data['only-view-settings']['city-name'] = f'{index[3]}-{index[7]}-{index[7]}'
+                    data['only-view-settings']['time'] = time.strftime("%a %b %d %Y %H:%M:%S", time.localtime())
+                    data['only-view-settings']['user'] = getpass.getuser()
+                with open('./config.yml', 'w', encoding='utf-8') as wf:
+                    YAML().dump(data, wf)
+                Logger.info(f'[Write]{language["write_successfully"]}:config.yml')
+                break
+            except (IndexError, ValueError) as e:
+                Logger.info(e)
+                Logger.error(f'[Write]{language["input_type_error"]}')
+                continue
+            finally:
+                Logger.info('[Done]Program has done.')
+                sys.exit(0)