改善循环退出机制

zhanghe06 · zhanghe06 · commit 8a2a4d5a5267 · 2016-01-01T01:36:29.000+08:00
diff --git a/test/test_loop.py b/test/test_loop.py
@@ -4,6 +4,10 @@
 """
 __author__ = 'zhanghe'
 
+
+import time
+
+
 friend_dict = {
     # 'QQ&#21495;': '&#22995;&#21517;',
     'A_f': 'info_a',
@@ -27,20 +31,13 @@ def show_list(dict_info):
 
 
 def get_msg():
-    print '&#26368;&#26032;&#28040;&#24687;'
+    print '&#26368;&#26032;&#28040;&#24687;[%s]' % time.ctime()
+    time.sleep(2)
 
 
 msg_type = None
 while 1:
     try:
-        if msg_type in ['?', '&#65311;', '', '#', None]:  # ? &#30452;&#25509;&#22238;&#36710; &#22238;&#21040;&#33756;&#21333; &#21021;&#22987;&#36827;&#20837; &#26174;&#31034;&#24110;&#21161;
-            print '\n-------------'
-            print '0&#12289;&#25509;&#25910;&#26368;&#26032;&#28040;&#24687;'
-            print '1&#12289;&#21457;&#36865;&#22909;&#21451;&#28040;&#24687;'
-            print '2&#12289;&#21457;&#36865;&#32676;&#32452;&#28040;&#24687;'
-            print 'q&#12289;&#36864;&#20986;&#31243;&#24207;'
-            msg_type = raw_input('&#35831;&#36873;&#25321;&#31867;&#22411;: \n')
-            continue
         if msg_type == '0':  # &#25509;&#25910;&#26368;&#26032;&#28040;&#24687;
             get_msg()
             continue
@@ -61,8 +58,23 @@ def get_msg():
                 print raw_input_msg
             continue
         if msg_type == 'q':
-            print '&#31243;&#24207;&#24050;&#36864;&#20986;'
-            break
+            print '&#30830;&#35748;&#36864;&#20986;&#65288;Y&#65289;'
+            print '&#21462;&#28040;&#25805;&#20316;&#65288;N&#65289;'
+            raw_input_msg = raw_input("&#31243;&#24207;&#21363;&#23558;&#36864;&#20986;: \n")
+            if raw_input_msg in ['y', 'Y']:
+                print '&#31243;&#24207;&#24050;&#36864;&#20986;'
+                break
+            else:
+                msg_type = ''
+                continue
+        else:  # ? &#30452;&#25509;&#22238;&#36710; &#22238;&#21040;&#33756;&#21333; &#21021;&#22987;&#36827;&#20837; &#26174;&#31034;&#24110;&#21161;
+            print '\n-------------'
+            print '0&#12289;&#25509;&#25910;&#26368;&#26032;&#28040;&#24687;'
+            print '1&#12289;&#21457;&#36865;&#22909;&#21451;&#28040;&#24687;'
+            print '2&#12289;&#21457;&#36865;&#32676;&#32452;&#28040;&#24687;'
+            print 'q&#12289;&#36864;&#20986;&#31243;&#24207;'
+            msg_type = raw_input('&#35831;&#36873;&#25321;&#31867;&#22411;: \n')
+            continue
     except KeyboardInterrupt:
         print '&#31243;&#24207;&#24050;&#36864;&#20986;'
         break

-Original file line number
+Diff line change
 """
 __author__ = 'zhanghe'
++
 +import time
++
++
 friend_dict = {
     # 'QQ号': '姓名',
     'A_f': 'info_a',
 def get_msg():
 -    print '最新消息'
 +    print '最新消息[%s]' % time.ctime()
 +    time.sleep(2)
 msg_type = None
 while 1:
     try:
 -        if msg_type in ['?', '？', '', '#', None]:  # ? 直接回车 回到菜单 初始进入 显示帮助
 -            print '\n-------------'
 -            print '0、接收最新消息'
 -            print '1、发送好友消息'
 -            print '2、发送群组消息'
 -            print 'q、退出程序'
 -            msg_type = raw_input('请选择类型: \n')
 -            continue
         if msg_type == '0':  # 接收最新消息
             get_msg()
             continue
                 print raw_input_msg
             continue
         if msg_type == 'q':
 -            print '程序已退出'
 -            break
 +            print '确认退出（Y）'
 +            print '取消操作（N）'
 +            raw_input_msg = raw_input("程序即将退出: \n")
 +            if raw_input_msg in ['y', 'Y']:
 +                print '程序已退出'
 +                break
 +            else:
 +                msg_type = ''
 +                continue
 +        else:  # ? 直接回车 回到菜单 初始进入 显示帮助
 +            print '\n-------------'
 +            print '0、接收最新消息'
 +            print '1、发送好友消息'
 +            print '2、发送群组消息'
 +            print 'q、退出程序'
 +            msg_type = raw_input('请选择类型: \n')
 +            continue
     except KeyboardInterrupt:
         print '程序已退出'
         break