新增sqlite工具类

tools/sqlite.py

@@ -0,0 +1,72 @@
+# encoding: utf-8
+__author__ = 'zhanghe'
+
+import sqlite3
+
+
+class SqLite(object):
+    def __init__(self, db_name):
+        self.conn = sqlite3.connect(db_name)
+
+    def close(self):
+        """
+        关闭Connection
+        """
+        self.conn.close()
+
+    def create(self):
+        """
+        创建数据表
+        """
+        # 创建一个Cursor:
+        cursor = self.conn.cursor()
+        # 执行一条SQL语句，创建user表:
+        cursor.execute('create table user (id varchar(20) primary key, name varchar(20))')
+        # 继续执行一条SQL语句，插入一条记录:
+        cursor.execute('insert into user (id, name) values (\'1\', \'Michael\')')
+        # 通过rowcount获得插入的行数:
+        print cursor.rowcount
+        # 关闭Cursor:
+        cursor.close()
+        # 提交事务:
+        self.conn.commit()
+
+    def show_tables(self):
+        """
+        显示数据库表名
+        """
+        cursor = self.conn.cursor()
+        cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
+        result = cursor.fetchall()
+        cursor.close()
+        print result
+        return result
+
+    def get_row(self):
+        """
+        获取多行数据
+        :return:
+        """
+        cursor = self.conn.cursor()
+        # 执行查询语句:
+        cursor.execute('select * from user where id=?', '1')
+        # 获得查询结果集:
+        values = cursor.fetchall()
+        cursor.close()
+        print values
+        return values
+
+
+def test():
+    """
+    测试
+    """
+    db = SqLite('test.db')
+    # db.create()
+    db.show_tables()
+    db.get_row()
+    db.close()
+
+
+if __name__ == '__main__':
+    test()