Save always

pages/2_👨‍🏫_Начало_работы.py

@@ -230,16 +230,15 @@ if st.session_state.get('-LOGGED_IN_BOOL-'):
             PROGRESS_BAR.progress(100)
             PROGRESS_BAR_S.progress(100)
             LOGS.success('**Все готово! Готовые задания и/или онлайн-тест доступны в соответствующих вкладках.**')
-            if SAVE_IN_CLOUD:
-                save_data_in_database(
-                    user_task_database=user_save_text_table,
-                    save_type='download',
-                    save_name=USER__SAVE_IN_CLOUD_FILE_NAME,
-                    cefr_level=CEFR_TEXT_LEVEL,
-                    time_stamp=str(datetime.datetime.now())[:-7],
-                    creator_name=st.session_state.get('-USER_NAME-'),
-                    generated_result=__TASK_DATA__,
-                    distractor_model=DISTRACTOR_MODEL)
+            save_data_in_database(
+                user_task_database=user_save_text_table,
+                save_type='download',
+                save_name=USER__SAVE_IN_CLOUD_FILE_NAME,
+                cefr_level=CEFR_TEXT_LEVEL,
+                time_stamp=str(datetime.datetime.now())[:-7],
+                creator_name=st.session_state.get('-USER_NAME-'),
+                generated_result=__TASK_DATA__,
+                distractor_model=DISTRACTOR_MODEL, allow=SAVE_IN_CLOUD)
 
             PROGRESS_BAR = ''
             PROGRESS_BAR_S = ''

utilities_database/user_database_utils.py

@@ -192,7 +192,7 @@ def check_current_passwd(user_log_in_database, email_reset_passwd: str, current_
 
 def save_data_in_database(user_task_database, save_type, save_name, cefr_level, time_stamp, creator_name=None,
                           generated_result=None, test_taker_name=None, test_taker_answers=None, test_taker_result=None,
-                          comments=None, distractor_model=None):
+                          comments=None, distractor_model=None, allow=False):
     already_saved_names = user_task_database.fetch({'creator_name': creator_name,
                                                     'save_name': save_name,
                                                     'cefr_level': cefr_level}).items
@@ -235,7 +235,8 @@ def save_data_in_database(user_task_database, save_type, save_name, cefr_level,
                 'comments': comments}
         user_task_database.put(new_save_data)
         if save_type == 'download':
-            return st.success('Задания успешно сохранены! Можете переходить на следующие вкладки')
+            if allow:
+                return st.success('Задания успешно сохранены! Можете переходить на следующие вкладки')
         elif save_type == 'online_test':
             return st.success('Ответы успешно сохранены!')