88
99from database .models .reports import Upload
1010from helpers .exceptions import ReportEmptyError , ReportExpiredException
11- from helpers .labels import get_all_report_labels , get_labels_per_session
1211from services .path_fixer import PathFixer
1312from services .report .parser .types import ParsedRawReport
1413from services .report .report_builder import ReportBuilder
@@ -124,7 +123,7 @@ def process_raw_upload(
124123 # Adjust sessions removed carryforward sessions that are being replaced
125124 if session .flags :
126125 session_adjustment = clear_carryforward_sessions (
127- report , temporary_report , session .flags , commit_yaml , upload
126+ report , session .flags , commit_yaml
128127 )
129128 else :
130129 session_adjustment = SessionAdjustmentResult ([], [])
@@ -138,52 +137,21 @@ def process_raw_upload(
138137@sentry_sdk .trace
139138def clear_carryforward_sessions (
140139 original_report : Report ,
141- to_merge_report : Report ,
142140 to_merge_flags : list [str ],
143141 current_yaml : UserYaml ,
144- upload : Upload | None = None ,
145142) -> SessionAdjustmentResult :
146- flags_under_carryforward_rules = {
143+ to_fully_overwrite_flags = {
147144 f for f in to_merge_flags if current_yaml .flag_has_carryfoward (f )
148145 }
149- to_partially_overwrite_flags = {
150- f
151- for f in flags_under_carryforward_rules
152- if current_yaml .get_flag_configuration (f ).get ("carryforward_mode" ) == "labels"
153- }
154- to_fully_overwrite_flags = flags_under_carryforward_rules .difference (
155- to_partially_overwrite_flags
156- )
157-
158- if upload is None and to_partially_overwrite_flags :
159- log .warning ("Upload is None, but there are partial_overwrite_flags present" )
160146
161147 session_ids_to_fully_delete = []
162- session_ids_to_partially_delete = []
163-
164- if to_fully_overwrite_flags or to_partially_overwrite_flags :
148+ if to_fully_overwrite_flags :
165149 for session_id , session in original_report .sessions .items ():
166150 if session .session_type == SessionType .carriedforward and session .flags :
167151 if any (f in to_fully_overwrite_flags for f in session .flags ):
168152 session_ids_to_fully_delete .append (session_id )
169- if any (f in to_partially_overwrite_flags for f in session .flags ):
170- session_ids_to_partially_delete .append (session_id )
171153
172- actually_fully_deleted_sessions = set ()
173154 if session_ids_to_fully_delete :
174155 original_report .delete_multiple_sessions (session_ids_to_fully_delete )
175- actually_fully_deleted_sessions .update (session_ids_to_fully_delete )
176-
177- if session_ids_to_partially_delete :
178- all_labels = get_all_report_labels (to_merge_report )
179- original_report .delete_labels (session_ids_to_partially_delete , all_labels )
180- for s in session_ids_to_partially_delete :
181- labels_now = get_labels_per_session (original_report , s )
182- if not labels_now :
183- actually_fully_deleted_sessions .add (s )
184- original_report .delete_session (s )
185-
186- return SessionAdjustmentResult (
187- sorted (actually_fully_deleted_sessions ),
188- sorted (set (session_ids_to_partially_delete ) - actually_fully_deleted_sessions ),
189- )
156+
157+ return SessionAdjustmentResult (sorted (session_ids_to_fully_delete ), [])
0 commit comments