Moved files around to simplify deployment

1 files changed, 305 insertions, 0 deletions

diff --git a/src/collagen/matchbox/db/__init__.py b/src/collagen/matchbox/db/__init__.py
new file mode 100644
index 0000000..6dca0b1
--- /dev/null
+++ b/src/collagen/matchbox/db/__init__.py
@@ -0,0 +1,305 @@
+import sys
+import os
+import settings as config
+from django.conf import settings
+os.environ["DJANGO_SETTINGS_MODULE"]="db_config"
+
+
+
+from main.models import *
+from django.db import reset_queries, close_connection, _rollback_on_exception
+
+class DbIface(object):
+    """
+    Class representing interface to database for Matchbox
+
+    All of these functions have to be implemented in subclasses
+    for Matchbox to work correcly
+    """
+
+    def __init__(self, dbcon):
+        raise NotImplementedError("DbIface is only interface, create real object for database functions")
+
+    def add_package(self, name):
+        """
+        @param name: name of package to add without category or version (e.g. kdevelop)
+        @type name: string
+
+        @rtype: integer
+        @returns: id of package added to database
+        """
+        pass
+
+    def add_category(self, name):
+        """
+        @param name: name of category to add
+        @type name: string
+
+        @rtype: integer
+        @returns: id of category added to database
+        """
+        pass
+
+    def add_package_version(self, package_id, category_id, version):
+        """
+        @param package_id: id of row in package table
+        @type package_id: integer
+        @param category_id: id of row in category table
+        @type category_id: integer
+        @param version: version string to add (including revision)
+        @type version: string
+
+        @rtype: integer
+        @returns: id of package version added
+        """
+        pass
+
+    def add_dependency(self, packageversion_id, dependency_id):
+        """
+        Adds dependency that package was compiled with
+
+        @param packageversion_id: id of row in packageversion table
+        @type packageversion_id: integer
+        @param dependency_id: id of row in packageversion table
+        @type dependency_id: integer
+        """
+        pass
+
+    def add_package_info(self, pi):
+        """
+        @param pi: package info to add to database
+        @type pi: protocol.PackageInfo
+
+        @rtype: integer
+        @returns: id of package info added to database
+        """
+        pass
+
+    def add_tinderbox(self, ip):
+        """
+        @param ip: ip address of tinderbox slave
+        @type ip: string in form of xxx.xxx.xxx.xxx
+        """
+        pass
+
+    def add_attachment(self, packageproperties_id, name, content, mimetype):
+        """
+        @param packageproperties_id: id of row in packageproperties table
+        @type packageproperties_id: integer
+        @param name: name of attachment (usually filename)
+        @type name: string
+        @param content: data of attachment
+        @type content: string blob
+        @param mimetype: mime-type of attachment (e.g. text/html, text/plain etc)
+        @type mimetype: string
+        """
+        pass
+
+    def add_portage_profile(self, name):
+        """
+        @param name: name of portage profile to add
+        @type name: string
+
+        @rtype: integer
+        @returns: row id of profile added to database
+        """
+        pass
+
+
+    def add_useflag(self, name):
+        """
+        @param name: name of use flag to add
+        @type name: string
+
+        @rtype: integer
+        @returns: row id of use flag added to database
+        """
+        pass
+
+    def add_packageproperties(self, packageversion_id, profile_id, tinderbox_id, error_code):
+        """
+
+        @param packageversion_id: id of row in packageversion table
+        @type packageversion_id: integer
+        @param profile_id: id of row in portageprofile table
+        @type profile_id: integer
+        @param tinderbox_id: id of row in tinderbox_id
+        @type tinderbox_id: integer
+        @param error_code: error code for this compile try
+        @type error_code: integer
+
+        @returns: row id of packageproperties added to database
+        @rtype: integer
+        """
+        pass
+
+    def add_useflags_to_packageproperies(self, packageproperties_id, useflag_ids):
+        """
+        @param packageproperties_id: id of row in packageproperies table
+        @type packageproperties_id: integer
+        @param useflag_ids: list of row ids in useflag table
+        @type useflag_ids: list
+        """
+        pass
+
+    def add_contents_to_packageproperties(self, packageproperties_id, contents):
+        """
+        @param packageproperties_id: id of row in packageproperties table
+        @type packageproperties_id: integer
+        @param contents: dict of files in packageproperties (path is key, value is list of information about path)
+                         see dblink.getcontents() for more info
+        @type contents: dict
+        """
+        pass
+
+class DjangoDB(object):
+
+    def dbquery(f):
+        def newfunc(*args, **kwargs):
+            reset_queries()
+            try:
+                return f(*args, **kwargs)
+            except Exception, e:
+                _rollback_on_exception()
+                raise e
+        return newfunc
+
+    @dbquery
+    def add_package(self, name):
+        p = Package.objects.filter(name=name)
+        if len(p) > 0:
+            return p[0].id
+        p = Package(name=name)
+        p.save()
+        return p.id
+
+
+    @dbquery
+    def add_category(self, name):
+        c = PackageCategory.objects.filter(name=name)
+        if len(c) > 0:
+            return c[0].id
+        c = PackageCategory(name=name)
+        c.save()
+        return c.id
+
+    @dbquery
+    def add_package_version(self, package_id, category_id, version):
+        p = Package.objects.get(pk=package_id)
+        c = PackageCategory.objects.get(pk=category_id)
+
+        v = PackageVersion.objects.filter(version=version, category=c, package=p)
+        if len(v) > 0:
+            return v[0].id
+        v = PackageVersion(version=version, package=p, category=c)
+        v.save()
+        return v.id
+
+    @dbquery
+    def get_package_version(self, package_id, category_id, version):
+        p = Package.objects.get(pk=package_id)
+        c = PackageCategory.objects.get(pk=category_id)
+
+        v = PackageVersion.objects.filter(version=version, category=c, package=p)
+        if len(v) > 0:
+            return v[0].id
+        return None
+
+    @dbquery
+    def add_dependency(self, packageversion_id, dependency_id):
+        p = Package.objects.get(pk=packageversion_id)
+        dep = Package.objects.get(pk=dependency_id)
+        p.dependencies.add(dep)
+
+    @dbquery
+    def add_tinderbox(self, ip):
+        t = Tinderbox.objects.filter(ip=ip)
+        if len(t) > 0:
+            return t[0].id
+
+        t = Tinderbox(ip=ip)
+        t.save()
+        return t.id
+
+    @dbquery
+    def add_attachment(self, packageproperties_id, name, content, mimetype):
+        pp = PackageProperties.objects.get(pk=packageproperties_id)
+
+        attachments = Attachment.objects.filter(packageproperties=pp, name=name)
+        if len(attachments) > 0:
+            return attachments[0].id
+
+        a = Attachment(packageproperties=pp, name=name, content=content, mimetype=mimetype)
+        a.save()
+        return a.id
+
+    @dbquery
+    def add_portage_profile(self, name):
+        pp = PortageProfile.objects.filter(name=name)
+        if len(pp) > 0:
+            return pp[0].id
+        pp = PortageProfile(name=name)
+        pp.save()
+        return pp.id
+
+    @dbquery
+    def add_useflag(self, name):
+        u = Useflag.objects.filter(name=name)
+        if len(u) > 0:
+            return u[0].id
+        u = Useflag(name=name)
+        u.save()
+        return u.id
+
+    @dbquery
+    def add_packageproperties(self, packageversion_id, profile_id, tinderbox_id, error_code):
+        pv = PackageVersion.objects.get(pk=packageversion_id)
+        profile = PortageProfile.objects.get(pk=profile_id)
+        tinderbox = Tinderbox.objects.get(pk=tinderbox_id)
+
+        pp = PackageProperties(packageversion=pv,
+                               profile=profile,
+                               tinderbox=tinderbox,
+                               error_code=error_code)
+        pp.save()
+        return pp.id
+
+    @dbquery
+    def add_useflags_to_packageproperies(self, packageproperties_id, useflag_ids):
+        pp = PackageProperties.objects.get(pk=packageproperties_id)
+        pp.useflags.clear()
+        for id in useflag_ids:
+            useflag = Useflag.objects.get(pk=id)
+            pp.useflags.add(useflag)
+        pp.save()
+
+    @dbquery
+    def add_contents_to_packageproperties(self, packageproperties_id, contents):
+        pp = PackageProperties.objects.get(pk=packageproperties_id)
+
+        for path in contents.keys():
+            type = contents[path][0]
+            file = File.objects.filter(path=path)
+            if len(file) == 0:
+                file = File(path=path)
+                file.save()
+            else:
+                file = file[0]
+
+            filetype = FileType.objects.filter(name=type)
+            if len(filetype) == 0:
+                raise DatabaseError ("Database not initialized from fixtures!!\nFile type loading failed (for type %s)" % type)
+            filetype = filetype[0]
+            hash = ""
+            size = 0
+            if type == 'obj':
+                size = contents[path][1]
+                hash = contents[path][2]
+
+            ppf = PackageProperties_File(file=file,
+                                         filetype=filetype,
+                                         packageproperties=pp,
+                                         hash=hash,
+                                         size=size)
+            ppf.save()
+