73 lines
1.3 KiB
Python
73 lines
1.3 KiB
Python
#!/usr/bin/env python
|
|
# coding=utf8
|
|
|
|
"""
|
|
Created on 2018-8-16
|
|
|
|
@author: wd
|
|
|
|
@function:operate dameng database
|
|
|
|
@version: 1.0.0
|
|
"""
|
|
|
|
import dmPython
|
|
import sys
|
|
|
|
def login():
|
|
try:
|
|
conn = dmPython.connect(user='sysdba',password='123456',server='LOCALHOST',port=5236)
|
|
cursor = conn.cursor()
|
|
return cursor
|
|
except Exception as e:
|
|
print(e)
|
|
exit("error")
|
|
|
|
def logout(conn):
|
|
conn.close()
|
|
|
|
|
|
def create():
|
|
cursor = login()
|
|
drop(cursor)
|
|
cursor.execute("create table t01(c1 varchar(100), c2 varchar(100))") #创建表
|
|
for i in range(10):
|
|
cursor.execute("insert into t01 values('a'||%s, 'b'||%s)" %(i,i))
|
|
cursor.close()
|
|
return "success"
|
|
|
|
def query():
|
|
cursor = login()
|
|
try:
|
|
cursor.execute("select count(*) from t01") #查询表
|
|
count = cursor.fetchall()
|
|
except Exception as e:
|
|
print("error")
|
|
exit()
|
|
cursor.close()
|
|
return count[0][0]
|
|
|
|
def drop(cursor):
|
|
# cursor = login()
|
|
try:
|
|
cursor.execute("drop table t01") #删除表
|
|
return "success"
|
|
except Exception as e:
|
|
return "error"
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if sys.argv[1] == 'create' :
|
|
result = create()
|
|
print(result)
|
|
elif sys.argv[1] == 'query' :
|
|
count = query()
|
|
print(count)
|
|
elif sys.argv[1] == 'drop' :
|
|
cursor = login()
|
|
result = drop(cursor)
|
|
print(result)
|
|
else:
|
|
print("error")
|
|
|