chore: add main file and building data points
This commit is contained in:
parent
0a2ea1061f
commit
7cc7221ab8
1048576
cmm_points_function_vintage_surface.csv
Normal file
1048576
cmm_points_function_vintage_surface.csv
Normal file
File diff suppressed because it is too large
Load Diff
42
main.py
Normal file
42
main.py
Normal file
@ -0,0 +1,42 @@
|
||||
from hub.imports.geometry_factory import GeometryFactory
|
||||
from hub.helpers.dictionaries import Dictionaries
|
||||
import os
|
||||
import psycopg2
|
||||
|
||||
password = os.getenv('PGPASSWORD')
|
||||
|
||||
input_file = "cmm_points_function_vintage_surface.csv"
|
||||
|
||||
city = GeometryFactory(
|
||||
file_type="csv",
|
||||
path=input_file,
|
||||
year_of_construction_field="anne_cnstr",
|
||||
function_field="code_util",
|
||||
function_to_hub=Dictionaries().montreal_function_to_hub_function,
|
||||
total_floor_area_field="supfi_etag"
|
||||
).city
|
||||
|
||||
|
||||
conn_params = {
|
||||
'host': 'localhost',
|
||||
'port': '5432',
|
||||
'database': 'energydemanddb',
|
||||
'user': 'postgres',
|
||||
'password': password
|
||||
}
|
||||
|
||||
try:
|
||||
conn = psycopg2.connect(**conn_params)
|
||||
print("Connection established successfully.")
|
||||
except psycopg2.Error as e:
|
||||
print(f"An error occurred while connecting to the database: {e}")
|
||||
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT * FROM energy_data LIMIT 5;")
|
||||
results = cursor.fetchall()
|
||||
|
||||
for row in results:
|
||||
print(row)
|
||||
|
||||
cursor.close()
|
||||
conn.close()
|
Loading…
Reference in New Issue
Block a user