def testInsert(self): adao = AddressDAO() address = Address() address.ip = self.ip address.port = self.port aid = adao.insertOrUpdate(address) assert aid != None, 'error inserting address'
def find(self, id): query = "SELECT * FROM %s WHERE id = %s" % (self.tablename, id) self.cursor.execute(query) rs = self.cursor.fetchall() if not rs: return None address = Address() for row in rs: address.id = row[0] address.ip = row[1] address.port = row[2] return address
def findByIpPort(self, ip, port): query = "SELECT * FROM %s WHERE ip = '%s' AND port = %s" % (self.tablename, ip, port) logging.debug(query) self.cursor.execute(query) rs = self.cursor.fetchall() if not rs: return None address = Address() for row in rs: address.id = row[0] address.ip = row[1] address.port = row[2] return address