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()