Python sqllit实例

1、操作数据库

# -*- coding: UTF-8  -*-
'''
Created on 2010-6-7

@author: qxf
'''
import sys
import os
from sqlite3 import connect, Connection

import UpdateLog


class VisitSqlite(object):
    def __init__(self, dbname, tablename):
        self.dbname = dbname
        self.tablename = tablename
        self.__myInit()

    #表格初始化函数，若不存在，则新建表
    def __myInit(self):
        """
        如果数据库不存在，则创建数据库并新建数据库所需要的表
        """
        if not os.path.exists(self.dbname):
            connect(self.dbname)
            UpdateLog.log("DateBase has been created!")

            #下载任务表  TaskList
    #        strSql='create table TaskList 
    #        (gameID TEXT(64) primary key not null,\项目ID
    #        TaskState TEXT(64),\任务状态
    #        InstallMode TEXT(64),\安装方式 
    #        updateQz integer,\是否强制升级
    #        DownFileNumber integer,\文件数量
    #        MD5 TEXT(255),\MD5 值
    #        InstallPacketPath TEXT(255),\安装原路径
    #        InstallPacketTempPath TEXT(255),\安装临时路径
    #        DownOppPath TEXT(255),\下载的相对路径
    #        SetupFile TEXT(255),\安装包可执行文件路径
    #        szReleaseTime  TEXT(32),\发布时间
    #        iErroTimes integer,\安装失败次数
    #        szVestion TEXT(32),\版本信息
    #        rateOfLoad integer,\下载进度
    #        rateOfSetup integer,\安装进度
    #        IsReboot integer,\重启标志
    #        zipSize integer,\文件大小
    #        FileSize integer,\解压后大小
    #        dimRedu integer,\冗余量
    #        Flag1 TEXT(32),\预留字段1
    #        Flag2 TEXT(32),\预留字段2
    #        Flag3 TEXT(32) )预留字段3
    #        '''

            strSql='''create table TaskList 
            (id integer primary key autoincrement,
            gameID TEXT(64),\
            TaskState TEXT(64),\
            InstallMode TEXT(64),\
            updateQz integer,\
            DownFileNumber integer,\
            MD5 TEXT(255),\
            InstallPacketPath TEXT(255),\
            InstallPacketTempPath TEXT(255),\
            DownOppPath TEXT(255),\
            SetupFile TEXT(255),\
            szReleaseTime  TEXT(32),\
            iErroTimes integer,\
            szVestion TEXT(32),\
            rateOfLoad integer,\
            rateOfSetup integer,\
            IsReboot integer,\
            zipSize integer,\
            FileSize integer,\
            dimRedu integer,\
            Flag1 TEXT(32),\
            Flag2 TEXT(32),\
            Flag3 TEXT(32) )'''
            self.execute(strSql)


            #已安装项目信息表  InstallGameInfo
    #        '''
    #        (gameID TEXT(64) primary key not null,\项目ID
    #        gameName TEXT(64),\项目名称
    #        dispName TEXT(64),\显示名称
    #        lastVersion TEXT(64),\最新版本
    #        serVersion TEXT(64))"已安装版本
    #        '''
            strSql="create table InstallGameInfo \
            (gameID TEXT(64) primary key not null,\
            gameName TEXT(64),\
            dispName TEXT(64),\
            lastVersion TEXT(64),\
            serVersion TEXT(64))"
            self.execute(strSql)

            #任务执行情况表  WorkingInfo
            strSql="create table WorkingInfo \
            (ID integer primary key,\
            NumLoading integer,\
            NumSeting integer )"
            self.execute(strSql)

            #主版本号表
            strSql="create table IdongInfo (MainVersion TEXT(32) primary key, path TEXT(256))"
            self.execute(strSql)
            #写入初始主版本号
            strSql = "insert into IdongInfo(MainVersion, path) values('1.6.0', '')"
            self.execute(strSql)

    def execute(self, strSql):
        """
        此函数用户执行不带返回值的SQL语句

        strSql:执行的sql语句
        """
        try:
            dbc = Connection(self.dbname)
            dbc.execute(strSql)
            dbc.commit()
            dbc.close()
            return True
        except Exception, e:
            str = "execute '%s' exception with message '%s'" % (strSql, e)
            print str
            UpdateLog.log(str)
            return False

    def executeUpdate(self, strSql):
        """
        查询 flag==0 返回记录数；1返回查询结果
        fields表示要查询的字段集合 ,不为空则返回字段查询结果
        wsm 为条件查询语句 ，不为空，则返回条件查询结果（结果为字典列表）
        """
        try:
            dbc = Connection(self.dbname)
            cur = dbc.cursor()
            cur.execute(strSql)
            rowall = cur.fetchall()
            dbc.commit()
            dbc.close()
            #将结果转化成utf-8编码
            res = []
            for i in range(len(rowall)):
                tmp = []
                for j in range(len(rowall[i])):
                    if type(rowall[i][j]) == type(u"123"):
                        tmp.append(rowall[i][j].encode("utf-8"))
                    else:
                        tmp.append(rowall[i][j])
                res.append(tmp)
            return res

