54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
from fastapi import FastAPI, Depends
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from .database import engine, Base, SessionLocal
|
|
from .routers import auth, users, apps, sso
|
|
from . import models, auth_utils
|
|
from sqlalchemy.orm import Session
|
|
|
|
# Create tables
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
app = FastAPI(title="ASF SSO Service")
|
|
|
|
# CORS
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Routers
|
|
app.include_router(auth.router)
|
|
app.include_router(users.router)
|
|
app.include_router(apps.router)
|
|
app.include_router(sso.router)
|
|
|
|
# Mount static files (Frontend)
|
|
app.mount("/", StaticFiles(directory="../frontend", html=True), name="static")
|
|
|
|
# Create initial admin user if not exists
|
|
def create_initial_admin():
|
|
db = SessionLocal()
|
|
try:
|
|
admin = db.query(models.User).filter(models.User.username == "admin").first()
|
|
if not admin:
|
|
print("Creating initial admin user...")
|
|
hashed_pwd = auth_utils.get_password_hash("admin") # Default password, should be changed
|
|
admin_user = models.User(
|
|
username="admin",
|
|
email="admin@nabd-co.com",
|
|
hashed_password=hashed_pwd,
|
|
is_admin=True,
|
|
is_active=True
|
|
)
|
|
db.add(admin_user)
|
|
db.commit()
|
|
print("Admin user created.")
|
|
finally:
|
|
db.close()
|
|
|
|
create_initial_admin()
|