BaseTools/Source/Python/Table/TableFile.py

   1 ## @file
   2 # This file is used to create/update/query/erase table for files
   3 #
   4 # Copyright (c) 2008, Intel Corporation. All rights reserved.<BR>
   5 # This program and the accompanying materials
   6 # are licensed and made available under the terms and conditions of the BSD License
   7 # which accompanies this distribution.  The full text of the license may be found at
   8 # http://opensource.org/licenses/bsd-license.php
   9 #
  10 # THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
  12 #
  13
  14 ##
  15 # Import Modules
  16 #
  17 import Common.EdkLogger as EdkLogger
  18 from Table import Table
  19 from Common.String import ConvertToSqlString
  20 import os
  21 from CommonDataClass.DataClass import FileClass
  22
  23 ## TableFile
  24 #
  25 # This class defined a table used for file
  26 #
  27 # @param object:       Inherited from object class
  28 #
  29 class TableFile(Table):
  30     def __init__(self, Cursor):
  31         Table.__init__(self, Cursor)
  32         self.Table = 'File'
  33
  34     ## Create table
  35     #
  36     # Create table File
  37     #
  38     # @param ID:        ID of a File
  39     # @param Name:      Name of a File
  40     # @param ExtName:   ExtName of a File
  41     # @param Path:      Path of a File
  42     # @param FullPath:  FullPath of a File
  43     # @param Model:     Model of a File
  44     # @param TimeStamp: TimeStamp of a File
  45     #
  46     def Create(self):
  47         SqlCommand = """create table IF NOT EXISTS %s (ID INTEGER PRIMARY KEY,
  48                                                        Name VARCHAR NOT NULL,
  49                                                        ExtName VARCHAR,
  50                                                        Path VARCHAR,
  51                                                        FullPath VARCHAR NOT NULL,
  52                                                        Model INTEGER DEFAULT 0,
  53                                                        TimeStamp VARCHAR NOT NULL
  54                                                       )""" % self.Table
  55         Table.Create(self, SqlCommand)
  56
  57     ## Insert table
  58     #
  59     # Insert a record into table File
  60     #
  61     # @param ID:        ID of a File
  62     # @param Name:      Name of a File
  63     # @param ExtName:   ExtName of a File
  64     # @param Path:      Path of a File
  65     # @param FullPath:  FullPath of a File
  66     # @param Model:     Model of a File
  67     # @param TimeStamp: TimeStamp of a File
  68     #
  69     def Insert(self, Name, ExtName, Path, FullPath, Model, TimeStamp):
  70         self.ID = self.ID + 1
  71         (Name, ExtName, Path, FullPath) = ConvertToSqlString((Name, ExtName, Path, FullPath))
  72         SqlCommand = """insert into %s values(%s, '%s', '%s', '%s', '%s', %s, '%s')""" \
  73                                            % (self.Table, self.ID, Name, ExtName, Path, FullPath, Model, TimeStamp)
  74         Table.Insert(self, SqlCommand)
  75
  76         return self.ID
  77     ## InsertFile
  78     #
  79     # Insert one file to table
  80     #
  81     # @param FileFullPath:  The full path of the file
  82     # @param Model:         The model of the file
  83     #
  84     # @retval FileID:       The ID after record is inserted
  85     #
  86     def InsertFile(self, FileFullPath, Model):
  87         (Filepath, Name) = os.path.split(FileFullPath)
  88         (Root, Ext) = os.path.splitext(FileFullPath)
  89         TimeStamp = os.stat(FileFullPath)[8]
  90         File = FileClass(-1, Name, Ext, Filepath, FileFullPath, Model, '', [], [], [])
  91         return self.Insert(File.Name, File.ExtName, File.Path, File.FullPath, File.Model, TimeStamp)
  92
  93     ## Get ID of a given file
  94     #
  95     #   @param  FilePath    Path of file
  96     #
  97     #   @retval ID          ID value of given file in the table
  98     #
  99     def GetFileId(self, File):
 100         QueryScript = "select ID from %s where FullPath = '%s'" % (self.Table, str(File))
 101         RecordList = self.Exec(QueryScript)
 102         if len(RecordList) == 0:
 103             return None
 104         return RecordList[0][0]