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