-
Notifications
You must be signed in to change notification settings - Fork 0
/
manipulate_data.py
70 lines (60 loc) · 2.47 KB
/
manipulate_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import os
import os
import json
# Type hints
from typing import List, Dict, Union, Any
def load_json_file(folder_name: str, file_name: str) -> Union[List[Dict[str, str]], None]:
'''
Carrega as informações da lista armazenada em um arquivo JSON.
Parameters:
-----------
folder_name: str
Nome da pasta onde o arquivo JSON está armazenado.
file_name: str
Nome do arquivo JSON.
Returns:
--------
result: list
Lista com as informações do arquivo JSON.
'''
current_dir = os.path.dirname(os.path.abspath(__file__))
file_path = os.path.join(current_dir, folder_name, file_name + '.json')
if not os.path.exists(file_path):
return None
with open(file_path, 'r') as f:
data = json.load(f)
result = [[v for _,v in restaurant.items()] for restaurant in data]
return result
def save_json_file(folder_name: str, file_name: str, data: List[List[Any]], column_name: List[str]) -> None:
'''
Armazena as informações da lista para um arquivo JSON.
Parameters:
-----------
folder_name: str
Nome da pasta onde o arquivo JSON será armazenado.
file_name: str
Nome do arquivo JSON.
data: list
Lista com as informações a serem armazenadas.
column_name: list
Lista com os nomes das colunas da lista.
'''
current_dir = os.path.dirname(os.path.abspath(__file__))
# Se a pasta não existir, cria a pasta
if not os.path.exists(os.path.join(current_dir, folder_name)):
os.makedirs(os.path.join(current_dir, folder_name))
file_path = os.path.join(current_dir, folder_name, file_name + '.json')
data_json = [{k:v for k,v in zip(column_name, restaurant)} for restaurant in data]
with open(file_path, 'w') as f:
json.dump(data_json, f, indent=4)
if __name__ == '__main__':
# Cria uma lista referente aos restaurantes para serem salvos
restaurants = [['Restaurante A', 'Endereço A', 'Telefone A', 'Especialidade A'], ['Restaurante B', 'Endereço B', 'Telefone B', 'Especialidade B']]
# Cria uma lista com os nomes das colunas
column_names = ['Nome', 'Endereço', 'Telefone', 'Especialidade']
# Salva a lista no arquivo JSON
save_json_file('Cadastro', 'Restaurante', restaurants, column_names)
# Carrega a lista do arquivo JSON
result = load_json_file('Cadastro', 'Restaurante')
# Imprime a lista
print(result)