SimRecall/import_redist.py

#coding utf-8
import sys
import json
from db_help import RedisHelper

if __name__=="__main__":
     f2 = open(sys.argv[2], 'w')
     with  open(sys.argv[1]) as f:
         rec_json_list=json.load(f)
         #print(rec_json_list)
         import_data_dict = {}
         for line in rec_json_list:
             #print(line)
             rec_list = line[1][:10]
             rec_item_list = []
             for rec_item in rec_list:
                 rec_item_list.append((rec_item[0], round(rec_item[1],3)))
             res_info = json.dumps(rec_item_list)
             
             f2.write(str(line[0])+"\t"+res_info+"\n")
             key="sim_hot_"+str(line[0])
             import_data_dict[key] = res_info
         redis_helper = RedisHelper()
         redis_helper.update_batch_setnx_key(import_data_dict, 60*60*24*7)
         #con = redis_helper.connect()
         res = redis_helper.get_data_from_redis("sim_hot_14330133")
         print(res)
         f2.close()
     f.close()