redis_test.py 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. from utils import RedisHelper
  2. from config import set_config
  3. from log import Log
  4. config_, _ = set_config()
  5. log_ = Log()
  6. redis_helper = RedisHelper()
  7. if __name__ == '__main__':
  8. key = "RISK_SHIELD_FILTER_RULE_V1_JSON"
  9. value = "{\"0\": [\"110000\"]}"
  10. redis_helper.set_data_to_redis(key, value, 7200)
  11. value_get = redis_helper.get_data_from_redis(key_name=key)
  12. print('key:', key, type(key))
  13. print('value_get:', value_get, type(value_get))
  14. # key = "RISK_SHIELD_FILTER_VIDEO_V1_STR"
  15. # value = "7536230,1,2,3,4,5,6,7,8,9,10"
  16. # redis_helper.set_data_to_redis(key, value, 7200)
  17. # value_get = redis_helper.get_data_from_redis(key_name=key)
  18. # print('key:', key, type(key))
  19. # print('value_get:', value_get, type(value_get))
  20. key = "RISK_SHIELD_FILTER_EXPANSION_FACTOR_INT"
  21. value = "10"
  22. redis_helper.set_data_to_redis(key, value, 7200)
  23. value_get = redis_helper.get_data_from_redis(key_name=key)
  24. print('key:', key, type(key))
  25. print('value_get:', value_get, type(value_get))
  26. key = "RISK_SHIELD_FILTER_FLAG_BOOL"
  27. value = "True"
  28. redis_helper.set_data_to_redis(key, value, 7200)
  29. value_get = redis_helper.get_data_from_redis(key_name=key)
  30. print('key:', key, type(key))
  31. print('value_get:', value_get, type(value_get))
  32. # redis_helper.del_keys(key_name=key)
  33. # python zhangbo/redis_test.py
  34. # dt = '20231203'
  35. # data_key = 'test_lr_v1'
  36. # key_name = f"{config_.KEY_NAME_PREFIX_AD_GROUP}{data_key}:{dt}"
  37. # redis_data = {
  38. # 'yxh': 0.3,
  39. # 'zb': 0.13213231231,
  40. # }
  41. # redis_data = {
  42. # 'mz': 0.2,
  43. # 'zb': 0.6666,
  44. # }
  45. # print(key_name)
  46. # #redis_helper.add_data_with_zset(key_name=key_name, data=redis_data, expire_time=2 * 24 * 3600)
  47. #
  48. # print(redis_helper.get_score_with_value(key_name=key_name, value='zb'))
  49. # print(redis_helper.get_score_with_value(key_name=key_name, value='mz'))
  50. # print(redis_helper.get_score_with_value(key_name='ad:users:group:predict:share:rate:test_lr_v1:20231129', value='16537775'))
  51. #
  52. # model_key = 'ad_out_v1'
  53. # KEY_NAME_PREFIX_AD_OUT_MODEL_CONFIG = 'ad:out:model:config:'
  54. # abtest_id = '173'
  55. # abtest_config_tag = 'u'
  56. # config_key_prefix = f"{KEY_NAME_PREFIX_AD_OUT_MODEL_CONFIG}{model_key}:{abtest_id}:{abtest_config_tag}"
  57. # threshold_key = f"{config_key_prefix}:threshold"
  58. # use_mean_key = f"{config_key_prefix}:use_mean"
  59. # threshold = redis_helper.get_data_from_redis(key_name=threshold_key)
  60. # use_mean = redis_helper.get_data_from_redis(key_name=use_mean_key)
  61. # print('threshold:', threshold, type(threshold))
  62. # print('use_mean:', use_mean, type(use_mean))
  63. # expire_time = 30 * 24 * 3600
  64. # redis_helper.set_data_to_redis(threshold_key, 0.4, expire_time)
  65. # redis_helper.set_data_to_redis(use_mean_key, 1, expire_time)
  66. # print(threshold_key)
  67. # print(use_mean_key)