[mirror_edk2.git] / BaseTools / Source / Python / PackagingTool / IpiDb.py

## @file\r
# This file is for installed package information database operations\r
#\r
# Copyright (c) 2007 - 2008, Intel Corporation. All rights reserved.<BR>\r
# This program and the accompanying materials\r
# are licensed and made available under the terms and conditions of the BSD License\r
# which accompanies this distribution.  The full text of the license may be found at\r
# http://opensource.org/licenses/bsd-license.php\r
#\r
# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
#\r
\r
##\r
# Import Modules\r
#\r
import sqlite3\r
import os\r
import time\r
import Common.EdkLogger as EdkLogger\r
\r
from CommonDataClass import DistributionPackageClass\r
\r
## IpiDb\r
#\r
# This class represents the installed package information databse\r
# Add/Remove/Get installed distribution package information here.\r
# \r
# \r
# @param object:      Inherited from object class\r
# @param DbPath:      A string for the path of the database\r
#\r
# @var Conn:          Connection of the database\r
# @var Cur:           Cursor of the connection\r
#\r
class IpiDatabase(object):\r
    def __init__(self, DbPath):\r
        Dir = os.path.dirname(DbPath)\r
        if not os.path.isdir(Dir):\r
            os.mkdir(Dir)\r
        self.Conn = sqlite3.connect(DbPath, isolation_level = 'DEFERRED')\r
        self.Conn.execute("PRAGMA page_size=4096")\r
        self.Conn.execute("PRAGMA synchronous=OFF")\r
        self.Cur = self.Conn.cursor()\r
        self.DpTable = 'DpInfo'\r
        self.PkgTable = 'PkgInfo'\r
        self.ModInPkgTable = 'ModInPkgInfo'\r
        self.StandaloneModTable = 'StandaloneModInfo'\r
        self.ModDepexTable = 'ModDepexInfo'\r
        self.DpFileListTable = 'DpFileListInfo'\r
    \r
    ## Initialize build database\r
    #\r
    #\r
    def InitDatabase(self):\r
        EdkLogger.verbose("\nInitialize IPI database started ...")\r
        \r
        #\r
        # Create new table\r
        #\r
        SqlCommand = """create table IF NOT EXISTS %s (DpGuid TEXT NOT NULL,\r
                                                       DpVersion TEXT NOT NULL,\r
                                                       InstallTime REAL NOT NULL,\r
                                                       PkgFileName TEXT,\r
                                                       PRIMARY KEY (DpGuid, DpVersion)\r
                                                      )""" % self.DpTable\r
        self.Cur.execute(SqlCommand)\r
        \r
        SqlCommand = """create table IF NOT EXISTS %s (FilePath TEXT NOT NULL,\r
                                                       DpGuid TEXT,\r
                                                       DpVersion TEXT,\r
                                                       PRIMARY KEY (FilePath)\r
                                                      )""" % self.DpFileListTable\r
        self.Cur.execute(SqlCommand)\r
        \r
        SqlCommand = """create table IF NOT EXISTS %s (PackageGuid TEXT NOT NULL,\r
                                                       PackageVersion TEXT NOT NULL,\r
                                                       InstallTime REAL NOT NULL,\r
                                                       DpGuid TEXT,\r
                                                       DpVersion TEXT,\r
                                                       InstallPath TEXT NOT NULL,\r
                                                       PRIMARY KEY (PackageGuid, PackageVersion, InstallPath)\r
                                                      )""" % self.PkgTable\r
        self.Cur.execute(SqlCommand)\r
        \r
        SqlCommand = """create table IF NOT EXISTS %s (ModuleGuid TEXT NOT NULL,\r
                                                       ModuleVersion TEXT NOT NULL,\r
                                                       InstallTime REAL NOT NULL,\r
                                                       PackageGuid TEXT,\r
                                                       PackageVersion TEXT,\r
                                                       InstallPath TEXT NOT NULL,\r
                                                       PRIMARY KEY (ModuleGuid, ModuleVersion, InstallPath)\r
                                                      )""" % self.ModInPkgTable\r
        self.Cur.execute(SqlCommand)\r
        \r
        SqlCommand = """create table IF NOT EXISTS %s (ModuleGuid TEXT NOT NULL,\r
                                                       ModuleVersion TEXT NOT NULL,\r
                                                       InstallTime REAL NOT NULL,\r
                                                       DpGuid TEXT,\r
                                                       DpVersion TEXT,\r
                                                       InstallPath TEXT NOT NULL,\r
                                                       PRIMARY KEY (ModuleGuid, ModuleVersion, InstallPath)\r
                                                      )""" % self.StandaloneModTable\r
        self.Cur.execute(SqlCommand)\r
        \r
        SqlCommand = """create table IF NOT EXISTS %s (ModuleGuid TEXT NOT NULL,\r
                                                       ModuleVersion TEXT NOT NULL,\r
                                                       InstallPath TEXT NOT NULL,\r
                                                       DepexGuid TEXT,\r
                                                       DepexVersion TEXT\r
                                                      )""" % self.ModDepexTable\r
        self.Cur.execute(SqlCommand)\r
        \r
        self.Conn.commit()\r
        \r
        EdkLogger.verbose("Initialize IPI database ... DONE!")\r
