parameter_update.py 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. import sys
  2. sys.path.append("../utils_dir/")
  3. sys.path.append("utils_dir/")
  4. from my_utils import parse_json_for_risk_rule
  5. from my_utils import parse_json_for_risk_videos
  6. from db_helper import RedisHelper
  7. from config import set_config
  8. config_ = set_config()
  9. RISK_SHIELD_FILTER_RULE_V1_JSON = "RISK_SHIELD_FILTER_RULE_V1_JSON"
  10. RISK_SHIELD_FILTER_VIDEO_V1_STR = "RISK_SHIELD_FILTER_VIDEO_V1_STR"
  11. RISK_SHIELD_FILTER_EXPANSION_FACTOR_INT = "RISK_SHIELD_FILTER_EXPANSION_FACTOR_INT"
  12. RISK_SHIELD_FILTER_FLAG_BOOL = "RISK_SHIELD_FILTER_FLAG_BOOL"
  13. def param_update_risk_rule() -> dict:
  14. """
  15. 定时更新风险过滤的规则
  16. key=RISK_SHIELD_FILTER_RULE_V1_JSON
  17. value= "{\"0\": [\"110000\"]}"
  18. """
  19. redis_helper = RedisHelper()
  20. tmp = redis_helper.get_data_from_redis(key_name=RISK_SHIELD_FILTER_RULE_V1_JSON)
  21. if tmp is not None:
  22. data = parse_json_for_risk_rule(tmp)
  23. return data
  24. return {}
  25. def param_update_risk_videos() -> list:
  26. """
  27. 定时更新风险过滤的videos
  28. key = "RISK_SHIELD_FILTER_VIDEO_V1_STR"
  29. value = "7536230,1,2,3,4,5,6,7,8,9,10"
  30. """
  31. redis_helper = RedisHelper()
  32. tmp = redis_helper.get_data_from_redis(key_name=RISK_SHIELD_FILTER_VIDEO_V1_STR)
  33. if tmp is not None:
  34. data = parse_json_for_risk_videos(tmp)
  35. return data
  36. return []
  37. def param_update_expansion_factor() -> int:
  38. """
  39. 定时更新扩大倍数
  40. key = "RISK_SHIELD_FILTER_EXPANSION_FACTOR_INT"
  41. value = "5"
  42. """
  43. redis_helper = RedisHelper()
  44. tmp = redis_helper.get_data_from_redis(key_name=RISK_SHIELD_FILTER_EXPANSION_FACTOR_INT)
  45. data = 5
  46. if tmp is not None:
  47. try:
  48. data = int(tmp)
  49. except Exception as e:
  50. data = 5
  51. # 容灾
  52. if data < 5:
  53. data = 5
  54. elif data > 15:
  55. data = 5
  56. return data
  57. def param_update_risk_filter_flag() -> bool:
  58. """
  59. 定时更新 是否过滤
  60. key = "RISK_SHIELD_FILTER_FLAG_BOOL"
  61. value = "False"
  62. """
  63. redis_helper = RedisHelper()
  64. tmp = redis_helper.get_data_from_redis(key_name=RISK_SHIELD_FILTER_FLAG_BOOL)
  65. data = False
  66. if tmp is not None:
  67. try:
  68. data = bool(tmp)
  69. except Exception as e:
  70. data = False
  71. return data
  72. if __name__ == '__main__':
  73. pass
  74. d1 = param_update_risk_rule()
  75. d2 = param_update_risk_videos()
  76. print(d1, type(d1))
  77. print(d2, type(d2))
  78. d3 = param_update_expansion_factor()
  79. d4 = param_update_risk_filter_flag()
  80. print(d3, type(d3))
  81. print(d4, type(d4))