Upload init_db.py with huggingface_hub

init_db.py

@@ -0,0 +1,97 @@
+"""
+データベース初期化スクリプト
+新しい認証システムとワークスペース機能のテーブルを作成します
+NullAI用のデフォルトユーザーとワークスペースも作成します
+"""
+import sys
+import os
+
+# プロジェクトルートをパスに追加
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from backend.app.database.models import Base, User, Workspace
+
+def init_database():
+    """データベーステーブルを作成"""
+    print("Initializing database...")
+
+    # データベースURLを環境変数または設定から取得（直接インポートを避ける）
+    database_url = os.getenv("DATABASE_URL", "sqlite:///./sql_app.db")
+    print(f"Database URL: {database_url}")
+
+    # エンジンを作成
+    engine = create_engine(
+        database_url,
+        connect_args={"check_same_thread": False} if database_url.startswith("sqlite") else {}
+    )
+
+    # 全テーブルを作成
+    print("Creating tables...")
+    Base.metadata.create_all(bind=engine)
+
+    print("✓ Database tables created successfully!")
+    print("\nCreated tables:")
+    for table in Base.metadata.tables.keys():
+        print(f"  - {table}")
+
+    # NullAI用のデフォルトデータを作成
+    print("\nCreating NullAI default data...")
+    Session = sessionmaker(bind=engine)
+    session = Session()
+
+    try:
+        # デフォルトユーザーを作成（認証なしローカル版用）
+        default_user = session.query(User).filter_by(id="nullai_default_user").first()
+        if not default_user:
+            default_user = User(
+                id="nullai_default_user",
+                email="nullai@localhost",
+                username="nullai",
+                display_name="NullAI System",
+                auth_provider="local",
+                role="admin",
+                is_expert=True,
+                is_guest=False
+            )
+            session.add(default_user)
+            print("  ✓ Created default user: nullai_default_user")
+        else:
+            print("  - Default user already exists")
+
+        # デフォルトワークスペースを作成
+        default_workspace = session.query(Workspace).filter_by(id="default_workspace").first()
+        if not default_workspace:
+            default_workspace = Workspace(
+                id="default_workspace",
+                name="NullAI Default Workspace",
+                slug="default",
+                description="Default workspace for NullAI local operation (no authentication required)",
+                owner_id="nullai_default_user",
+                is_public=True,
+                allow_guest_edit=True,
+                allow_guest_view=True,
+                db_type="sqlite"
+            )
+            session.add(default_workspace)
+            print("  ✓ Created default workspace: default_workspace")
+        else:
+            print("  - Default workspace already exists")
+
+        session.commit()
+        print("\n✓ NullAI default data initialized successfully!")
+
+    except Exception as e:
+        session.rollback()
+        print(f"\n⚠️  Error creating default data: {e}")
+        print("  You may need to create default user and workspace manually.")
+    finally:
+        session.close()
+
+    print("\nYou can now start the application with:")
+    print("  python -m uvicorn backend.app.main:app --reload")
+    print("  or use: ./start_null_ai.sh")
+
+if __name__ == "__main__":
+    init_database()