\r
    ## Add a distribution install information from DpObj\r
    #\r
    # @param DpObj:\r
    #\r
    def AddDPObject(self, DpObj):\r
        \r
        for PkgKey in DpObj.PackageSurfaceArea.keys():\r
            PkgGuid = PkgKey[0]\r
            PkgVersion = PkgKey[1]\r
            PkgInstallPath = PkgKey[2]\r
            self.AddPackage(PkgGuid, PkgVersion, DpObj.Header.Guid, DpObj.Header.Version, PkgInstallPath)\r
            PkgObj = DpObj.PackageSurfaceArea[PkgKey]\r
            for ModKey in PkgObj.Modules.keys():\r
                ModGuid = ModKey[0]\r
                ModVersion = ModKey[1]\r
                ModInstallPath = ModKey[2]\r
                self.AddModuleInPackage(ModGuid, ModVersion, PkgGuid, PkgVersion, ModInstallPath)\r
                ModObj = PkgObj.Modules[ModKey]\r
                for Dep in ModObj.PackageDependencies:\r
                    DepexGuid = Dep.PackageGuid\r
                    DepexVersion = Dep.PackageVersion\r
                    self.AddModuleDepex(ModGuid, ModVersion, ModInstallPath, DepexGuid, DepexVersion)\r
            for FilePath in PkgObj.FileList:\r
                self.AddDpFilePathList(DpObj.Header.Guid, DpObj.Header.Version, FilePath)\r
