base: Integrate Data Objects better into SQL queries

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/buildservice/base.py b/src/buildservice/base.py
index cd979995c9a257aa751f166afc28008baa4b5aaf..13160fdc76363d152e531de9df22a9b7dbb03c60 100644 (file)
--- a/src/buildservice/base.py
+++ b/src/buildservice/base.py
@@ -1,5 +1,7 @@
 #!/usr/bin/python
 
+import psycopg2.extensions
+
 from .decorators import *
 
 class Object(object):
@@ -89,3 +91,12 @@ class DataObject(Object):
                # Update cached attribute
                if res:
                        self.data[key] = res[key]
+
+       # SQL Integration
+
+       def __conform__(self, proto):
+               if proto == psycopg2.extensions.ISQLQuote:
+                       return self
+
+       def getquoted(self):
+               return "(%d)" % self.id