add python_redis.py

xianhu · xianhu · commit fab50b24eea0 · 2019-07-20T22:22:51.000+08:00
diff --git a/README.md b/README.md
@@ -58,6 +58,8 @@
 
 ### python_mail.py: &#20351;&#29992;Python&#33258;&#21160;&#21457;&#36865;&#37038;&#20214;&#65292;&#21253;&#25324;&#21457;&#36865;HTML&#20197;&#21450;&#22270;&#29255;&#12289;&#38468;&#20214;&#31561;
 
+### python_redis.py: Python&#25805;&#20316;Redis&#23454;&#29616;&#28040;&#24687;&#30340;&#21457;&#24067;&#19982;&#35746;&#38405;
+
 ### Plotly&#30446;&#24405;: &#19968;&#20123;plotly&#30011;&#22270;&#30340;&#23454;&#20363;&#65292;&#20351;&#29992;jupyter notebook&#32534;&#20889;
 ===================================================================================================
 
diff --git a/python_redis.py b/python_redis.py
@@ -0,0 +1,55 @@
+# _*_ coding: utf-8 _*_
+
+"""
+Python&#25805;&#20316;Redis&#23454;&#29616;&#28040;&#24687;&#30340;&#21457;&#24067;&#19982;&#35746;&#38405;
+"""
+
+import sys
+import time
+import redis
+
+# &#20840;&#23616;&#21464;&#37327;
+conn_pool = redis.ConnectionPool(host="localhost", port=6379, db=1)
+conn_inst = redis.Redis(connection_pool=conn_pool)
+channel_name = "fm-101.1"
+
+
+def public_test():
+    while True:
+        # &#21457;&#24067;&#28040;&#24687;
+        conn_inst.publish(channel_name, "hello " + str(time.time()))
+        if int(time.time()) % 10 == 1:
+            conn_inst.publish(channel_name, "over")
+        time.sleep(1)
+
+
+def subscribe_test(_type=0):
+    pub = conn_inst.pubsub()
+    pub.subscribe(channel_name)
+
+    if _type == 0:
+        # &#35746;&#38405;&#28040;&#24687;
+        for item in pub.listen():
+            print("Listen on channel: %s" % item)
+            if item["type"] == "message" and item["data"].decode() == "over":
+                print(item["channel"].decode(), "&#24050;&#20572;&#27490;&#21457;&#24067;")
+                break
+    else:
+        # &#21478;&#19968;&#31181;&#35746;&#38405;&#27169;&#24335;
+        while True:
+            item = pub.parse_response()
+            print("Listen on channel: %s" % item)
+            if item[0].decode() == "message" and item[2].decode() == "over":
+                print(item[1].decode(), "&#24050;&#20572;&#27490;&#21457;&#24067;")
+                break
+
+    # &#21462;&#28040;&#35746;&#38405;
+    pub.unsubscribe()
+    return
+
+
+if __name__ == '__main__':
+    if sys.argv[1] == "public":
+        public_test()
+    else:
+        subscribe_test(int(sys.argv[2]))

-Original file line number
+Diff line change
 ### python_mail.py: 使用Python自动发送邮件，包括发送HTML以及图片、附件等
 +### python_redis.py: Python操作Redis实现消息的发布与订阅
++
 ### Plotly目录: 一些plotly画图的实例，使用jupyter notebook编写
 ===================================================================================================