\r
        for ModKey in DpObj.ModuleSurfaceArea.keys():\r
            ModGuid = ModKey[0]\r
            ModVersion = ModKey[1]\r
            ModInstallPath = ModKey[2]\r
            self.AddStandaloneModule(ModGuid, ModVersion, DpObj.Header.Guid, DpObj.Header.Version, ModInstallPath)\r
            ModObj = DpObj.ModuleSurfaceArea[ModKey]\r
            for Dep in ModObj.PackageDependencies:\r
                DepexGuid = Dep.PackageGuid\r
                DepexVersion = Dep.PackageVersion\r
                self.AddModuleDepex(ModGuid, ModVersion, ModInstallPath, DepexGuid, DepexVersion)\r
            for FilePath in ModObj.FileList:\r
                self.AddDpFilePathList(DpObj.Header.Guid, DpObj.Header.Version, FilePath)\r
                \r
        self.AddDp(DpObj.Header.Guid, DpObj.Header.Version, DpObj.Header.FileName)    \r
    ## Add a distribution install information\r
    #\r
    # @param Guid:  \r
    # @param Version:\r
    # @param PkgFileName:\r
    #\r
    def AddDp(self, Guid, Version, PkgFileName = None):\r
        \r
        if Version == None or len(Version.strip()) == 0:\r
            Version = 'N/A'\r
        \r
        #\r
        # Add newly installed DP information to DB.\r
        #\r
        if PkgFileName == None or len(PkgFileName.strip()) == 0:\r
            PkgFileName = 'N/A'\r
        (Guid, Version, PkgFileName) = (Guid, Version, PkgFileName)\r
        CurrentTime = time.time()\r
        SqlCommand = """insert into %s values('%s', '%s', %s, '%s')""" % (self.DpTable, Guid, Version, CurrentTime, PkgFileName)\r
        self.Cur.execute(SqlCommand)\r
        self.Conn.commit()\r
        \r
    ## Add a file list from DP\r
    #\r
    # @param DpGuid:  \r
    # @param DpVersion:\r
    # @param Path\r
    #\r
    def AddDpFilePathList(self, DpGuid, DpVersion, Path):\r
        \r
        SqlCommand = """insert into %s values('%s', '%s', '%s')""" % (self.DpFileListTable, Path, DpGuid, DpVersion)\r
        self.Cur.execute(SqlCommand)\r
        self.Conn.commit()\r
            \r
    ## Add a package install information\r
    #\r
    # @param Guid:  \r
    # @param Version:\r
    # @param DpGuid:  \r
    # @param DpVersion:\r
    # @param Path\r
    #\r
    def AddPackage(self, Guid, Version, DpGuid = None, DpVersion = None, Path = ''):\r
        \r
        if Version == None or len(Version.strip()) == 0:\r
            Version = 'N/A'\r
        \r
        if DpGuid == None or len(DpGuid.strip()) == 0:\r
            DpGuid = 'N/A'\r
        \r
        if DpVersion == None or len(DpVersion.strip()) == 0:\r
            DpVersion = 'N/A'\r
        \r
        #\r
        # Add newly installed package information to DB.\r
        #\r
        \r
        CurrentTime = time.time()\r
        SqlCommand = """insert into %s values('%s', '%s', %s, '%s', '%s', '%s')""" % (self.PkgTable, Guid, Version, CurrentTime, DpGuid, DpVersion, Path)\r
        self.Cur.execute(SqlCommand)\r
        self.Conn.commit()\r
        \r
    ## Add a module that from a package install information\r
    #\r
    # @param Guid:  \r
    # @param Version:\r
    # @param PkgFileName:\r
    #\r
    def AddModuleInPackage(self, Guid, Version, PkgGuid = None, PkgVersion = None, Path = ''):\r
        \r
        if Version == None or len(Version.strip()) == 0:\r
            Version = 'N/A'\r
        \r
        if PkgGuid == None or len(PkgGuid.strip()) == 0:\r
            PkgGuid = 'N/A'\r
        \r
        if PkgVersion == None or len(PkgVersion.strip()) == 0:\r
            PkgVersion = 'N/A'\r
        \r
        #\r
        # Add module from package information to DB.\r
        #\r
        CurrentTime = time.time()\r
        SqlCommand = """insert into %s values('%s', '%s', %s, '%s', '%s', '%s')""" % (self.ModInPkgTable, Guid, Version, CurrentTime, PkgGuid, PkgVersion, Path)\r
        self.Cur.execute(SqlCommand)\r
        self.Conn.commit()\r
    \r
    ## Add a module that is standalone install information\r
    #\r
    # @param Guid:  \r
    # @param Version:\r
    # @param PkgFileName:\r
    #\r
    def AddStandaloneModule(self, Guid, Version, DpGuid = None, DpVersion = None, Path = ''):\r
        \r
        if Version == None or len(Version.strip()) == 0:\r
            Version = 'N/A'\r
        \r
        if DpGuid == None or len(DpGuid.strip()) == 0:\r
            DpGuid = 'N/A'\r
        \r
        if DpVersion == None or len(DpVersion.strip()) == 0:\r
            DpVersion = 'N/A'\r
        \r
        #\r
        # Add module standalone information to DB.\r
        #\r
        CurrentTime = time.time()\r
        SqlCommand = """insert into %s values('%s', '%s', %s, '%s', '%s', '%s')""" % (self.StandaloneModTable, Guid, Version, CurrentTime, DpGuid, DpVersion, Path)\r
        self.Cur.execute(SqlCommand)\r
        self.Conn.commit()\r
    \r
    ## Add a module depex\r
    #\r
    # @param Guid:  \r
    # @param Version:\r
    # @param DepexGuid:\r
    # @param DepexVersion:\r
    #\r
    def AddModuleDepex(self, Guid, Version, Path, DepexGuid = None, DepexVersion = None):\r
                \r
        if DepexGuid == None or len(DepexGuid.strip()) == 0:\r
            DepexGuid = 'N/A'\r
        \r
        if DepexVersion == None or len(DepexVersion.strip()) == 0:\r
            DepexVersion = 'N/A'\r
        \r
        #\r
        # Add module depex information to DB.\r
        #\r
        \r
        SqlCommand = """insert into %s values('%s', '%s', '%s', '%s', '%s')""" % (self.ModDepexTable, Guid, Version, Path, DepexGuid, DepexVersion)\r
        self.Cur.execute(SqlCommand)\r
        self.Conn.commit()\r
        \r
    ## Remove a distribution install information, if no version specified, remove all DPs with this Guid.\r
    #\r
    # @param DpObj:  \r
    #\r
    def RemoveDpObj(self, DpGuid, DpVersion):\r
        \r
        PkgList = self.GetPackageListFromDp(DpGuid, DpVersion)\r
        \r
        # delete from ModDepex the standalone module's dependency\r
        SqlCommand = """delete from ModDepexInfo where ModDepexInfo.ModuleGuid in \r
                        (select ModuleGuid from StandaloneModInfo as B where B.DpGuid = '%s' and B.DpVersion = '%s')\r
                         and ModDepexInfo.ModuleVersion in\r
                        (select ModuleVersion from StandaloneModInfo as B where B.DpGuid = '%s' and B.DpVersion = '%s')\r
                         and ModDepexInfo.InstallPath in\r
                        (select InstallPath from StandaloneModInfo as B where B.DpGuid = '%s' and B.DpVersion = '%s') """ \\r
                        %(DpGuid, DpVersion, DpGuid, DpVersion, DpGuid, DpVersion)\r
