From a148dedb99f72f6b5a4b55a06cb8b794d9741e85 Mon Sep 17 00:00:00 2001
From: cuiweizhi <560397@gree.com.cn>
Date: Sat, 4 Nov 2023 12:19:21 +0800
Subject: [PATCH] Update project

---
 application/common/config.py         |  8 +++++---
 application/libs/__init__.py         |  1 +
 application/libs/error/__init__.py   | 13 +++++++++++++
 application/libs/error/exceptions.py | 20 ++++++++++++++++++++
 4 files changed, 39 insertions(+), 3 deletions(-)
 create mode 100644 application/libs/error/__init__.py
 create mode 100644 application/libs/error/exceptions.py

diff --git a/application/common/config.py b/application/common/config.py
index 86c695c..6b463d5 100644
--- a/application/common/config.py
+++ b/application/common/config.py
@@ -16,6 +16,8 @@ from typing import Any
 from flask import Flask
 from loguru import logger
 
+from application.libs import ConfigKeyError
+
 
 class ConfigHelper:
     """
@@ -28,7 +30,7 @@ class ConfigHelper:
     def __init__(self, app: Flask):
         self.app = app
 
-    def __getattr__(self, name):
+    def __getattr__(self, name: str) -> Any:
         """
         Get a config value as an attribute.
 
@@ -38,8 +40,8 @@ class ConfigHelper:
         try:
             return self.app.config[name]
         except KeyError:
-            logger.error(f'Key {name} error')
-            raise KeyError(f'Key {name} error')
+            logger.error(f'Key {name} not found in configuration')
+            raise ConfigKeyError(name)
 
     def __repr__(self):
         return f"<ConfigHelper with app {self.app}>"
diff --git a/application/libs/__init__.py b/application/libs/__init__.py
index 620fe13..20aa673 100644
--- a/application/libs/__init__.py
+++ b/application/libs/__init__.py
@@ -11,5 +11,6 @@
 """
 
 from .config import ConsulConfig, LocalConfig
+from .error import ConfigKeyError
 from .flask_elasticsearch import FlaskElasticsearch
 from .flask_loguru import FlaskLoguru
diff --git a/application/libs/error/__init__.py b/application/libs/error/__init__.py
new file mode 100644
index 0000000..c3ac498
--- /dev/null
+++ b/application/libs/error/__init__.py
@@ -0,0 +1,13 @@
+#!/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 12:13
+# @File        : __init__.py
+# @Description : error 自定义模块
+"""
+
+from .exceptions import ConfigKeyError
diff --git a/application/libs/error/exceptions.py b/application/libs/error/exceptions.py
new file mode 100644
index 0000000..7caf33f
--- /dev/null
+++ b/application/libs/error/exceptions.py
@@ -0,0 +1,20 @@
+#!/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 12:14
+# @File        : exceptions.py
+# @Description :
+"""
+
+
+class ConfigKeyError(Exception):
+    """
+    An error that is raised when a configuration key is not found.
+    """
+    def __init__(self, key):
+        self.key = key
+        super().__init__(f'Key {self.key} not found in configuration')
-- 
GitLab