Change varible import to new util function
This commit is contained in:
1 parent
16dd4ad287
commit
1dbcb252e3
1 file changed
+5
-7
+5
-7
@@ -4,15 +4,13 @@
|
||||
The base definitions for database access and definition
|
||||
"""
|
||||
from datetime import datetime
|
||||
import os
|
||||
|
||||
from dotenv import load_dotenv
|
||||
|
||||
from sqlalchemy import create_engine, Date, Integer, MetaData, String
|
||||
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, sessionmaker
|
||||
|
||||
from util.environment import load_environment_variables
|
||||
|
||||
load_dotenv()
|
||||
project_variables = load_environment_variables()
|
||||
|
||||
"""
|
||||
Databse naming conventions.
|
||||
@@ -38,7 +36,7 @@ class DataAccessLayer:
|
||||
|
||||
def __init__(self):
|
||||
self.engine = None
|
||||
self.conn_string = os.getenv("DATABASE_URL")
|
||||
self.conn_string = project_variables["LP_DATABASE_URL"]
|
||||
|
||||
|
||||
def connect(self):
|
||||
@@ -53,7 +51,7 @@ dal = DataAccessLayer()
|
||||
class PowerballDraw(Base):
|
||||
__tablename__ = 'powerball_draws'
|
||||
|
||||
draw_date: Mapped[datetime.date] = mapped_column(Date, primary_key=True)
|
||||
draw_date: Mapped[datetime] = mapped_column(Date, primary_key=True)
|
||||
main_ball1: Mapped[int] = mapped_column(Integer)
|
||||
main_ball2: Mapped[int] = mapped_column(Integer)
|
||||
main_ball3: Mapped[int] = mapped_column(Integer)
|
||||
@@ -81,7 +79,7 @@ class PowerballDraw(Base):
|
||||
class MegaMillionsDraw(Base):
|
||||
__tablename__ = 'mega_millions_draws'
|
||||
|
||||
draw_date: Mapped[datetime.date] = mapped_column(Date, primary_key=True)
|
||||
draw_date: Mapped[datetime] = mapped_column(Date, primary_key=True)
|
||||
main_ball1: Mapped[int] = mapped_column(Integer)
|
||||
main_ball2: Mapped[int] = mapped_column(Integer)
|
||||
main_ball3: Mapped[int] = mapped_column(Integer)
|
||||
|
||||
Reference in new issue
Block a user