CERC/api_v1.4
58
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Renamed export factory

Browse Source
This commit is contained in:
Peter Yefi 2023-01-17 20:13:54 -05:00
parent 582c18ef0c
commit c462d75e08
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
hub_api/config.py
View File

@ -12,13 +12,13 @@ import pickle
class Config: class Config:
def __init__(self): def __init__(self):
self.factory = CityExportFactory(db_name='hub_prod', app_env='PROD', self.export_db_factory = CityExportFactory(db_name='hub_prod', app_env='PROD',
dotenv_path="{}/.env".format(os.path.expanduser('~'))) dotenv_path="{}/.env".format(os.path.expanduser('~')))
self.import_db_factory = DBFactory(db_name='hub_prod', app_env='PROD', self.import_db_factory = DBFactory(db_name='hub_prod', app_env='PROD',
dotenv_path="{}/.env".format(os.path.expanduser('~'))) dotenv_path="{}/.env".format(os.path.expanduser('~')))
def get_city(self, city_id): def get_city(self, city_id):
city_obj = self.factory.get_city(city_id) city_obj = self.export_db_factory.get_city(city_id)
city = pickle.loads(city_obj.city) city = pickle.loads(city_obj.city)
for building in city.buildings: for building in city.buildings:
building.heated = True building.heated = True