🎨 Improve OOP Logic, DRYyed the codebase

InovaFiscaliza · Nov 30, 2023 · 039017a · 039017a
1 parent 00920a9
commit 039017a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/extracao/estacoes.py b/extracao/estacoes.py
@@ -253,6 +253,7 @@ def _simplify_sources(df):
 
 	@staticmethod
 	def _remove_invalid_frequencies(df):
+		df['Frequência'] = df['Frequência'].astype('float')
 		df.sort_values(['Frequência', 'Latitude', 'Longitude'], ignore_index=True, inplace=True)
 		return df[df['Frequência'] <= LIMIT_FREQ]
 		# TODO: save to discarded and log
@@ -270,6 +271,7 @@ def _format(
 		df = merge_on_frequency(anatel, aero)
 		df = self.validate_coordinates(df)
 		df = Estacoes._simplify_sources(df)
-		df = SRD._format_types(df)
 		df = Estacoes._remove_invalid_frequencies(df)
+		df = SRD._format_types(df)
+		df = df.astype('string', copy=False).replace('-1.0', '-1').astype('category', copy=False)
 		return df.loc[:, self.columns]