\r
#        SqlCommand = """delete from %s where %s.DpGuid ='%s' and %s.DpVersion = '%s' and \r
#                        %s.ModuleGuid = %s.ModuleGuid and %s.ModuleVersion = %s.ModuleVersion and \r
#                        %s.InstallPath = %s.InstallPath""" \\r
#                        % (self.ModDepexTable, self.StandaloneModTable, DpGuid, self.StandaloneModTable, DpVersion, self.ModDepexTable, self.StandaloneModTable, self.ModDepexTable, self.StandaloneModTable, self.ModDepexTable, self.StandaloneModTable)\r
#        print SqlCommand\r
        self.Cur.execute(SqlCommand)\r
        \r
        # delete from ModDepex the from pkg module's dependency\r
        for Pkg in PkgList:\r
#            SqlCommand = """delete from %s where %s.PackageGuid ='%s' and %s.PackageVersion = '%s' and \r
#                        %s.ModuleGuid = %s.ModuleGuid and %s.ModuleVersion = %s.ModuleVersion and \r
#                        %s.InstallPath = %s.InstallPath""" \\r
#                        % (self.ModDepexTable, self.ModInPkgTable, Pkg[0], self.ModInPkgTable, Pkg[1], self.ModDepexTable, self.ModInPkgTable, self.ModDepexTable, self.ModInPkgTable, self.ModDepexTable, self.ModInPkgTable)\r
            SqlCommand = """delete from ModDepexInfo where ModDepexInfo.ModuleGuid in \r
                            (select ModuleGuid from ModInPkgInfo where ModInPkgInfo.PackageGuid ='%s' and ModInPkgInfo.PackageVersion = '%s')\r
                            and ModDepexInfo.ModuleVersion in\r
                            (select ModuleVersion from ModInPkgInfo where ModInPkgInfo.PackageGuid ='%s' and ModInPkgInfo.PackageVersion = '%s')\r
                            and ModDepexInfo.InstallPath in\r
                            (select InstallPath from ModInPkgInfo where ModInPkgInfo.PackageGuid ='%s' and ModInPkgInfo.PackageVersion = '%s')""" \\r
                            % (Pkg[0], Pkg[1],Pkg[0], Pkg[1],Pkg[0], Pkg[1])\r
            \r
            self.Cur.execute(SqlCommand)\r
        \r
        # delete the standalone module\r
        SqlCommand = """delete from %s where DpGuid ='%s' and DpVersion = '%s'""" % (self.StandaloneModTable, DpGuid, DpVersion)\r
        self.Cur.execute(SqlCommand)\r
        \r
        # delete the from pkg module\r
        for Pkg in PkgList:\r
            SqlCommand = """delete from %s where %s.PackageGuid ='%s' and %s.PackageVersion = '%s'""" \\r
                        % (self.ModInPkgTable, self.ModInPkgTable, Pkg[0], self.ModInPkgTable, Pkg[1])\r
            self.Cur.execute(SqlCommand)\r
        \r
        # delete packages\r
        SqlCommand = """delete from %s where DpGuid ='%s' and DpVersion = '%s'""" % (self.PkgTable, DpGuid, DpVersion)\r
        self.Cur.execute(SqlCommand)\r
        \r
        # delete file list from DP\r
        SqlCommand = """delete from %s where DpGuid ='%s' and DpVersion = '%s'""" % (self.DpFileListTable, DpGuid, DpVersion)\r
        self.Cur.execute(SqlCommand)\r
            \r
        # delete DP\r
        SqlCommand = """delete from %s where DpGuid ='%s' and DpVersion = '%s'""" % (self.DpTable, DpGuid, DpVersion)\r
        self.Cur.execute(SqlCommand)\r
        \r
        self.Conn.commit()\r
        \r
    ## Get a list of distribution install information.\r
    #\r
    # @param Guid:  \r
    # @param Version:  \r
    #\r
    def GetDp(self, Guid, Version):\r
        \r
        if Version == None or len(Version.strip()) == 0:\r
            Version = 'N/A'\r
            EdkLogger.verbose("\nGetting list of DP install information started ...")\r
            (DpGuid, DpVersion) = (Guid, Version)\r
            SqlCommand = """select * from %s where DpGuid ='%s'""" % (self.DpTable, DpGuid)\r
            self.Cur.execute(SqlCommand)\r
        \r
        else:\r
            EdkLogger.verbose("\nGetting DP install information started ...")\r
            (DpGuid, DpVersion) = (Guid, Version)\r
            SqlCommand = """select * from %s where DpGuid ='%s' and DpVersion = '%s'""" % (self.DpTable, DpGuid, DpVersion)\r
            self.Cur.execute(SqlCommand)\r
