m_osman/city_retrofit
1
0
Fork 0
forked from s_ranjbar/city_retrofit
Code Pull Requests Activity

Add load_compressed function to city.py

Browse Source
This commit is contained in:
Koa Wells 2023-09-12 14:57:13 -04:00
parent cf6542cb0c
commit 70cc659c14
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
hub/city_model_structure/city.py
View File

@ -299,6 +299,21 @@ class City:
with open(city_filename, 'rb') as file: with open(city_filename, 'rb') as file:
return pickle.load(file) return pickle.load(file)
@staticmethod
def load_compressed(compressed_city_filename, destination_filename) -> City:
"""
Load the pickle file name in the current city
:param source_file: Compressed pickle as source
:param destination_file: Pickle file as destination
:return: City
"""
with open(str(compressed_city_filename), 'rb') as source, open(str(destination_filename), 'wb') as destination:
destination.write(bz2.decompress(source.read()))
return City.load(destination_filename)
@staticmethod
def load_compressed(city_filename, ):
def save(self, city_filename): def save(self, city_filename):
""" """
Save a city into the given filename Save a city into the given filename