2023-01-17 19:00:22 -05:00
|
|
|
"""
|
|
|
|
Config
|
|
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
|
|
Copyright © 2023 Project Peter Yefi peteryefi@gmail.com
|
|
|
|
"""
|
2023-01-30 19:45:02 -05:00
|
|
|
from hub.exports.db_factory import DBFactory as CityExportFactory
|
|
|
|
from hub.imports.db_factory import DBFactory
|
2023-02-13 08:09:48 -05:00
|
|
|
import os
|
|
|
|
import pickle
|
2023-01-17 19:00:22 -05:00
|
|
|
|
|
|
|
|
|
|
|
class Config:
|
|
|
|
|
|
|
|
def __init__(self):
|
2023-02-13 08:09:48 -05:00
|
|
|
self.export_db_factory = CityExportFactory(db_name='hub_prod', app_env='PROD',
|
2023-01-17 20:13:54 -05:00
|
|
|
dotenv_path="{}/.env".format(os.path.expanduser('~')))
|
2023-02-13 08:09:48 -05:00
|
|
|
self.import_db_factory = DBFactory(db_name='hub_prod', app_env='PROD',
|
2023-01-17 20:10:20 -05:00
|
|
|
dotenv_path="{}/.env".format(os.path.expanduser('~')))
|
2023-01-17 19:00:22 -05:00
|
|
|
|
|
|
|
def get_city(self, city_id):
|
2023-02-13 08:09:48 -05:00
|
|
|
city_obj = self.export_db_factory.get_city(city_id)
|
|
|
|
city = pickle.loads(city_obj.city)
|
|
|
|
for building in city.buildings:
|
|
|
|
building.heated = True
|
|
|
|
building.cooled = True
|
|
|
|
building.attic_heated = 0
|
|
|
|
building.basement_heated = 0
|
|
|
|
for surface in building.surfaces:
|
|
|
|
surface.swr = 0.2
|
|
|
|
return city
|