#            if flag==0:
#                return len(rowall)
#            else:
#                #根据字段列表和返回值生成 返回字典列表
#                if strfield=='':
#                    fields = self.getTableinfo(self.tablename)
#                else:
#                    fields = strfield.rsplit(',')
#                
##                print 'The Fields of ',tablename,'is:',fields            
#                rowlist=[]
#                for row in rowall:
#                    dict = {}
##                    print len(row),len(fields)
#                    for i in range(0,len(row)):
##                        print fields[i],row[i]
#                        dict[fields[i]] = row[i]
#                    rowlist.append(dict)
#                
##                print rowlist
#                return rowlist
#  
        except:
            str = "executeUpdate '%s' exception with message '%s'" % (strSql, sys.exc_info()[1][0])
            print str
            UpdateLog.log(str)
            return None

    def getTableinfo(self):
        """
        获取表的字段信息
        """
        try:
            dbc = Connection(self.dbname)
            cur = dbc.cursor()
            strsql = "select * from sqlite_master where type='table' and name='" + self.tablename+"'"
#            print strsql
            cur.execute(strsql)
            dbinfo = cur.fetchall()
            #print dbinfo
            dbc.commit()
            dbc.close()
        except Exception, e:
            str = "getTableinfo err with message %s" % e
            print str
            UpdateLog.log(str)
            return None
        res = self.__createsqlToFieldlist(dbinfo[0][4])
        #转化成utf-8
        for i in range(len(res)):
            res[i] = res[i].encode("utf-8")
        return res

    def __createsqlToFieldlist(self,createsql):
        """
        根据create sql 语句返回字段列表
        """
        strtemp = createsql
        list =[]
        ifind=strtemp.find('(')
        while ifind!=-1:
            strtemp = strtemp[ifind+1:].lstrip()
            list.append(strtemp[:strtemp.find(' ')])
            ifind = strtemp.find(',')
        return list

#    #删除某条记录
#    def deletedb(self,tablename,wsm=''):
#        log = open('log.txt','w')
#        try:
#            dbc = Connection(self.dbname)
#            if wsm=='':
#                dbc.execute("delete from "+tablename)
#            else:
#                dbc.execute("delete from "+tablename+' where '+wsm)
#            dbc.commit()
#            dbc.close()
#        except:
#            print sys.exc_info()
#            log.write(sys.exc_info()[1][0])
#            log.close()

#    #查询数据库表结构信息
#    #输入表名
#    #返回表字段列表

#        
#    def dropTable(self,tablename):
#        dbc = Connection(self.dbname)
#        dbc.execute('drop table '+tablename)
#        dbc.commit()
#        dbc.close()

