reimplemented functions using the Directory class

Author: fccoelho

pysus/online_data/territory.py

@@ -1,26 +1,27 @@
 
 from pathlib import Path
-from typing import Union, Dict
+from typing import Union, Dict, List
 import pandas as pd
 
-from pysus.ftp import CACHEPATH
+from pysus.ftp import CACHEPATH, File, Directory
 from pysus.ftp.databases.territory import Territory
 
 ter = Territory().load()
 
-def list_tables()-> Dict:
-    files = ter.get_files()
-    tabelas = [f for f in files if 'territor' in f.name]
+def list_tables()-> List[File]:
+    d = Directory('/territorio/tabelas')
+    tabelas = [f for f in d.content if 'territor' in f.name]
     return tabelas
 
-def list_maps():
-    files = ter.get_files()
-    mapas = [f for f in files if 'mapas' in f.name]
+def list_maps()-> List[File]:
+    d = Directory('/territorio/mapas')
+    mapas = [f for f in d.content if 'mapas' in f.name]
     return mapas
 
 def download(fname: Union[str,list], data_path: str = CACHEPATH):
-    files = ter.get_files()
-    
-    dfiles = ter.download(list(fname), data_path)
+    files = Directory('/territorio/tabelas').content + Directory('/territorio/mapas').content
+    for file in files:
+        if fname in [str(file), file.name]:  # handles suffixed and no suffixed `fname`s
+            return file.download()