Commit 3df1721e authored by 张彦钊's avatar 张彦钊

change test file

parent 53d1971d
......@@ -77,15 +77,16 @@ def tractate_write(device_id, cid):
cursor.execute(sql)
result = cursor.fetchall()
db.close()
cids = [i[0] for i in result]
if len(cids) != 0:
if len(result) > 0:
cids = [i[0] for i in result]
r = redis.Redis(host="172.16.40.135", port=5379, password="",db = 2)
key = str(device_id) + "_dislike_tractate"
if r.exists(key):
value = json.loads(r.get(key).decode('utf-8'))
value.extend(cids)
cids = json.dumps(value)
cids = json.dumps(list(set(value)))
r.set(key, cids)
print("cunza")
......@@ -117,15 +118,16 @@ def answer_write(device_id,cid):
cursor.execute(sql)
result = cursor.fetchall()
db.close()
cids = [i[0] for i in result]
if len(cids) != 0:
if len(result) > 0:
cids = [i[0] for i in result]
r = redis.Redis(host="172.16.40.135", port=5379, password="", db=2)
key = str(device_id) + "_dislike_answer"
if r.exists(key):
value = json.loads(r.get(key).decode('utf-8'))
value.extend(cids)
cids = json.dumps(value)
cids = json.dumps(list(set(value)))
r.set(key, cids)
print("cunza")
......@@ -158,15 +160,15 @@ def question_write(device_id,cid):
cursor.execute(sql)
result = cursor.fetchall()
db.close()
cids = [i[0] for i in result]
if len(cids) != 0:
if len(result) > 0:
cids = [i[0] for i in result]
r = redis.Redis(host="172.16.40.135", port=5379, password="", db=2)
key = str(device_id) + "_dislike_question"
if r.exists(key):
value = json.loads(r.get(key).decode('utf-8'))
value.extend(cids)
cids = json.dumps(value)
cids = json.dumps(list(set(value)))
r.set(key, cids)
print("cunza")
......@@ -199,34 +201,30 @@ def diary_write(device_id,cid):
cursor.execute(sql)
result = cursor.fetchall()
db.close()
cids = [i[0] for i in result]
if len(cids) != 0:
if len(result) > 0:
cids = [i[0] for i in result]
r = redis.Redis(host="172.16.40.135", port=5379, password="", db=2)
key = str(device_id) + "_dislike_diary"
if r.exists(key):
value = json.loads(r.get(key).decode('utf-8'))
value.extend(cids)
cids = json.dumps(value)
cids = json.dumps(list(set(value)))
r.set(key, cids)
print("cunza")
else:
r.set(key, json.dumps(cids))
r.expire(key, 7 * 24 * 60 * 60)
print("bucunza")
return "diary good"
except Exception as e:
print("diary insert redis fail")
print(e)
# sc = SparkContext(conf=SparkConf().setMaster("spark://nvwa01:7077").setAppName("filter").set("spark.io.compression.codec", "lzf"))
# sc = SparkContext(conf=SparkConf().setMaster("spark://nvwa01:7077").setAppName("dislike").set("spark.io.compression.codec", "lzf"))
# ssc = StreamingContext(sc,4)
# sc.setLogLevel("WARN")
# kafkaParams = {"metadata.broker.list": "172.16.44.25:9092,172.16.44.31:9092,172.16.44.45:9092",
# "group.id": "filter",
# "group.id": "dislike",
# "socket.timeout.ms": "600000",
# "auto.offset.reset": "largest"}
#
......@@ -238,8 +236,9 @@ def diary_write(device_id,cid):
# ssc.start()
# ssc.awaitTermination()
question_write("hello",176661)
question_write("hello",226749)
diary_write("9C5E7C73-380C-4623-8F48-A64C8034E315",16952842)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment