Skip to content

Commit

Permalink
Pass test locally.
Browse files Browse the repository at this point in the history
  • Loading branch information
toddbirchard committed Dec 28, 2024
1 parent febfc79 commit 28decbd
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions clients/database.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
from pandas.core.groupby.generic import DataFrameGroupBy
from sqlalchemy import create_engine
from sqlalchemy.types import DateTime, Integer, String, Text
from sqlalchemy.engine import Engine


class Database:
Expand All @@ -16,8 +17,12 @@ def __init__(self, uri, table, args=None):
self.args = args

@property
def engine(self):
"""Database connection engine"""
def engine(self) -> Engine:
"""
Database connection engine
:returns: Engine
"""
return create_engine(self.uri, connect_args=self.args)

def upload_dataframe(self, df: DataFrame) -> str:
Expand Down

0 comments on commit 28decbd

Please sign in to comment.