\r
        DpList = []\r
        for DpInfo in self.Cur:\r
            DpGuid = DpInfo[0]\r
            DpVersion = DpInfo[1]\r
            InstallTime = DpInfo[2]\r
            PkgFileName = DpInfo[3]\r
            DpList.append((DpGuid, DpVersion, InstallTime, PkgFileName))\r
        \r
        EdkLogger.verbose("Getting DP install information ... DONE!")\r
        return DpList\r
    \r
    ## Get a list of distribution install file path information.\r
    #\r
    # @param Guid:  \r
    # @param Version:  \r
    #\r
    def GetDpFileList(self, Guid, Version):\r
        \r
        (DpGuid, DpVersion) = (Guid, Version)\r
        SqlCommand = """select * from %s where DpGuid ='%s' and DpVersion = '%s'""" % (self.DpFileListTable, DpGuid, DpVersion)\r
        self.Cur.execute(SqlCommand)\r
\r
        PathList = []\r
        for Result in self.Cur:\r
            Path = Result[0]\r
            PathList.append(Path)\r
        \r
        return PathList\r
    \r
    ## Get a list of package information.\r
    #\r
    # @param Guid:  \r
    # @param Version:  \r
    #\r
    def GetPackage(self, Guid, Version, DpGuid = '', DpVersion = ''):\r
        \r
        if DpVersion == '' or DpGuid == '':\r
\r
            (PackageGuid, PackageVersion) = (Guid, Version)\r
            SqlCommand = """select * from %s where PackageGuid ='%s' and PackageVersion = '%s'""" % (self.PkgTable, PackageGuid, PackageVersion)\r
            self.Cur.execute(SqlCommand)\r
        \r
        elif Version == None or len(Version.strip()) == 0:\r
            \r
            SqlCommand = """select * from %s where PackageGuid ='%s'""" % (self.PkgTable, Guid)\r
            self.Cur.execute(SqlCommand)\r
        else:\r
            (PackageGuid, PackageVersion) = (Guid, Version)\r
            SqlCommand = """select * from %s where PackageGuid ='%s' and PackageVersion = '%s'\r
                            and DpGuid = '%s' and DpVersion = '%s'""" % (self.PkgTable, PackageGuid, PackageVersion, DpGuid, DpVersion)\r
            self.Cur.execute(SqlCommand)\r
\r
        PkgList = []\r
        for PkgInfo in self.Cur:\r
            PkgGuid = PkgInfo[0]\r
            PkgVersion = PkgInfo[1]\r
            InstallTime = PkgInfo[2]\r
            InstallPath = PkgInfo[5]\r
            PkgList.append((PkgGuid, PkgVersion, InstallTime, DpGuid, DpVersion, InstallPath))\r
        \r
        return PkgList\r
    \r
    ## Get a list of module in package information.\r
    #\r
    # @param Guid:  \r
    # @param Version:  \r
    #\r
    def GetModInPackage(self, Guid, Version, PkgGuid = '', PkgVersion = ''):\r
        \r
        if PkgVersion == '' or PkgGuid == '':\r
