CERC/hub
61
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases 95 Wiki Activity

Add removal of decompressed pickle file

Browse Source
This commit is contained in:
Koa Wells 2023-10-04 15:11:35 -04:00
parent 29ba43f1d9
commit a97c068513
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
hub/city_model_structure/city.py
View File

@ -14,6 +14,7 @@ import math
import pickle import pickle
import sys import sys
import pathlib import pathlib
import os
from pathlib import Path from pathlib import Path
from typing import List, Union from typing import List, Union
@ -309,7 +310,9 @@ class City:
""" """
with open(str(compressed_city_filename), 'rb') as source, open(str(destination_filename), 'wb') as destination: with open(str(compressed_city_filename), 'rb') as source, open(str(destination_filename), 'wb') as destination:
destination.write(bz2.decompress(source.read())) destination.write(bz2.decompress(source.read()))
return City.load(destination_filename) loaded_city = City.load(destination_filename)
os.unlink(destination_filename)
return loaded_city
def save(self, city_filename): def save(self, city_filename):
""" """