提交 4bcb26a7 编写于 作者: S slguan

fix the problem that affectRows in python is incorrect

上级 62f9a90f
......@@ -31,6 +31,7 @@ class TDengineCursor(object):
self._block = None
self._block_rows = -1
self._block_iter = 0
self._affected_rows = 0
if connection is not None:
self._connection = connection
......@@ -66,6 +67,12 @@ class TDengineCursor(object):
"""
return self._rowcount
@property
def affected_rows(self):
"""Return the affected_rows of the object
"""
return self._affected_rows
def callproc(self, procname, *args):
"""Call a stored database procedure with the given name.
......@@ -105,6 +112,7 @@ class TDengineCursor(object):
res = CTaosInterface.query(self._connection._conn, stmt)
if res == 0:
if CTaosInterface.fieldsCount(self._connection._conn) == 0:
self._affected_rows += CTaosInterface.affectedRows(self._connection._conn)
return CTaosInterface.affectedRows(self._connection._conn)
else:
self._result, self._fields = CTaosInterface.useResult(self._connection._conn)
......@@ -167,6 +175,7 @@ class TDengineCursor(object):
self._block = None
self._block_rows = -1
self._block_iter = 0
self._affected_rows = 0
def _handle_result(self):
"""Handle the return result from query.
......@@ -175,4 +184,4 @@ class TDengineCursor(object):
for ele in self._fields:
self._description.append((ele['name'], ele['type'], None, None, None, None, False))
return self._result
\ No newline at end of file
return self._result
......@@ -31,6 +31,7 @@ class TDengineCursor(object):
self._block = None
self._block_rows = -1
self._block_iter = 0
self._affected_rows = 0
if connection is not None:
self._connection = connection
......@@ -66,6 +67,12 @@ class TDengineCursor(object):
"""
return self._rowcount
@property
def affected_rows(self):
"""Return the rowcount of insertion
"""
return self._affected_rows
def callproc(self, procname, *args):
"""Call a stored database procedure with the given name.
......@@ -105,6 +112,7 @@ class TDengineCursor(object):
res = CTaosInterface.query(self._connection._conn, stmt)
if res == 0:
if CTaosInterface.fieldsCount(self._connection._conn) == 0:
self._affected_rows += CTaosInterface.affectedRows(self._connection._conn)
return CTaosInterface.affectedRows(self._connection._conn)
else:
self._result, self._fields = CTaosInterface.useResult(self._connection._conn)
......@@ -167,6 +175,7 @@ class TDengineCursor(object):
self._block = None
self._block_rows = -1
self._block_iter = 0
self._affected_rows = 0
def _handle_result(self):
"""Handle the return result from query.
......@@ -175,4 +184,4 @@ class TDengineCursor(object):
for ele in self._fields:
self._description.append((ele['name'], ele['type'], None, None, None, None, False))
return self._result
\ No newline at end of file
return self._result
......@@ -31,6 +31,7 @@ class TDengineCursor(object):
self._block = None
self._block_rows = -1
self._block_iter = 0
self._affected_rows = 0
if connection is not None:
self._connection = connection
......@@ -66,6 +67,12 @@ class TDengineCursor(object):
"""
return self._rowcount
@property
def affected_rows(self):
"""Return the affected_rows of the object
"""
return self._affected_rows
def callproc(self, procname, *args):
"""Call a stored database procedure with the given name.
......@@ -105,6 +112,7 @@ class TDengineCursor(object):
res = CTaosInterface.query(self._connection._conn, stmt)
if res == 0:
if CTaosInterface.fieldsCount(self._connection._conn) == 0:
self._affected_rows += CTaosInterface.affectedRows(self._connection._conn)
return CTaosInterface.affectedRows(self._connection._conn)
else:
self._result, self._fields = CTaosInterface.useResult(self._connection._conn)
......@@ -167,7 +175,8 @@ class TDengineCursor(object):
self._block = None
self._block_rows = -1
self._block_iter = 0
self._affected_rows = 0
def _handle_result(self):
"""Handle the return result from query.
"""
......@@ -175,4 +184,4 @@ class TDengineCursor(object):
for ele in self._fields:
self._description.append((ele['name'], ele['type'], None, None, None, None, False))
return self._result
\ No newline at end of file
return self._result
......@@ -31,6 +31,7 @@ class TDengineCursor(object):
self._block = None
self._block_rows = -1
self._block_iter = 0
self._affected_rows = 0
if connection is not None:
self._connection = connection
......@@ -66,6 +67,12 @@ class TDengineCursor(object):
"""
return self._rowcount
@property
def affected_rows(self):
"""Return the affected_rows of the object
"""
return self._affected_rows
def callproc(self, procname, *args):
"""Call a stored database procedure with the given name.
......@@ -105,6 +112,7 @@ class TDengineCursor(object):
res = CTaosInterface.query(self._connection._conn, stmt)
if res == 0:
if CTaosInterface.fieldsCount(self._connection._conn) == 0:
self._affected_rows += CTaosInterface.affectedRows(self._connection._conn)
return CTaosInterface.affectedRows(self._connection._conn)
else:
self._result, self._fields = CTaosInterface.useResult(self._connection._conn)
......@@ -167,6 +175,7 @@ class TDengineCursor(object):
self._block = None
self._block_rows = -1
self._block_iter = 0
self._affected_rows = 0
def _handle_result(self):
"""Handle the return result from query.
......@@ -175,4 +184,4 @@ class TDengineCursor(object):
for ele in self._fields:
self._description.append((ele['name'], ele['type'], None, None, None, None, False))
return self._result
\ No newline at end of file
return self._result
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册