time_check.py 1.9KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. import time,re,hashlib
  2. import config
  3. from common.log import logger
  4. def time_checker(f):
  5. def _time_checker(self, *args, **kwargs):
  6. _config = config.conf()
  7. chat_time_module = _config.get("chat_time_module", False)
  8. if chat_time_module:
  9. chat_start_time = _config.get("chat_start_time", "00:00")
  10. chat_stopt_time = _config.get("chat_stop_time", "24:00")
  11. time_regex = re.compile(r'^([01]?[0-9]|2[0-4])(:)([0-5][0-9])$') #时间匹配,包含24:00
  12. starttime_format_check = time_regex.match(chat_start_time) # 检查停止时间格式
  13. stoptime_format_check = time_regex.match(chat_stopt_time) # 检查停止时间格式
  14. chat_time_check = chat_start_time < chat_stopt_time # 确定启动时间<停止时间
  15. # 时间格式检查
  16. if not (starttime_format_check and stoptime_format_check and chat_time_check):
  17. logger.warn('时间格式不正确,请在config.json中修改您的CHAT_START_TIME/CHAT_STOP_TIME,否则可能会影响您正常使用,开始({})-结束({})'.format(starttime_format_check,stoptime_format_check))
  18. if chat_start_time>"23:59":
  19. logger.error('启动时间可能存在问题,请修改!')
  20. # 服务时间检查
  21. now_time = time.strftime("%H:%M", time.localtime())
  22. if chat_start_time <= now_time <= chat_stopt_time: # 服务时间内,正常返回回答
  23. f(self, *args, **kwargs)
  24. return None
  25. else:
  26. if args[0]['Content'] == "#更新配置": # 不在服务时间内也可以更新配置
  27. f(self, *args, **kwargs)
  28. else:
  29. logger.info('非服务时间内,不接受访问')
  30. return None
  31. else:
  32. f(self, *args, **kwargs) # 未开启时间模块则直接回答
  33. return _time_checker