\r
            (ModuleGuid, ModuleVersion) = (Guid, Version)\r
            SqlCommand = """select * from %s where ModuleGuid ='%s' and ModuleVersion = '%s'""" % (self.ModInPkgTable, ModuleGuid, ModuleVersion)\r
            self.Cur.execute(SqlCommand)\r
        \r
        else:\r
            (ModuleGuid, ModuleVersion) = (Guid, Version)\r
            SqlCommand = """select * from %s where ModuleGuid ='%s' and ModuleVersion = '%s' and PackageGuid ='%s' and PackageVersion = '%s'\r
                            """ % (self.ModInPkgTable, ModuleGuid, ModuleVersion, PkgGuid, PkgVersion)\r
            self.Cur.execute(SqlCommand)\r
\r
        ModList = []\r
        for ModInfo in self.Cur:\r
            ModGuid = ModInfo[0]\r
            ModVersion = ModInfo[1]\r
            InstallTime = ModInfo[2]\r
            InstallPath = ModInfo[5]\r
            ModList.append((ModGuid, ModVersion, InstallTime, PkgGuid, PkgVersion, InstallPath))\r
        \r
        return ModList\r
    \r
    ## Get a list of module standalone.\r
    #\r
    # @param Guid:  \r
    # @param Version:  \r
    #\r
    def GetStandaloneModule(self, Guid, Version, DpGuid = '', DpVersion = ''):\r
        \r
        if DpGuid == '':\r
\r
            (ModuleGuid, ModuleVersion) = (Guid, Version)\r
            SqlCommand = """select * from %s where ModuleGuid ='%s' and ModuleVersion = '%s'""" % (self.StandaloneModTable, ModuleGuid, ModuleVersion)\r
            self.Cur.execute(SqlCommand)\r
        \r
        else:\r
            (ModuleGuid, ModuleVersion) = (Guid, Version)\r
            SqlCommand = """select * from %s where ModuleGuid ='%s' and ModuleVersion = '%s' and DpGuid ='%s' and DpVersion = '%s'\r
                            """ % (self.StandaloneModTable, ModuleGuid, ModuleVersion, DpGuid, DpVersion)\r
            self.Cur.execute(SqlCommand)\r
\r
        ModList = []\r
        for ModInfo in self.Cur:\r
            ModGuid = ModInfo[0]\r
            ModVersion = ModInfo[1]\r
            InstallTime = ModInfo[2]\r
            InstallPath = ModInfo[5]\r
            ModList.append((ModGuid, ModVersion, InstallTime, DpGuid, DpVersion, InstallPath))\r
        \r
        return ModList\r
    \r
    ## Get a list of module information that comes from DP.\r
    #\r
    # @param DpGuid:  \r
    # @param DpVersion:  \r
    #\r
    def GetStandaloneModuleInstallPathListFromDp(self, DpGuid, DpVersion):\r
\r
        PathList = []\r
        SqlCommand = """select t1.InstallPath from %s t1 where t1.DpGuid ='%s' and t1.DpVersion = '%s'\r
                        """ % (self.StandaloneModTable, DpGuid, DpVersion)\r
        self.Cur.execute(SqlCommand)\r
\r
        for Result in self.Cur:\r
            InstallPath = Result[0]\r
            PathList.append(InstallPath)\r
        \r
        return PathList\r
    \r
    ## Get a list of package information.\r
    #\r
    # @param DpGuid:  \r
    # @param DpVersion:  \r
    #\r
    def GetPackageListFromDp(self, DpGuid, DpVersion):\r
        \r
\r
        SqlCommand = """select * from %s where DpGuid ='%s' and DpVersion = '%s'\r
                        """ % (self.PkgTable, DpGuid, DpVersion)\r
        self.Cur.execute(SqlCommand)\r
