mirror of
https://github.com/CloverHackyColor/CloverBootloader.git
synced 2024-12-11 14:28:08 +01:00
307 lines
8.5 KiB
Python
307 lines
8.5 KiB
Python
## @file
|
|
# This file is used to create/update/query/erase table for files
|
|
#
|
|
# Copyright (c) 2008 - 2018, Intel Corporation. All rights reserved.<BR>
|
|
# SPDX-License-Identifier: BSD-2-Clause-Patent
|
|
#
|
|
|
|
##
|
|
# Import Modules
|
|
#
|
|
import Common.LongFilePathOs as os
|
|
|
|
import Common.EdkLogger as EdkLogger
|
|
from CommonDataClass import DataClass
|
|
from CommonDataClass.DataClass import FileClass
|
|
|
|
## Convert to SQL required string format
|
|
def ConvertToSqlString(StringList):
|
|
return list(map(lambda s: "'" + s.replace("'", "''") + "'", StringList))
|
|
|
|
## TableFile
|
|
#
|
|
# This class defined a common table
|
|
#
|
|
# @param object: Inherited from object class
|
|
#
|
|
# @param Cursor: Cursor of the database
|
|
# @param TableName: Name of the table
|
|
#
|
|
class Table(object):
|
|
_COLUMN_ = ''
|
|
_ID_STEP_ = 1
|
|
_ID_MAX_ = 0x80000000
|
|
_DUMMY_ = 0
|
|
|
|
def __init__(self, Db, Name='', IdBase=0, Temporary=False):
|
|
self.Db = Db
|
|
self.Table = Name
|
|
self.IdBase = int(IdBase)
|
|
self.ID = int(IdBase)
|
|
self.Temporary = Temporary
|
|
self.Contents = []
|
|
|
|
def __str__(self):
|
|
return self.Table
|
|
|
|
## Create table
|
|
#
|
|
# Create a table
|
|
#
|
|
def Create(self, NewTable=True):
|
|
self.Db.CreateEmptyTable(self.Table)
|
|
self.ID = self.GetId()
|
|
|
|
## Insert table
|
|
#
|
|
# Insert a record into a table
|
|
#
|
|
def Insert(self, *Args):
|
|
self.ID = self.ID + self._ID_STEP_
|
|
if self.ID >= (self.IdBase + self._ID_MAX_):
|
|
self.ID = self.IdBase + self._ID_STEP_
|
|
row = [self.ID]
|
|
row.extend(Args)
|
|
self.Contents.append(row)
|
|
|
|
return self.ID
|
|
|
|
|
|
## Get count
|
|
#
|
|
# Get a count of all records of the table
|
|
#
|
|
# @retval Count: Total count of all records
|
|
#
|
|
def GetCount(self):
|
|
tab = self.Db.GetTable(self.Table)
|
|
return len(tab)
|
|
|
|
|
|
def GetId(self):
|
|
tab = self.Db.GetTable(self.Table)
|
|
Id = max([int(item[0]) for item in tab])
|
|
if Id is None:
|
|
Id = self.IdBase
|
|
return Id
|
|
|
|
## Init the ID of the table
|
|
#
|
|
# Init the ID of the table
|
|
#
|
|
def InitID(self):
|
|
self.ID = self.GetId()
|
|
|
|
## Exec
|
|
#
|
|
# Exec Sql Command, return result
|
|
#
|
|
# @param SqlCommand: The SqlCommand to be executed
|
|
#
|
|
# @retval RecordSet: The result after executed
|
|
#
|
|
def Exec(self, SqlCommand):
|
|
EdkLogger.debug(EdkLogger.DEBUG_5, SqlCommand)
|
|
self.Db.execute(SqlCommand)
|
|
RecordSet = self.Db.fetchall()
|
|
return RecordSet
|
|
|
|
def SetEndFlag(self):
|
|
Tab = self.Db.GetTable(self.Table)
|
|
Tab.append(self._DUMMY_)
|
|
|
|
|
|
def IsIntegral(self):
|
|
tab = self.Db.GetTable(self.Table)
|
|
Id = min([int(item[0]) for item in tab])
|
|
if Id != -1:
|
|
return False
|
|
return True
|
|
|
|
def GetAll(self):
|
|
tab = self.Db.GetTable(self.Table)
|
|
return tab
|
|
|
|
|
|
## TableFile
|
|
#
|
|
# This class defined a table used for file
|
|
#
|
|
# @param object: Inherited from object class
|
|
#
|
|
class TableFile(Table):
|
|
_COLUMN_ = '''
|
|
ID INTEGER PRIMARY KEY,
|
|
Name VARCHAR NOT NULL,
|
|
ExtName VARCHAR,
|
|
Path VARCHAR,
|
|
FullPath VARCHAR NOT NULL,
|
|
Model INTEGER DEFAULT 0,
|
|
TimeStamp SINGLE NOT NULL,
|
|
FromItem REAL NOT NULL
|
|
'''
|
|
def __init__(self, Cursor):
|
|
Table.__init__(self, Cursor, 'File')
|
|
|
|
## Insert table
|
|
#
|
|
# Insert a record into table 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, FromItem=0):
|
|
(Name, ExtName, Path, FullPath) = ConvertToSqlString((Name, ExtName, Path, FullPath))
|
|
return Table.Insert(
|
|
self,
|
|
Name,
|
|
ExtName,
|
|
Path,
|
|
FullPath,
|
|
Model,
|
|
TimeStamp,
|
|
FromItem
|
|
)
|
|
|
|
## 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, File, Model, FromItem=''):
|
|
if FromItem:
|
|
return self.Insert(
|
|
File.Name,
|
|
File.Ext,
|
|
File.Dir,
|
|
File.Path,
|
|
Model,
|
|
File.TimeStamp,
|
|
FromItem
|
|
)
|
|
return self.Insert(
|
|
File.Name,
|
|
File.Ext,
|
|
File.Dir,
|
|
File.Path,
|
|
Model,
|
|
File.TimeStamp
|
|
)
|
|
|
|
## Get type of a given file
|
|
#
|
|
# @param FileId ID of a file
|
|
#
|
|
# @retval file_type Model value of given file in the table
|
|
#
|
|
def GetFileType(self, FileId):
|
|
QueryScript = "select Model from %s where ID = '%s'" % (self.Table, FileId)
|
|
RecordList = self.Exec(QueryScript)
|
|
if len(RecordList) == 0:
|
|
return None
|
|
return RecordList[0][0]
|
|
|
|
## Get file timestamp of a given file
|
|
#
|
|
# @param FileId ID of file
|
|
#
|
|
# @retval timestamp TimeStamp value of given file in the table
|
|
#
|
|
def GetFileTimeStamp(self, FileId):
|
|
QueryScript = "select TimeStamp from %s where ID = '%s'" % (self.Table, FileId)
|
|
RecordList = self.Exec(QueryScript)
|
|
if len(RecordList) == 0:
|
|
return None
|
|
return RecordList[0][0]
|
|
|
|
## Update the timestamp of a given file
|
|
#
|
|
# @param FileId ID of file
|
|
# @param TimeStamp Time stamp of file
|
|
#
|
|
def SetFileTimeStamp(self, FileId, TimeStamp):
|
|
self.Exec("update %s set TimeStamp=%s where ID='%s'" % (self.Table, TimeStamp, FileId))
|
|
|
|
## Get list of file with given type
|
|
#
|
|
# @param FileType Type value of file
|
|
#
|
|
# @retval file_list List of files with the given type
|
|
#
|
|
def GetFileList(self, FileType):
|
|
RecordList = self.Exec("select FullPath from %s where Model=%s" % (self.Table, FileType))
|
|
if len(RecordList) == 0:
|
|
return []
|
|
return [R[0] for R in RecordList]
|
|
|
|
## TableDataModel
|
|
#
|
|
# This class defined a table used for data model
|
|
#
|
|
# @param object: Inherited from object class
|
|
#
|
|
#
|
|
class TableDataModel(Table):
|
|
_COLUMN_ = """
|
|
ID INTEGER PRIMARY KEY,
|
|
CrossIndex INTEGER NOT NULL,
|
|
Name VARCHAR NOT NULL,
|
|
Description VARCHAR
|
|
"""
|
|
def __init__(self, Cursor):
|
|
Table.__init__(self, Cursor, 'DataModel')
|
|
|
|
## Insert table
|
|
#
|
|
# Insert a record into table DataModel
|
|
#
|
|
# @param ID: ID of a ModelType
|
|
# @param CrossIndex: CrossIndex of a ModelType
|
|
# @param Name: Name of a ModelType
|
|
# @param Description: Description of a ModelType
|
|
#
|
|
def Insert(self, CrossIndex, Name, Description):
|
|
(Name, Description) = ConvertToSqlString((Name, Description))
|
|
return Table.Insert(self, CrossIndex, Name, Description)
|
|
|
|
## Init table
|
|
#
|
|
# Create all default records of table DataModel
|
|
#
|
|
def InitTable(self):
|
|
EdkLogger.verbose("\nInitialize table DataModel started ...")
|
|
Count = self.GetCount()
|
|
if Count is not None and Count != 0:
|
|
return
|
|
for Item in DataClass.MODEL_LIST:
|
|
CrossIndex = Item[1]
|
|
Name = Item[0]
|
|
Description = Item[0]
|
|
self.Insert(CrossIndex, Name, Description)
|
|
EdkLogger.verbose("Initialize table DataModel ... DONE!")
|
|
|
|
## Get CrossIndex
|
|
#
|
|
# Get a model's cross index from its name
|
|
#
|
|
# @param ModelName: Name of the model
|
|
# @retval CrossIndex: CrossIndex of the model
|
|
#
|
|
def GetCrossIndex(self, ModelName):
|
|
CrossIndex = -1
|
|
SqlCommand = """select CrossIndex from DataModel where name = '""" + ModelName + """'"""
|
|
self.Db.execute(SqlCommand)
|
|
for Item in self.Db:
|
|
CrossIndex = Item[0]
|
|
|
|
return CrossIndex
|
|
|