add vc redis

vc.py

@@ -3,6 +3,7 @@ import json
 import time
 
 import pymysql
+import redis
 
 from common import request_to_jd_vc
 
@@ -231,10 +232,52 @@ def insert_vc_ext_attr(conf: dict):
             db.commit()
             print(insert_sql % (class_id, attr_id, attr_name, unit, enum_value, input_type, is_required, status, platform, create_time))
 
+def insert_ext_attr_to_redis(conf: dict):
+    host = conf["mysql"]["host"]
+    port = conf["mysql"]["port"]
+    user = conf["mysql"]["user"]
+    password = conf["mysql"]["password"]
+    database = conf["mysql"]["database"]
+    method_name = "jingdong.vc.item.extProps.find"
+    token = "845ce8478b074103b9e78a769d5fa4831y2u"
+    app_key = "CA52430E90209F51D8F5D7B615DDE9AD"
+    app_secret = "c92691b2379c48de87e699c4c2f7fb32"
+    select_sql = "select class_id from lie_shop_class where platform = 1"
+    db = pymysql.connect(host=host, port=port, user=user, password=password, database=database)
+    cursor = db.cursor()
+    cursor.execute(query=select_sql)
+    db_results = cursor.fetchall()
+    class_attr_mapping = {}
+    for i in range(len(db_results)):
+        class_id = db_results[i][0]
+        class_attr_mapping[class_id] = []
+        param = {
+            "category_leaf_id": class_id,
+        }
+        ans = request_to_jd_vc(method_name=method_name, token=token, app_key=app_key, app_secret=app_secret, param=param)
+        print("request to jd ....")
+        if "result" not in ans["jingdong_vc_item_extProps_find_responce"]["jos_result_dto"]:
+            continue
+        jd_results = ans["jingdong_vc_item_extProps_find_responce"]["jos_result_dto"]["result"]
+        for j in range(len(jd_results)):
+            result = jd_results[j]
+            attr_id = result["att_id"]
+            class_attr_mapping[class_id].append(attr_id)
+    redis_host = conf["redis"]["host"]
+    redis_port = conf["redis"]["port"]
+    redis_password = conf["redis"]["password"]
+    redis_db = redis.Redis(host=redis_host, port=redis_port, password=redis_password)
+    for class_id in class_attr_mapping:
+        key = str(class_id)
+        attr_ids = class_attr_mapping[class_id]
+        if len(attr_ids) == 0:
+            continue
+        value = json.dumps(attr_ids, ensure_ascii=False)
+        redis_db.hset(name="jd_ext_attr", key=key, value=value)
+        print("jd_ext_attr", key, value)
+
 
 
 if __name__ == '__main__':
-    conf = config_prod
-    insert_vc_attr(conf)
-    print("======================= start ext attr!!! ========================>")
-    insert_vc_ext_attr(conf)
+    conf = config_dev
+    insert_ext_attr_to_redis(conf)
\ No newline at end of file