def InsertFileRecord(self, file_name_tid, display_file_name, data_intro,
data_privacy):
with session_scope() as session:
uploadFile = UploadFile(
file_name_tid=file_name_tid,
display_file_name=display_file_name,
data_intro=data_intro,
data_privacy=data_privacy,
)
session.add(uploadFile)
def InsertPdData(self, file_name_id, sheet_index, row_index, col_index,
unit_value):
with session_scope() as session:
pdData = PdData(
file_name_id=file_name_id,
sheet_index=sheet_index,
row_index=row_index,
col_index=col_index,
unit_value=unit_value,
)
session.add(pdData)
def UpdateFileRecord(self, id, uploadFile_):
with session_scope() as session:
uploadFile = session.query(UploadFile).filter(
UploadFile.id == id).first()
if not uploadFile:
return
uploadFile.data_intro = uploadFile_.data_intro
uploadFile.data_privacy = uploadFile_.data_privacy
uploadFile.file_name_tid = uploadFile_.file_name_tid
uploadFile.display_file_name = uploadFile_.display_file_name
return uploadFile
def UpdatePdData(self, id, pdData_):
with session_scope() as session:
pdData = session.query(PdData).filter(PdData.id == id).first()
if not pdData:
return
pdData.file_name_id = pdData_.file_name_id
pdData.sheet_index = pdData_.sheet_index
pdData.row_index = pdData_.row_index
pdData.col_index = pdData_.col_index
pdData.unit_value = pdData_.unit_value
pdData.invalid = pdData_.invalid
return pdData
def GetSheetAllData(self, file_name_id, sheet_index=0):
with session_scope() as session:
records = session.query(PdData).filter(
PdData.sheet_index == sheet_index,
PdData.file_name_id == file_name_id).all()
all_data = {}
for record in records:
temp = {}
temp[record.col_index] = record.unit_value
all_data[record.row_index] = temp
# print(all_data)
return all_data
def GetShape(self, file_name_id, sheet_index=0):
with session_scope() as session:
results = session.query(PdData.row_index, PdData.col_index).filter(
PdData.sheet_index == sheet_index,
PdData.file_name_id == file_name_id).all()
row_index_list = []
col_index_list = []
for result in results:
row_index_list.append(result[0])
for result in results:
col_index_list.append(result[1])
# print(max(row_index_list), max(col_index_list))
return max(row_index_list), max(col_index_list)
def InsertManyPdData(self, pdDataReords):
with session_scope() as session:
session.add_all(pdDataReords)
def GetUploadFileIdByTid(self, file_name_tid):
with session_scope() as session:
uploadFiles = session.query(UploadFile).filter(
UploadFile.file_name_tid == file_name_tid).first()
return uploadFiles.id
def GetAllUploadFileRecord(self):
with session_scope() as session:
uploadFiles = session.query(UploadFile).filter().all()
return uploadFiles
def DeleteFileRecord(self, id):
with session_scope() as session:
uploadFile = session.query(UploadFile).filter(
UploadFile.id == id).first()
if uploadFile:
session.delete(uploadFile)
def GetUploadFileRecordByFileNameTid(self, file_name_tid):
with session_scope() as session:
return session.query(UploadFile).filter(
UploadFile.file_name_tid == file_name_tid).first()
def GetFileRecord(self, id=None):
with session_scope() as session:
return session.query(UploadFile).filter(
UploadFile.id == id).first()
def DeletePdData(self, id):
with session_scope() as session:
pdData = session.query(PdData).filter(PdData.id == id).first()
if pdData:
session.delete(pdData)
def GetPdData(self, id=None):
with session_scope() as session:
return session.query(PdData).filter(PdData.id == id).first()