aboutsummaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorlian-manonog <lian.manonog@gmail.com>2024-08-27 09:30:34 +0100
committerlian-manonog <lian.manonog@gmail.com>2024-08-27 09:30:34 +0100
commit72ebda950c84d7b519db9a236b35a7fafcbb1899 (patch)
tree0a05ff5e42e8587c53fefa64eb86c6b1dad0d210 /src
parent14fcca02de9e0771df5ad38e092abc285cab172d (diff)
downloadde-project-bentley-72ebda950c84d7b519db9a236b35a7fafcbb1899.tar.gz
de-project-bentley-72ebda950c84d7b519db9a236b35a7fafcbb1899.zip
wip: added a bracket in dataframes
Diffstat (limited to 'src')
-rw-r--r--src/dataframes.py4
-rw-r--r--src/transform_lambda.py3
2 files changed, 4 insertions, 3 deletions
diff --git a/src/dataframes.py b/src/dataframes.py
index f2cae5d..d0479f1 100644
--- a/src/dataframes.py
+++ b/src/dataframes.py
@@ -115,9 +115,9 @@ def create_dim_location(dict_of_df):
df_loc = (
dict_of_df["address"]
.drop(labels=["created_at", "last_updated"], axis=1)
- .rename(columns={"address_id": "location_id"})
+ .rename(columns={"address_id": "location_id"}))
return df_loc
-
+
def create_dim_counterparty(dict_of_df):
diff --git a/src/transform_lambda.py b/src/transform_lambda.py
index 7677f66..57e9042 100644
--- a/src/transform_lambda.py
+++ b/src/transform_lambda.py
@@ -5,12 +5,13 @@ import logging
import pandas as pd
import pyarrow as pa
import pyarrow.parquet as pq
-from src.dataframes import *
+from dataframes import *
from botocore.exceptions import ClientError
from pg8000.native import Connection, InterfaceError
from datetime import datetime
+
class DBConnectionException(Exception):
"""Wraps pg8000.native Error or DatabaseError."""
git.ajschof.me — hosted by ajschofield — powered by cgit