From d1b91790f529df108d556efec21dea36a8ff1355 Mon Sep 17 00:00:00 2001 From: Bernardo Damele Date: Thu, 17 Jan 2013 22:13:59 +0000 Subject: [PATCH] fixed --count on DB2 --- plugins/generic/databases.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/plugins/generic/databases.py b/plugins/generic/databases.py index 9d4956366..5f22ef80e 100644 --- a/plugins/generic/databases.py +++ b/plugins/generic/databases.py @@ -732,11 +732,11 @@ class Databases: return kb.data.cachedColumns def _tableGetCount(self, db, table): - if Backend.isDbms(DBMS.DB2): - query = "SELECT %s FROM %s.%s--" % (queries[Backend.getIdentifiedDbms()].count.query % '*', safeSQLIdentificatorNaming(db.upper()), safeSQLIdentificatorNaming(table.upper(), True)) - else: - query = "SELECT %s FROM %s.%s" % (queries[Backend.getIdentifiedDbms()].count.query % '*', safeSQLIdentificatorNaming(db), safeSQLIdentificatorNaming(table, True)) + if Backend.getIdentifiedDbms() in (DBMS.ORACLE, DBMS.DB2): + db = db.upper() + table = table.upper() + query = "SELECT %s FROM %s.%s" % (queries[Backend.getIdentifiedDbms()].count.query % '*', safeSQLIdentificatorNaming(db), safeSQLIdentificatorNaming(table, True)) count = inject.getValue(query, expected=EXPECTED.INT, charsetType=CHARSET_TYPE.DIGITS) if isNumPosStrValue(count):