File size: 554 Bytes
1130076
a00fee9
 
1130076
a00fee9
1130076
 
 
2617189
a00fee9
1130076
 
 
 
 
 
 
a00fee9
1130076
 
 
 
a00fee9
1130076
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import os
import sys

from dotenv import load_dotenv
from sqlalchemy import create_engine

sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from absa.pipeline.db_models import Base


def init_db():
    load_dotenv()
    database_url = os.getenv("DATABASE_URL")
    if not database_url:
        print("DATABASE_URL not set in .env")
        return

    engine = create_engine(database_url)
    Base.metadata.create_all(bind=engine)
    print("Database tables created successfully.")


if __name__ == "__main__":
    init_db()