Fixed the return type of update_host

2012-09-27 13:05:42 +10:00 · 2012-09-27 13:05:42 +10:00 · 82c56de835
commit 82c56de835
parent d3ffdf5a3e
1 changed files with 2 additions and 2 deletions
--- a/neat/db.py
+++ b/neat/db.py
@ -144,13 +144,13 @@ class Database(object):
                ram=ram).inserted_primary_key[0]
            log.info('Created a new DB record for a host %s, id=%d',
                     hostname, id)
-            return id
+            return int(id)
        else:
            self.connection.execute(self.hosts.update().
                                    where(self.hosts.c.id == row['id']).
                                    values(cpu_mhz=cpu_mhz,
                                           ram=ram))
-            return row['id']
+            return int(row['id'])

    @contract
    def select_host_characteristics(self):