python和sqlserver_python连接sqlserver和MySQL实现增删改查

#python 3.6#统一输出函数，使用distutils.log.warn()函数代替版本2中的print语句和版本3中的print函数

from distutils.log importwarn as printffrom random importrandrange#检测内置函数中是否包括raw_input()，如果存在说明版本是2（或者1），不存在说明是3。这样的话，无论是哪个版本，最终scanf函数表示了该功能

if isinstance(__builtins__,dict) and 'raw_input' in __builtins__:

scanf=raw_inputelif hasattr(__builtins__,'raw_input'):

scanf=raw_inputelse:

scanf=input

FIELDS= ('login','userid','projid')

COLSIZ= 10RDBMSs= {'s':'sql server','m':'mysql'}

DBNAME= 'test'DB_EXC=None

DBUSER= 'root'DBSERVER= '127.0.0.1'NAMELEN= 10tformat= lambda s: str(s).title().ljust(COLSIZ)#str().title()将单词首字母大写；ljust()左对齐，并使用空格填充直至达到指定长度

cformat = lambda s: s.upper().ljust(COLSIZ) #

defsetup():return RDBMSs[scanf('''请选择数据库类型：

(Sql) Server

(M)ySQL

请键入：''').strip().lower()[0]]defconnect(db,DBNAME):globalDB_EXC

dbDir= '%s_%s' %(db,DBNAME)if db == 'mysql':try:#python3.0以前的版本

importMySQLdbimport_mysql_exceptions as DB_EXCtry:

cxn= MySQLdb.connect(user = DBUSER,password = 'Jwxjs123456',db=DBNAME)exceptDB_EXC.OperationalError:try:

cxn= MySQLdb.connect(user = DBUSER,password = 'Jwxjs123456',db=DBNAME)

cxn.query('create database %s' %DBNAME)

cxn.commit()

cxn.close()

cxn= MySQLdb.connect(db=DBNAME)exceptDB_EXC.OperationalError:returnNoneexcept ModuleNotFoundError: #ImportError

try:#python3.0以上的版本

importpymysqlimportpymysql.err as DB_EXCtry:

cxn= pymysql.connect(**{'host':DBSERVER,'database':DBNAME,'user':DBUSER,'password':'Jwxjs123456'})exceptDB_EXC.InternalError:#连接成功但是没有指定的数据库

cxn = pymysql.connect(**{'host':DBSERVER,'user':DBUSER,'password':'Jwxjs123456'})

cur=cxn.cursor()

cur.execute('create database %s;' %DBNAME)

cxn.commit()

cxn= pymysql.connect(**{'host':DBSERVER,'database':DBNAME,'user':DBUSER,'password':'Jwxjs123456'})exceptImportError:returnNoneelif db == 'sql server':importpymssqlimport_mssql#import pymssql.StandardError as DB_EXC

try:

cxn= pymssql.connect(**{'server':DBSERVER,'database':DBNAME,'password':'Jwxjs123456','user':'sa'})

printf('成功链接数据库%s' %DBNAME)exceptBaseException:#这里一定要设置为自动提交模式，否则创建数据库会失败

cxn = pymssql.connect(**{'server':DBSERVER,'password':'Jwxjs123456','user':'sa','autocommit':True})

cursor=cxn.cursor()

cxn.commit()

cursor.execute('create database %s;' %DBNAME)#cxn.close()

cxn = pymssql.connect(**{'server':DBSERVER,'database':DBNAME,'password':'Jwxjs123456','user':'sa','autocommit':True})

printf('数据库%s 不存在，新建该数据库' %DBNAME)returncxndefcreate(cur):try:

cur.execute('''create table users(login varchar(%s),

userid int,

projid int

)''' %NAMELEN)

printf('……………新建users表成功')exceptBaseException:

drop(cur)

create(cur)

printf('已存在表users,删除后并新建此表……………………')

drop= lambda cur:cur.execute('drop table users')

NAMES=(

('arron',8312),('angle',7603),('dane',7306),

('jess',7912),('jim',7512),('larry',7311),

)defrandName():

pick= set(NAMES)#内置函数set()创建一个无序不重复元素集，可进行关系测试，删除重复数据，还可以计算交集、差集、并集等

whilepick:yieldpick.pop()definsert(cur,db):

cur.executemany('insert into users values(%s,%s,%s)',[(who,uid,randrange(1,5)) for who,uid inrandName()])#返回最后一次操作影响的行数，这里考虑右边对象是不是支持该属性，不支持的话则返回-1

getRC = lambda cur:cur.rowcount if hasattr(cur,'rowcount') else -1

defupdate(cur):

fr= randrange(1,5)

to= randrange(1,5)

cur.execute('update users set projid = %s where projid = %s' %(to,fr))returnfr,to,getRC(cur)defdbDump(cur):

cur.execute('select * from users')

printf('\n%s' % ''.join(map(cformat,FIELDS)))for data incur.fetchall():

printf(''.join(map(tformat,data)))#join()将序列中的元素以指定的字符连接生成一个新的字符串。

defdelete(cur):

rm= randrange(1,5)

cur.execute('delete from users where projid = %s' %rm)returnrm,getRC(cur)

drop= lambda cur:cur.execute('drop table users')defmain():#用户选择数据库类型

db =setup()

printf('\n****将要连接到 %s(%s) 数据库****' %(DBNAME,db))#创建数据库连接

cxn =connect(db,DBNAME)if notcxn:

printf('\n 连接数据库%s(%s) 失败，程序退出!!!' %(DBNAME,db))return

else:

printf('\n 成功连接数据库%s(%s)' %(DBNAME,db))

cur=cxn.cursor()

printf('\n 创建 users表……')

create(cur)

printf('\n*****初始化users表******')

insert(cur,db)

dbDump(cur)

printf('\n ***随机更改projid****')

fr,to,num=update(cur)

printf('\t(%s 个用户被更改了) from (%s) to(%s)' %(num,fr,to))

dbDump(cur)

printf('\n***随机删除某一组数据***')

rm,num=delete(cur)

printf('\t(组%s中共%s个用户被删除了)' %(rm,num))

dbDump(cur)

printf('\n*****删除users表*******')

drop(cur)if db == 'mysql':

cxn.commit()#sqlserver 的链接已经被设置为自动提交

printf('\n 关闭数据库连接')

cxn.close()if __name__ == '__main__':

main()