Add load_compressed function to city.py
This commit is contained in:
parent
cf6542cb0c
commit
70cc659c14
@ -299,6 +299,21 @@ class City:
|
||||
with open(city_filename, 'rb') as 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):
|
||||
"""
|
||||
Save a city into the given filename
|
||||
|
Loading…
Reference in New Issue
Block a user