__init__.py 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. """
  2. utils/__init__.py
  3. 该模块初始化文件,导入了多个辅助工具类,并定义了一系列便捷函数,用于日志记录、配置管理、文件操作、字符串处理和邮件发送等功能。
  4. """
  5. import json
  6. from tools.utils.ai_helper import AiHelper
  7. from tools.utils.config_helper import ConfigHelper
  8. from tools.utils.logger_helper import LoggerHelper
  9. #import tools.utils.logger_helper as logger_helper
  10. def get_logger():
  11. """
  12. 获取日志记录器实例。
  13. 该函数通过调用LoggerHelper类的静态方法get_logger()来获取一个日志记录器实例。
  14. 主要用于需要记录日志的位置,通过该函数获取日志记录器实例,然后进行日志记录。
  15. 这样做可以保持日志记录的一致性和集中管理。
  16. :return: Logger实例,用于记录日志。
  17. """
  18. #return logger_helper.LoggerHelper.get_logger()
  19. return LoggerHelper.get_logger()
  20. def clean_log_file(day: int):
  21. """
  22. 清理指定天数之前的日志文件。
  23. :param day: 整数,表示清理多少天前的日志文件。
  24. """
  25. #logger_helper.LoggerHelper.clean_log_file(day)
  26. LoggerHelper.clean_log_file(day)
  27. def get_config():
  28. """
  29. 获取配置管理器实例。
  30. 该函数返回一个ConfigHelper实例,用于读取和管理应用程序的配置信息。
  31. :return: ConfigHelper实例,用于配置管理。
  32. """
  33. return ConfigHelper()
  34. def reload_config():
  35. """
  36. 重新加载配置文件。
  37. 该函数会重新加载配置文件中的内容,适用于配置文件发生更改后需要重新加载的情况。
  38. """
  39. get_config().load_config()
  40. def get_config_value(key: str, default: str = None):
  41. """
  42. 获取配置项的值。
  43. :param key: 字符串,配置项的键。
  44. :param default: 字符串,默认值(可选)。
  45. :return: 配置项的值,如果不存在则返回默认值。
  46. """
  47. return get_config().get(key, default)
  48. def get_config_int(key: str, default: int = None):
  49. """
  50. 获取配置项的整数值。
  51. :param key: 字符串,配置项的键。
  52. :param default: 整数,默认值(可选)。
  53. :return: 配置项的整数值,如果不存在则返回默认值。
  54. """
  55. return get_config().get_int(key, default)
  56. def get_config_bool(key: str):
  57. """
  58. 获取配置项的布尔值。
  59. :param key: 字符串,配置项的键。
  60. :return: 配置项的布尔值。
  61. """
  62. return get_config().get_bool(key)
  63. def call_openai(system_prompt: str, user_prompt: str) -> json:
  64. """
  65. 调用OpenAI API进行对话。
  66. :param system_prompt: 字符串,系统提示信息。
  67. :param user_prompt: 字符串,用户输入的提示信息。
  68. :return: JSON对象,API返回的结果。
  69. """
  70. return AiHelper().call_openai(system_prompt, user_prompt)