minor formatting

Author: fccoelho

pysus/online_data/ESUS.py

@@ -7,6 +7,7 @@
 import time
 from datetime import date
 
+
 def download(uf, cache=True, checkmemory=True):
     """
     Download ESUS data by UF
@@ -19,7 +20,7 @@ def download(uf, cache=True, checkmemory=True):
     pwd = 'Za4qNXdyQNSa9YaA'
     today = date.today()
     dt = today.strftime("_%d_%m_%Y")
-    base = f'desc-notificacoes-esusve-{uf}'  #desc-notificacoes-esusve-
+    base = f'desc-notificacoes-esusve-{uf}'  # desc-notificacoes-esusve-
     url = f'https://{user}:{pwd}@elasticsearch-saps.saude.gov.br'
     out = f'ESUS_{uf}_{dt}.parquet'
 
@@ -33,7 +34,7 @@ def download(uf, cache=True, checkmemory=True):
         fname = fetch(base, uf, url)
         size = os.stat(fname).st_size
         if size > 50e6 and checkmemory:
-            print(f"Downloaded data is to large:{size/1e6} MB compressed.")
+            print(f"Downloaded data is to large:{size / 1e6} MB compressed.")
             print("Only loading the first 1000 rows. If your computer has enough memory, set 'checkmemory' to False")
             print(f"The full data is in {fname}")
             df = pd.read_csv(fname, chunksize=1000)
@@ -47,7 +48,6 @@ def download(uf, cache=True, checkmemory=True):
     return df
 
 
-
 def fetch(base, uf, url):
     UF = uf.upper()
     print(f"Reading ESUS data for {UF}")
@@ -61,16 +61,14 @@ def fetch(base, uf, url):
     tempfile = os.path.join(CACHEPATH, f'ESUS_temp_{UF}.csv.gz')
     for ch in chunker:
         df = pd.DataFrame.from_dict(ch)
-        df.sintomas = df['sintomas'].str.replace(';', '',)  ## remove os  ;
+        df.sintomas = df['sintomas'].str.replace(';', '', )  ## remove os  ;
         if h:
             df.to_csv(tempfile)
             h = 0
         else:
             df.to_csv(tempfile, mode='a', header=False)
     # df = pd.read_csv('temp.csv.gz')
 
-
-
     return tempfile
 
 

pysus/utilities/readdbc.py

@@ -7,7 +7,6 @@
 from tempfile import NamedTemporaryFile
 from io import BytesIO
 import pandas as pd
-import geopandas as gpd
 from dbfread import DBF
 import geopandas as gpd
 
@@ -16,7 +15,7 @@
 except (ImportError, ModuleNotFoundError):
     from _readdbc import ffi, lib
 
-    
+
 def read_dbc(filename, encoding='utf-8', raw=False):
     """
     Opens a DATASUS .dbc file and return its contents as a pandas
@@ -55,7 +54,7 @@ def dbc2dbf(infile, outfile):
     # print(os.path.exists(outfile))
 
 
-def read_dbc_geopandas(filename,encoding='utf-8'):
+def read_dbc_geopandas(filename, encoding='utf-8'):
     """
     Opens a DATASUS .dbc file and return its contents as a pandas
     Dataframe, using geopandas
@@ -68,9 +67,8 @@ def read_dbc_geopandas(filename,encoding='utf-8'):
     with NamedTemporaryFile(delete=False) as tf:
         out = tf.name + '.dbf'
         dbc2dbf(filename, out)
-        dbf = gpd.read_file(out, encoding=encoding).drop("geometry",axis=1)
+        dbf = gpd.read_file(out, encoding=encoding).drop("geometry", axis=1)
         df = pd.DataFrame(dbf)
     os.unlink(tf.name)
 
-    return df 
-
+    return df