codekingpro/portable-devtools

1from flask_login import current_user2from pgadmin.model import Setting3 4 5def get_dialog_type(file_type):6    """7    This function return dialog type8    :param file_type:9    :return: dialog type.10    """11    if 'pgerd' in file_type:12        return 'erd_file_type'13    elif 'backup' in file_type:14        return 'backup_file_type'15    elif 'csv' in file_type and 'txt' in file_type:16        return 'import_export_file_type'17    elif 'csv' in file_type and 'txt' not in file_type:18        return 'storage_manager_file_type'19    else:20        return 'sqleditor_file_format'21 22 23def get_file_type_setting(file_types):24    """25    This function return last file format setting based on file types26    :param file_types:27    :return: file format setting.28    """29    file_type = get_dialog_type(list(file_types))30 31    data = Setting.query.filter_by(32        user_id=current_user.id, setting=file_type).first()33    if data is None:34        return '*'35    else:36        return data.value37