YellowAlberto commited on
Commit
2c61f29
·
verified ·
1 Parent(s): 14ed572

Update databaseProyecto.py

Browse files
Files changed (1) hide show
  1. databaseProyecto.py +5 -14
databaseProyecto.py CHANGED
@@ -5,21 +5,12 @@ from sqlalchemy.ext.declarative import declarative_base
5
  from sqlalchemy.orm import sessionmaker
6
 
7
  load_dotenv()
 
8
 
9
- user = os.getenv("DB_USER")
10
- password = os.getenv("DB_PASSWORD")
11
- host = os.getenv("DB_HOST")
12
- port = os.getenv("DB_PORT")
13
- db_name = os.getenv("DB_NAME")
14
- if not host or host == "localhost":
15
- SQLALCHEMY_DATABASE_URL = "sqlite:///./sql_app.db"
16
- # El argumento connect_args es necesario solo para SQLite
17
- engine = create_engine(
18
- SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
19
- )
20
- else:
21
- SQLALCHEMY_DATABASE_URL = f"postgresql://{user}:{password}@{host}:{port}/{db_name}"
22
- engine = create_engine(SQLALCHEMY_DATABASE_URL)
23
 
 
24
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
25
  Base = declarative_base()
 
5
  from sqlalchemy.orm import sessionmaker
6
 
7
  load_dotenv()
8
+ SQLALCHEMY_DATABASE_URL = os.getenv("DATABASE_URL")
9
 
10
+ # Si no hay URL (local), usamos SQLite por si acaso
11
+ if not SQLALCHEMY_DATABASE_URL:
12
+ SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"
 
 
 
 
 
 
 
 
 
 
 
13
 
14
+ engine = create_engine(SQLALCHEMY_DATABASE_URL)
15
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
16
  Base = declarative_base()