\r
        PkgList = []\r
        for PkgInfo in self.Cur:\r
            PkgGuid = PkgInfo[0]\r
            PkgVersion = PkgInfo[1]\r
            InstallPath = PkgInfo[5]\r
            PkgList.append((PkgGuid, PkgVersion, InstallPath))\r
        \r
        return PkgList\r
    \r
    ## Get a list of modules that depends on package information from a DP.\r
    #\r
    # @param DpGuid:  \r
    # @param DpVersion:  \r
    #\r
    def GetDpDependentModuleList(self, DpGuid, DpVersion):\r
        \r
        ModList = []\r
        PkgList = self.GetPackageListFromDp(DpGuid, DpVersion)\r
        if len(PkgList) > 0:\r
            return ModList\r
        \r
        for Pkg in PkgList:\r
            SqlCommand = """select t1.ModuleGuid, t1.ModuleVersion, t1.InstallPath \r
                            from %s as t1, %s as t2, where t1.ModuleGuid = t2.ModuleGuid and \r
                            t1.ModuleVersion = t2.ModuleVersion and t2.DepexGuid ='%s' and (t2.DepexVersion = '%s' or t2.DepexVersion = 'N/A') and\r
                            t1.PackageGuid != '%s' and t1.PackageVersion != '%s'\r
                        """ % (self.ModInPkgTable, self.ModDepexTable, Pkg[0], Pkg[1], Pkg[0], Pkg[1])\r
            self.Cur.execute(SqlCommand)\r
            for ModInfo in self.Cur:\r
                ModGuid = ModInfo[0]\r
                ModVersion = ModInfo[1]\r
                InstallPath = ModInfo[2]\r
                ModList.append((ModGuid, ModVersion, InstallPath))\r
\r
            SqlCommand = """select t1.ModuleGuid, t1.ModuleVersion, t1.InstallPath \r
                            from %s as t1, %s as t2, where t1.ModuleGuid = t2.ModuleGuid and \r
                            t1.ModuleVersion = t2.ModuleVersion and t2.DepexGuid ='%s' and (t2.DepexVersion = '%s' or t2.DepexVersion = 'N/A') and\r
                            t1.DpGuid != '%s' and t1.DpVersion != '%s'\r
                        """ % (self.StandaloneModTable, self.ModDepexTable, Pkg[0], Pkg[1], DpGuid, DpVersion)\r
            self.Cur.execute(SqlCommand)\r
            for ModInfo in self.Cur:\r
                ModGuid = ModInfo[0]\r
                ModVersion = ModInfo[1]\r
                InstallPath = ModInfo[2]\r
                ModList.append((ModGuid, ModVersion, InstallPath))\r
        \r
        \r
        return ModList\r
    \r
    ## Get a module depex\r
    #\r
    # @param Guid:  \r
    # @param Version:\r
    # @param Path:\r
    #\r
    def GetModuleDepex(self, Guid, Version, Path):\r
                \r
        #\r
        # Get module depex information to DB.\r
        #\r
        \r
        SqlCommand = """select * from %s where ModuleGuid ='%s' and ModuleVersion = '%s' and InstallPath ='%s'\r
                            """ % (self.ModDepexTable, Guid, Version, Path)\r
        self.Cur.execute(SqlCommand)\r
        self.Conn.commit()\r
        \r
        DepexList = []\r
        for DepInfo in self.Cur:\r
            DepexGuid = DepInfo[3]\r
            DepexVersion = DepInfo[4]\r
            DepexList.append((DepexGuid, DepexVersion))\r
        \r
        return DepexList\r
    \r
    ## Close entire database\r
    #\r
    # Close the connection and cursor\r
    #\r
    def CloseDb(self):\r
        \r
        self.Cur.close()\r
        self.Conn.close()\r
\r
    ## Convert To Sql String\r
    #\r
    # 1. Replace "'" with "''" in each item of StringList\r
    # \r
    # @param StringList:  A list for strings to be converted\r
    #\r
    def __ConvertToSqlString(self, StringList):\r
        return map(lambda s: s.replace("'", "''") , StringList)\r
##\r
#\r
# This acts like the main() function for the script, unless it is 'import'ed into another\r
# script.\r
#\r
if __name__ == '__main__':\r
    EdkLogger.Initialize()\r
    EdkLogger.SetLevel(EdkLogger.DEBUG_0)\r
    DATABASE_PATH = "C://MyWork//Conf//.cache//XML.db"\r
    Db = IpiDatabase(DATABASE_PATH)\r
    Db.InitDatabase()\r
\r