🐛 Create log column when there isn't one

InovaFiscaliza · Mar 13, 2024 · b2c6e6b · b2c6e6b
1 parent d11c1ec
commit b2c6e6b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/extracao/datasources/base.py b/extracao/datasources/base.py
@@ -93,8 +93,11 @@ def register_log(
 		if row_filter is None:
 			row_filter = pd.Series(True, index=df.index)
 
-		df['Log'] = df['Log'].astype('string', copy=False).fillna('[]')
-		df['Log'] = df['Log'].str.replace('^$', r'[]', regex=True)
+		if 'Log' not in df:
+			df['Log'] = '[]'
+		else:
+			df['Log'] = df['Log'].astype('string', copy=False).fillna('[]')
+			df['Log'] = df['Log'].str.replace('^$', r'[]', regex=True)
 		print(f'Logging: {processing}')
 		log_function = partial(Base.format_log, processing=processing, column=column)
 		df.loc[row_filter, 'Log'] = df.loc[row_filter].progress_apply(log_function, axis=1)