From 7805912f2990d78abeac024f241918a5692d8f78 Mon Sep 17 00:00:00 2001
From: winsert <winsert@163.com>
Date: Fri, 27 Mar 2020 13:25:11 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cbond/readcb.py | 128 +++++++++++-------------------------------------
 1 file changed, 28 insertions(+), 100 deletions(-)

diff --git a/cbond/readcb.py b/cbond/readcb.py
index 95ed291..73f2f81 100644
--- a/cbond/readcb.py
+++ b/cbond/readcb.py
@@ -11,16 +11,25 @@
 
 import sqlite3
 
-# 读出 code>0 所有可转债、可交换债的所有信息
-def readCB():
+# 读出指定 code 的所有可转债、可交换债的所有信息
+def readCB(code):
     cblists = []
-    conn = sqlite3.connect('cb.db')
-    curs = conn.cursor()
-    sql = "select * from cb where code > 0" 
-    curs.execute(sql)
-    tmp = curs.fetchall()
-    curs.close()
-    conn.close()
+    if code >= 0 :
+        conn = sqlite3.connect('cb.db')
+        curs = conn.cursor()
+        sql = "select * from cb where code = %r" %code
+        curs.execute(sql)
+        tmp = curs.fetchall()
+        curs.close()
+        conn.close()
+    else:
+        conn = sqlite3.connect('cb.db')
+        curs = conn.cursor()
+        sql = "select * from cb where code > 0"
+        curs.execute(sql)
+        tmp = curs.fetchall()
+        curs.close()
+        conn.close()
 
     for tlist in tmp:
         cblist = list(tlist) #将元组tuple转换成list
@@ -28,98 +37,17 @@ def readCB():
         
     return cblists
 
-# 读出 code=0 的可转债、可交换债的所有信息
-def readCB0():
-    cblists = []
-    conn = sqlite3.connect('cb.db')
-    curs = conn.cursor()
-    sql = "select * from cb where code=0" 
-    curs.execute(sql)
-    tmp = curs.fetchall()
-    curs.close()
-    conn.close()
-
-    for tlist in tmp:
-        cblist = list(tlist) #将元组tuple转换成list
-        cblists.append(cblist)
-        
-    return cblists
-
-# 读出 code=1 的可转债、可交换债的所有信息
-def readCB1():
-    cblists = []
-    conn = sqlite3.connect('cb.db')
-    curs = conn.cursor()
-    sql = "select * from cb where code=1" 
-    curs.execute(sql)
-    tmp = curs.fetchall()
-    curs.close()
-    conn.close()
-    
-    for tlist in tmp:
-        cblist = list(tlist) #将元组tuple转换成list
-        cblists.append(cblist)
-        
-    return cblists
-
-# 读出 code=2 的可转债、可交换债的所有信息
-def readCB2():
-    cblists = []
-    conn = sqlite3.connect('cb.db')
-    curs = conn.cursor()
-    sql = "select * from cb where code=2" 
-    curs.execute(sql)
-    tmp = curs.fetchall()
-    curs.close()
-    conn.close()
-
-    for tlist in tmp:
-        cblist = list(tlist) #将元组tuple转换成list
-        cblists.append(cblist)
-        
-    return cblists
-
-# 读出 code=3 的可转债、可交换债的所有信息
-def readCB3():
-    cblists = []
-    conn = sqlite3.connect('cb.db')
-    curs = conn.cursor()
-    sql = "select * from cb where code=3" 
-    curs.execute(sql)
-    tmp = curs.fetchall()
-    curs.close()
-    conn.close()
-
-    for tlist in tmp:
-        cblist = list(tlist) #将元组tuple转换成list
-        cblists.append(cblist)
-
-    return cblists
-
 if __name__ == '__main__':
 
-    list0 = readCB0()
-    print "\ncode = 0 共有：" + str(len(list0))
-    for i in list0:
-        print i[3],
-
-    list1 = readCB1()
-    print "\n\ncode = 1 共有：" + str(len(list1))
-    for i in list1:
-        print i[3],
-
-    list2 = readCB2()
-    print "\n\ncode = 2 共有：" + str(len(list2))
-    for i in list2:
-        print i[3],
+    print u"\n列表code =-1: 普通+关注+持仓"
+    print u"列表code = 0: 已强赎"
+    print u"列表code = 1: 普通"
+    print u"列表code = 2: 关注"
+    print u"列表code = 3: 持仓\n"
     
-    list3 = readCB3()
-    print "\n\ncode = 3 共有：" + str(len(list3))
-    for i in list3:
-        print i[3],
-
-    listNo0 = readCB()
-    print "\n\ncode > 0 共有：" + str(len(listNo0))
-    for i in listNo0:
+    code = int(raw_input("要查询的code?"))
+    cblists = readCB(code)
+    print "\ncode = " + str(code)+ u" 共有：" + str(len(cblists))
+    for i in cblists:
         print i[3],
-    print "\n"
\ No newline at end of file
+    print '\n'
\ No newline at end of file