#!/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")