def export_csv(self): if self.table_name in dtl_names: decision_table.D_table_dtl(self.file_name, file_type=self.table_name).write() elif self.table_name == 'mgt_sch': lum.Management_sch(self.file_name).write() else: ignored_cols = [] initial_headers = [] custom_query = None if self.table_name == 'rec_dat': ignored_cols.append('recall_rec') custom_query = self.table.select().where( self.table.recall_rec_id == self.related_id) elif self.table_name == 'rec_cnst': ignored_cols.append('recall_rec') ignored_cols.append('yr') ignored_cols.append('t_step') initial_headers.append('name') custom_query = Recall_dat.select( Recall_rec.name, Recall_dat).join(Recall_rec).where(Recall_rec.rec_typ == 4) try: fileio.write_csv(self.file_name, self.table, ignore_id_col=self.ignore_id_col, ignored_cols=ignored_cols, custom_query=custom_query, initial_headers=initial_headers) except PermissionError: sys.exit( 'Permission error. Please check to make sure the file is not open.' )
def import_csv(self): if self.table_name == 'rec_dat': recall.Recall_rec(self.file_name).read_data(self.related_id, self.delete_existing, self.rec_typ) elif self.table_name == 'rec_cnst': recall.Recall_rec(self.file_name).read_const_data() elif self.table_name in dtl_names: decision_table.D_table_dtl(self.file_name, file_type=self.table_name).read() elif self.table_name == 'mgt_sch': lum.Management_sch(self.file_name).read() else: fileio.read_csv_file(self.file_name, self.table, project_base.db, 0, ignore_id_col=self.ignore_id_col, overwrite=fileio.FileOverwrite.replace, remove_spaces_cols=['name'])
def write_lum(self, start_prog, allocated_prog): num_files = 5 files = self.get_file_names("lum", num_files) prog_step = round(allocated_prog / num_files) prog = start_prog landuse_lum_file = files[0].strip() if landuse_lum_file != NULL_FILE: self.update_file_status(prog, landuse_lum_file) lum.Landuse_lum(os.path.join(self.__dir, landuse_lum_file), self.__version, self.__swat_version).write() prog += prog_step management_sch_file = files[1].strip() if management_sch_file != NULL_FILE: self.update_file_status(prog, management_sch_file) lum.Management_sch(os.path.join(self.__dir, management_sch_file), self.__version, self.__swat_version).write() prog += prog_step cntable_lum_file = files[2].strip() if cntable_lum_file != NULL_FILE: self.update_file_status(prog, cntable_lum_file) lum.Cntable_lum(os.path.join(self.__dir, cntable_lum_file), self.__version, self.__swat_version).write() prog += prog_step cons_prac_lum_file = files[3].strip() if cons_prac_lum_file != NULL_FILE: self.update_file_status(prog, cons_prac_lum_file) lum.Cons_prac_lum(os.path.join(self.__dir, cons_prac_lum_file), self.__version, self.__swat_version).write() prog += prog_step ovn_table_lum_file = files[4].strip() if ovn_table_lum_file != NULL_FILE: self.update_file_status(prog, ovn_table_lum_file) lum.Ovn_table_lum(os.path.join(self.__dir, ovn_table_lum_file), self.__version, self.__swat_version).write()