#    #根据字段类型和值返回字符串
#    def fromtypetovalue(self,tablename,field,value):
#        try:
#            dbc = Connection(self.dbname)
#            dbc.execute(strSql)
#            dbc.commit()
#            dbc.close()
#        except:
#            print sys.exc_info()[1][0]


if __name__ == '__main__':
    UpdateLog.init_log("UpdateLog.log")

    vs = VisitSqlite('LanucherInfo.db', "WorkingInfo")
    print vs.getTableinfo()

2、简易sql控制台

# -*- coding: UTF-8  -*- 
import VisitSqlite

##getattr(TaskListDBFace)

class Console:
    @staticmethod 
    def start():

        while True:
            tip1 = "Please enter your sql to perform...\n"
            tip2 = "ERROR:your enter sql can not empry...\n"
            tip3 = "ERROR:your operate table is not exist...\n"
            executeErrorTip = 'Execute sql error,please check your excute sql...'
            executeSuccTip = 'Excute sql success...'
            sql = raw_input(tip1)
            if sql == '':
                print tip2
            else:
                if 'TaskList' in sql:
                    dbc = VisitSqlite.VisitSqlite('LanucherInfo.db', 'TaskList')
                elif 'InstallGameInfo' in sql:
                    dbc = VisitSqlite.VisitSqlite('LanucherInfo.db', 'InstallGameInfo')
                elif 'WorkingInfo' in sql:
                    dbc = VisitSqlite.VisitSqlite('LanucherInfo.db', 'WorkingInfo')
                else:
                    print tip3
                    continue
                tableInfo = dbc.getTableinfo()
                if ('select' in sql) or ('SELECT' in sql):
                    res = dbc.executeUpdate(sql)
                    if res is None:
                        print executeErrorTip
                    resDict = []
                    for row in res:
                        dict = {}
                        for i in range(0,len(row)):
                            dict[tableInfo[i]] = row[i]
                        resDict.append(dict)
                    if len(resDict) != 0:
                        print 'The query result is:\n'
                        print FormatUtil.format(resDict)
                    else:
                        print 'The query table is empty!'
                else:
                    ret = dbc.execute(sql)
                    if ret:
                        print executeSuccTip
                    else:
                        print executeErrorTip

class FormatUtil():
    '''
    format the select result
    '''
    @staticmethod 
    def format(res):
        '''
        the res is a list that contains dict
        '''
        resStr = ''
        topLine = ''
        nameStr = ''
        valuesStr = ''
        for record in res:
            keys = record.keys()
            values = record.values()
            strLine = ''
            valueStr = ''
            for key in keys:
                strLine += FormatUtil.printLine(20)

            nameStr = FormatUtil.printValue(keys)

            valueStr = FormatUtil.printValue(values)+'\n'+strLine

            valuesStr += valueStr+'\n'
        resStr = strLine+'\n'+nameStr+'\n'+strLine+'\n'+valuesStr
        return resStr

    @staticmethod         
    def printValue(valueList):
        '''
        '''
        resStr = ''
        spac = ''
        for i in range(0,len(valueList)):
            intSpac = 0
            if valueList[i] is not None:
                intSpac = len(str(valueList[i]))
            else:
                valueList[i] = ''
            for j in range(0,20-intSpac):
                spac += ' '
            resStr += str(valueList[i])+spac+'|'
            spac = ''
        return resStr 

    @staticmethod         
    def printLine(length):

        resStr = ''
        tems = ''
        for i in range(0,length):
            tems += '-'
        resStr += tems+'|'
        return resStr 



if __name__ == '__main__':
#    print FormatUtil.printCell('dfdf sdfs d')
#    print FormatUtil.format([{'name':'Dream','age':'12','job':'coder','school':'xinhua university'},{'name':'Dream2','age':'23','job':'coder2','school':'xinhua university2'}])
#    print FormatUtil.printLine('name')
#    print FormatUtil.printValue(['name','age','job'])
    Console.start()