1 files changed, 91 insertions, 0 deletions
diff --git a/BaseTools/Source/Python/Table/TableFile.py b/BaseTools/Source/Python/Table/TableFile.py
new file mode 100644
index 0000000000..9be64942ec
--- /dev/null
+++ b/BaseTools/Source/Python/Table/TableFile.py
@@ -0,0 +1,91 @@
+## @file
+# This file is used to create/update/query/erase table for files
+#
+# Copyright (c) 2008, Intel Corporation
+# All rights reserved. This program and the accompanying materials
+# are licensed and made available under the terms and conditions of the BSD License
+# which accompanies this distribution.  The full text of the license may be found at
+# http://opensource.org/licenses/bsd-license.php
+#
+# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+#
+
+##
+# Import Modules
+#
+import Common.EdkLogger as EdkLogger
+from Table import Table
+from Common.String import ConvertToSqlString
+import os
+from CommonDataClass.DataClass import FileClass
+
+## TableFile
+#
+# This class defined a table used for file
+# 
+# @param object:       Inherited from object class
+#
+class TableFile(Table):
+    def __init__(self, Cursor):
+        Table.__init__(self, Cursor)
+        self.Table = 'File'
+    
+    ## Create table
+    #
+    # Create table File
+    #
+    # @param ID:        ID of a File
+    # @param Name:      Name of a File
+    # @param ExtName:   ExtName of a File
+    # @param Path:      Path of a File
+    # @param FullPath:  FullPath of a File
+    # @param Model:     Model of a File
+    # @param TimeStamp: TimeStamp of a File
+    #
+    def Create(self):
+        SqlCommand = """create table IF NOT EXISTS %s (ID INTEGER PRIMARY KEY,
+                                                       Name VARCHAR NOT NULL,
+                                                       ExtName VARCHAR,
+                                                       Path VARCHAR,
+                                                       FullPath VARCHAR NOT NULL,
+                                                       Model INTEGER DEFAULT 0,
+                                                       TimeStamp VARCHAR NOT NULL
+                                                      )""" % self.Table
+        Table.Create(self, SqlCommand)
+
+    ## Insert table
+    #
+    # Insert a record into table File
+    #
+    # @param ID:        ID of a File
+    # @param Name:      Name of a File
+    # @param ExtName:   ExtName of a File
+    # @param Path:      Path of a File
+    # @param FullPath:  FullPath of a File
+    # @param Model:     Model of a File
+    # @param TimeStamp: TimeStamp of a File
+    #
+    def Insert(self, Name, ExtName, Path, FullPath, Model, TimeStamp):
+        self.ID = self.ID + 1
+        (Name, ExtName, Path, FullPath) = ConvertToSqlString((Name, ExtName, Path, FullPath))
+        SqlCommand = """insert into %s values(%s, '%s', '%s', '%s', '%s', %s, '%s')""" \
+                                           % (self.Table, self.ID, Name, ExtName, Path, FullPath, Model, TimeStamp)
+        Table.Insert(self, SqlCommand)
+        
+        return self.ID
+    ## InsertFile
+    #
+    # Insert one file to table
+    #
+    # @param FileFullPath:  The full path of the file
+    # @param Model:         The model of the file 
+    # 
+    # @retval FileID:       The ID after record is inserted
+    #
+    def InsertFile(self, FileFullPath, Model):
+        (Filepath, Name) = os.path.split(FileFullPath)
+        (Root, Ext) = os.path.splitext(FileFullPath)
+        TimeStamp = os.stat(FileFullPath)[8]
+        File = FileClass(-1, Name, Ext, Filepath, FileFullPath, Model, '', [], [], [])
+        return self.Insert(File.Name, File.ExtName, File.Path, File.FullPath, File.Model, TimeStamp)