Skip to content

Commit 545de52

Browse files
author
Uttam Singh
committed
Fixed database import and user model
1 parent e25676d commit 545de52

File tree

2 files changed

+9
-11
lines changed

2 files changed

+9
-11
lines changed

backend/app/database.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,14 @@
11
from sqlalchemy import create_engine
22
from sqlalchemy.orm import sessionmaker, declarative_base
33
import os
4-
from dotenv import load_dotenv
5-
6-
load_dotenv()
74

85
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./app.db")
96

10-
engine = create_engine(DATABASE_URL, pool_pre_ping=True)
7+
engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False} if DATABASE_URL.startswith("sqlite") else {})
118
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
129
Base = declarative_base()
1310

11+
# Dependency for FastAPI routes
1412
def get_db():
1513
db = SessionLocal()
1614
try:

backend/app/models/user.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
1-
from sqlalchemy import Column, Integer, String, Boolean
1+
from sqlalchemy import Column, Integer, String
22
from app.database import Base
33

44
class User(Base):
55
__tablename__ = "users"
6+
67
id = Column(Integer, primary_key=True, index=True)
7-
name = Column(String(120), nullable=False)
8-
email = Column(String(255), unique=True, index=True, nullable=False)
8+
name = Column(String(100), nullable=False)
9+
email = Column(String(100), unique=True, nullable=False)
910
hashed_password = Column(String(255), nullable=False)
10-
department = Column(String(100)) # e.g., Accounts, Compliance, Ops
11-
role = Column(String(50)) # admin, auditor, auditee, manager
12-
manager_email = Column(String(255)) # reporting manager
13-
is_active = Column(Boolean, default=True)
11+
department = Column(String(100), nullable=True)
12+
role = Column(String(50), nullable=False, default="auditee")
13+
manager_email = Column(String(100), nullable=True)

0 commit comments

Comments
 (0)