Update project

application/common/file.py

@@ -10,6 +10,10 @@
 # @Description :
 """
 
+from typing import Optional
+
+from application.constants import DEFAULT_CONFIG, ENV_CONFIG_MAP
+
 
 class FileHelper:
     """
@@ -19,20 +23,15 @@ class FileHelper:
     """
 
     @classmethod
-    def get_filename(cls, env: str) -> str:
+    def get_filename(cls, env: Optional[str]) -> str:
         """
         Get the configuration file name based on the environment.
 
         :param env: a string representing the environment.
-                    It should be 'PRODUCTION' or None.
+                    It should be one of the keys in ENV_CONFIG_MAP, or None.
         :return: a string representing the configuration file name.
-                 If env is 'PRODUCTION', return 'production_config.yaml'.
-                 Otherwise, return 'config.yaml'.
         """
-        if env == 'PRODUCTION':
-            return f'{env.lower()}_config.yaml'
-        else:
-            return 'config.yaml'
+        return ENV_CONFIG_MAP.get(env, DEFAULT_CONFIG)
 
     def __repr__(self):
         """

application/constants/__init__.py

+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+# @Version     : Python 3.11.4
+# @Software    : Sublime Text 4
+# @Author      : StudentCWZ
+# @Email       : StudentCWZ@outlook.com
+# @Date        : 2023/11/4 15:08
+# @File        : __init__.py.py
+# @Description :
+"""
+
+from .constants import DEFAULT_CONFIG, ENV_CONFIG_MAP

application/constants/constants.py

+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+# @Version     : Python 3.11.4
+# @Software    : Sublime Text 4
+# @Author      : StudentCWZ
+# @Email       : StudentCWZ@outlook.com
+# @Date        : 2023/11/4 15:08
+# @File        : constants.py
+# @Description :
+"""
+
+DEFAULT_CONFIG = 'config.yaml'
+
+ENV_CONFIG_MAP = {
+    'PRODUCTION': 'production_config.yaml',
+    'DEVELOPMENT': 'config.yaml',
+    # Add more environments here if necessary
+}